关注
先考虑不存在连续且重复的子串的字符串(即字符串的每个字符与其左右相邻的字符均不相同)的数目,令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
相关推荐
点赞 评论 收藏
分享
05-12 17:12
河南科技大学 Java 点赞 评论 收藏
分享
04-25 07:58
清华大学 C++ 点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 聊聊这家公司值得去吗 #
228116次浏览 2150人参与
# 秋招最大的收获是什么? #
33730次浏览 295人参与
# 你认为哪个岗位找工作最卷 #
7591次浏览 28人参与
# 职场上哪些事情令人讨厌 #
16383次浏览 79人参与
# 一人一个landing小技巧 #
77723次浏览 1111人参与
# 职场人,说说你的烦心事 #
8119次浏览 62人参与
# kpi面有什么特征 #
28848次浏览 160人参与
# 为了找工作你投递了多少公司? #
7597次浏览 100人参与
# 大家每天通勤多久? #
41796次浏览 328人参与
# tplink提前批进度交流 #
162474次浏览 1375人参与
# 找工作前vs找工作后的心路变化 #
9508次浏览 101人参与
# 聊聊你的职场新体验 #
157174次浏览 1367人参与
# 经纬恒润求职进展汇总 #
118822次浏览 1027人参与
# 通信硬件牛牛的实习日记 #
7034次浏览 65人参与
# 硬件人你反向读研了吗 #
41694次浏览 629人参与
# 入职第一天,你准备什么时候下班 #
55233次浏览 351人参与
# 好未来求职进展汇总 #
17982次浏览 153人参与
# 担心入职之后被发现很菜怎么办 #
126148次浏览 752人参与
# 晒一晒你收到的礼盒 #
67751次浏览 398人参与
# 一觉醒来,秋招难度下降一万倍…… #
77092次浏览 632人参与