先考虑不存在连续且重复的子串的字符串(即字符串的每个字符与其左右相邻的字符均不相同)的数目,令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

相关推荐

不愿透露姓名的神秘牛友
09-30 19:49
起名星人:蛮离谱的,直接要求转投销售
投递汇川技术等公司10个岗位
点赞 评论 收藏
分享
在校生实习:我觉得平时学校肯定有各种大作业吧。包装一下写项目里。特长那块喧宾夺主了,项目肯定是大头。特长里比如:熟悉vscode,这个感觉不具有吸引性。简要介绍你会什么语言,什么工具等就行了。同26找实习,我是个超级菜鸡😭大家一起加油
点赞 评论 收藏
分享
牛客网
牛客企业服务