leetcode-467
题目描述:
把字符串 s 看作是“abcdefghijklmnopqrstuvwxyz”的无限环绕字符串,所以 s 看起来是这样的:"...zabcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyzabcd....". 现在我们有了另一个字符串 p 。你需要的是找出 s 中有多少个唯一的 p 的非空子串,尤其是当你的输入是字符串 p ,你需要输出字符串 s 中 p 的不同的非空子串的数目。 注意: p 仅由小写的英文字母组成,p 的大小可能超过 10000。
from collections import defaultdict
class Solution:
def findSubstringInWraproundString(self, p: str) -> int:
dp = defaultdict(int)
n = len(p)
mx = 0
for i in range(n):
if i> 0 and (ord(p[i])-ord(p[i-1])==1or (p[i-1]=="z" and p[i]=="a")):
mx += 1
else:
mx = 1
dp[p[i]] = max(mx,dp[p[i]])
return sum(dp.values())