关注
先考虑不存在连续且重复的子串的字符串(即字符串的每个字符与其左右相邻的字符均不相同)的数目,令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
相关推荐
点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# AI面会问哪些问题? #
22043次浏览 438人参与
# 从事AI岗需要掌握哪些技术栈? #
6502次浏览 207人参与
# 米连集团26产品管培生项目 #
12698次浏览 284人参与
# 你的实习产出是真实的还是包装的? #
17569次浏览 320人参与
# 找AI工作可以去哪些公司? #
6382次浏览 156人参与
# 春招至今,你的战绩如何? #
55523次浏览 498人参与
# 厦门银行科技岗值不值得投 #
7015次浏览 181人参与
# 你做过最难的笔试是哪家公司 #
25798次浏览 153人参与
# 面试被问期望薪资时该如何回答 #
382251次浏览 2163人参与
# 阿里笔试 #
173272次浏览 1276人参与
# 一张图晒出你司的标语 #
3431次浏览 63人参与
# 晶盛机电求职进展汇总 #
35154次浏览 318人参与
# 沪漂/北漂你觉得哪个更苦? #
8488次浏览 179人参与
# 长得好看会提高面试通过率吗? #
20603次浏览 239人参与
# AI时代,哪个岗位还有“活路” #
9906次浏览 306人参与
# 春招你拿到offer了吗 #
827836次浏览 9973人参与
# HR最不可信的一句话是__ #
5102次浏览 99人参与
# 学历对求职的影响 #
661611次浏览 4233人参与
# 投递几十家公司,到现在0offer,大家都一样吗 #
338753次浏览 2152人参与
# 什么专业适合考公 #
59268次浏览 307人参与
# 通信和硬件还有转码的必要吗 #
99648次浏览 637人参与
# 聊聊你的职场新体验 #
335379次浏览 1889人参与

