关注
先考虑不存在连续且重复的子串的字符串(即字符串的每个字符与其左右相邻的字符均不相同)的数目,令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
相关推荐
点赞 评论 收藏
分享
点赞 评论 收藏
分享
点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 实习生的蛐蛐区 #
984048次浏览 4933人参与
# 父母对你找工作是助力还是阻力? #
50133次浏览 424人参与
# 27届实习投递记录 #
154922次浏览 1600人参与
# 你上一次给父母打电话是什么时候 #
45811次浏览 281人参与
# 万物皆可发面经 #
1580次浏览 22人参与
# 找工作时的取与舍 #
139555次浏览 927人参与
# 从mentor身上学到了__ #
66413次浏览 914人参与
# 我和mentor的爱恨情仇 #
120234次浏览 1011人参与
# 你觉得mentor喜欢什么样的实习生 #
62678次浏览 1052人参与
# 你的mentor是什么样的人? #
65353次浏览 811人参与
# 实习,不懂就问 #
223787次浏览 1732人参与
# 多益网络工作体验 #
74655次浏览 316人参与
# 多益网络求职进展汇总 #
109529次浏览 409人参与
# 如何一边实习一边找下家? #
131986次浏览 648人参与
# 一起聊华为 #
222198次浏览 973人参与
# 求职中的尴尬瞬间 #
42838次浏览 127人参与
# 薪资一样,你会选择去大厂还是小公司 #
36098次浏览 133人参与
# 实习的内耗时刻 #
243438次浏览 1670人参与
# 发工资后,你做的第一件事是什么 #
108237次浏览 348人参与
# 第一次找实习,我建议__ #
88598次浏览 885人参与

