09-18 10:47
华为云计算技术有限公司_算法工程师 0 点赞 评论 收藏
分享
09-14 11:38
华为云计算技术有限公司_算法工程师 0 点赞 评论 收藏
分享
swott:先考虑不存在连续且重复的子串的字符串(即字符串的每个字符与其左右相邻的字符均不相同)的数目,令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()
投递拼多多集团-PDD等公司10个岗位 >
0 点赞 评论 收藏
分享
我家栗子大美人:欢迎大家投递~
投递百度等公司10个岗位 >
0 点赞 评论 收藏
分享
我家栗子大美人:我加hr了,说的确实是月底出结果
投递华为北京研究所等公司10个岗位 >
0 点赞 评论 收藏
分享
关注他的用户也关注了: