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

相关推荐

程序员牛肉:这一眼假啊,基本上都是骗人的,不然就涉及到职位贪腐了,就像之前华为的OD事件,看你运气好不好了
点赞 评论 收藏
分享
爱吃肉的伊登在写日记:好棒,27届简历能做成这个样子,但是第一个项目感觉cover住难度还是不小的,特别是二面的时候肯定要对分布式系统设计这一块儿有高出正常面试者的水平才行
点赞 评论 收藏
分享
不想投了,不想面了,不想找了感觉自己像个小丑
用微笑面对困难:不是你去大学生就业平台看看啊,boss很多就是冲kpi的
点赞 评论 收藏
分享
牛客网
牛客网在线编程
牛客网题解
牛客企业服务