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

相关推荐

大鹏随风起:不用打开评论区我就知道会有什么评论
点赞 评论 收藏
分享
08-14 14:32
东北大学 Java
ybluea:当你的列表里出现一个🌸的hr,就说明你的列表里有无数🌸的hr
点赞 评论 收藏
分享
牛客网
牛客企业服务