关注
先考虑不存在连续且重复的子串的字符串(即字符串的每个字符与其左右相邻的字符均不相同)的数目,令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
相关推荐
牛客热帖
正在热议
# 软件开发薪资爆料 #
1426109次浏览 16473人参与
# vivo求职进展汇总 #
46070次浏览 347人参与
# 联影秋招 #
15842次浏览 260人参与
# 晒一晒我的offer #
5993139次浏览 75534人参与
# 你都收到了哪些公司的感谢信? #
46490次浏览 687人参与
# 我想象的工作vs实际工作 #
265396次浏览 3518人参与
# 锐明校招来了 #
12963次浏览 241人参与
# 许愿池 #
145307次浏览 2128人参与
# 视觉/交互/设计工作体验 #
10618次浏览 162人参与
# 机械人怎么评价今年的华为 #
111103次浏览 857人参与
# 校招过来人的经验分享 #
570808次浏览 16890人参与
# 你最近一次加班是什么时候? #
10060次浏览 101人参与
# 没有实习经历,还有机会进大厂吗 #
646356次浏览 11723人参与
# 国企还是互联网,你怎么选? #
38767次浏览 321人参与
# 职场新人生存指南 #
152750次浏览 4858人参与
# 选了这个offer,你有没有后悔? #
204176次浏览 1939人参与
# 安利/避雷我的岗位 #
274491次浏览 3924人参与
# 25届暑期实习 #
730239次浏览 14768人参与
# 投了多少份简历才上岸 #
98928次浏览 1431人参与
# 投递几十家公司,到现在0offer,大家都一样吗 #
29076次浏览 294人参与
# 你觉得技术面多长时间合理? #
11188次浏览 72人参与
# 正在实习的碎碎念 #
1097354次浏览 11976人参与