关注
先考虑不存在连续且重复的子串的字符串(即字符串的每个字符与其左右相邻的字符均不相同)的数目,令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
相关推荐
点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 如果秋招能重来,我会____ #
10868次浏览 102人参与
# 苦尽甘来时,再讲来时路 #
10730次浏览 182人参与
# “vivo”个offer #
19478次浏览 150人参与
# 如果上班像打游戏,你最想解锁什么技能 #
2595次浏览 32人参与
# 我是面试官,请用一句话让我破防 #
2186次浏览 19人参与
# 为了实习逃课值吗? #
12072次浏览 98人参与
# 快手技术岗信息交流阵地 #
12498次浏览 74人参与
# 校招生月薪1W算什么水平 #
3051次浏览 22人参与
# 机械求职避坑tips #
71416次浏览 485人参与
# 一份好的简历长什么样? #
6873次浏览 171人参与
# 选完offer后,你后悔学机械吗? #
43105次浏览 249人参与
# 秋招许愿,本周能____ #
14430次浏览 94人参与
# 选择和努力,哪个更重要? #
135058次浏览 1037人参与
# 班味很重的人是啥样的? #
4346次浏览 30人参与
# 应届生第一份工资要多少合适 #
3654次浏览 36人参与
# 投递无反馈,如何优化求职策略? #
2454次浏览 26人参与
# 材料专业可以靠半导体脱坑吗? #
26925次浏览 138人参与
# 机械制造秋招总结 #
82597次浏览 817人参与
# 大学最后一个寒假,我想…… #
60677次浏览 654人参与
# 职场新人体验 #
120631次浏览 826人参与
# 你觉得实习能学到东西吗 #
114657次浏览 1248人参与
# 新凯来求职进展汇总 #
58094次浏览 150人参与
海康威视公司福利 1125人发布
查看19道真题和解析