关注
先考虑不存在连续且重复的子串的字符串(即字符串的每个字符与其左右相邻的字符均不相同)的数目,令dp[i]表示长度为i,字符集大小为s的这样的字符串的数目,容易得到其递推式子为dp[i]=(s - 1) * dp[i - 1],
dp[1] = s。之后考虑从这样的字符串中任意挑选一个字符c,将c替换成ccc,另外,可以选择将其他的字符c替换成cc。这样时间复杂度大概为O(l^2),代码如下:
def C(n, j):
p1, p2 = 1, 1
if j > n // 2:
j = n - j
for i in range(j):
p1 *= n - i
p2 *= i + 1
return p1 // p2
def solve(s, l):
if l < 3:
return 0
dp = [0] * (l + 1)
dp[1] = s
for i in range(2, l + 1):
dp[i] = dp[i - 1] * (s - 1)
ret = 0
for i in range((l - 3) // 2 + 1):
t = l - i - 2
ret += (i + 1) * C(t, i + 1) * dp[t]
return ret % (10 ** 9 + 7)
if __name__ == "__main__":
line = input().strip()
while line:
l, s = map(int, line.split())
print(solve(s, l))
line = input().strip()
查看原帖
2 1
相关推荐
Lorn的意义:今年是未来十年最好的一年

点赞 评论 收藏
分享
07-10 11:59
复旦大学 Java 点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 大厂面试初体验 #
5373次浏览 42人参与
# 如果可以,你希望哪个公司来捞你 #
100870次浏览 457人参与
# 如何提高实习转正率? #
2277次浏览 30人参与
# leader认为你工作不认真怎么办 #
30872次浏览 140人参与
# 你遇到过哪些神仙同事 #
100332次浏览 724人参与
# 我的国央企投递进展 #
46665次浏览 292人参与
# 国企是理工四大天坑的最好选择吗 #
13702次浏览 95人参与
# 五一之后,实习真的很难找吗? #
78521次浏览 515人参与
# 机械人,你被简历秒挂的企业有哪些? #
43010次浏览 281人参与
# 招聘要求与实际实习内容不符怎么办 #
113005次浏览 770人参与
# 如果公司给你放一天假,你会怎么度过? #
17107次浏览 128人参与
# 找工作时的取与舍 #
80466次浏览 568人参与
# 投递几十家公司,到现在0offer,大家都一样吗 #
246317次浏览 1792人参与
# 三一重工求职进展汇总 #
15068次浏览 67人参与
# OPPO求职进展汇总 #
662889次浏览 5041人参与
# 你的秋招第一场笔试是哪家 #
142780次浏览 1453人参与
# 总结:哪家公司面试体验感最差 #
61093次浏览 276人参与
# 如果重来一次你还会读研吗 #
176922次浏览 1786人参与
# 机械人,说说你的烦心事 #
69713次浏览 839人参与
# 面试时被问的最奇葩的问题 #
22995次浏览 130人参与