【刷题】467. 环绕字符串中唯一的子字符串

---

题目

leetCode链接：https://leetcode.cn/problems/unique-substrings-in-wraparound-string/

467. 环绕字符串中唯一的子字符串

把字符串 s 看作是 “abcdefghijklmnopqrstuvwxyz” 的无限环绕字符串，所以 s 看起来是这样的：

“…zabcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyzabcd…” .

现在给定另一个字符串 p 。返回 s 中 唯一 的 p 的 非空子串 的数量 。

示例 1:

输入: p = "a"
输出: 1
解释: 字符串 s 中只有一个"a"子字符。

示例 2:

输入: p = "cac"
输出: 2
解释: 字符串 s 中的字符串“cac”只有两个子串“a”、“c”。.

示例 3:

输入: p = "zab"
输出: 6
解释: 在字符串 s 中有六个子串“z”、“a”、“b”、“za”、“ab”、“zab”。

---

思路

看似一个滑动窗口，但是越做越不对劲，发现情况非常复杂。。
看了答案，是一个dp+滑动窗口，不过个人dp一塌糊涂，差点放弃。
看了代码发现其实思路还是很清楚的。
对于一个长度为n的子串，不难看出其中有1+2…+n个子串符合题目要求。
那么用一个哈希桶来维护以每个以某个字母结尾的子串最大长度，最后统计桶中数量即可。
具体实现可以从第二位开始遍历字符串p，如果发现遍历到的字符（假设记为i）正好ASCII值和前一位相差1或者25，则内置计时器(记为len)+1，并记录count[i]=max(count[i],len)。
举例来说，对于字符串abcab
遍历开始，由于abc每一位都是前一位ASCII值+1，则遍历到c记录结果为count[“a”]=1, count[“b”]=2, count[“c”]=3。
继续遍历，由于下一位a和前一位不构成子串，充值计时器len，记录为count[“a”]=max(count[“a”],1)=1。
最终count中以a结尾子串最大长度为1，b为2，c为3，则最终结果为1+2+3=6。

---

解题

python实现

class Solution:
    def findSubstringInWraproundString(self, p):
        count = collections.defaultdict(int)
        N = len(p)
        _len = 0
        for i in range(N):
            # if i > 0 and (ord(p[i]) - ord(p[i - 1]) == 1 or ord(p[i]) - ord(p[i - 1]) == -25 ):
            if i > 0 and (ord(ch) - ord(p[i - 1])) % 26 == 1: # 抄答案，更好的写法
                _len += 1
            else:
                _len = 1
            count[p[i]] = max(count[p[i]], _len)
        return sum(count.values())

golang实现

func findSubstringInWraproundString(p string) (ans int) {
    dp := [26]int{}
    k := 0
    for i, ch := range p {
        if i > 0 && (byte(ch)-p[i-1]+26)%26 == 1 { // 字符之差为 1 或 -25
            k++
        } else {
            k = 1
        }
        dp[ch-'a'] = max(dp[ch-'a'], k)
    }
    for _, v := range dp {
        ans += v
    }
    return
}

func max(a, b int) int {
    if b > a {
        return b
    }
    return a
}

---

复杂度

时间复杂度：$O(n)$,其中 n 为字符串p长度。

空间复杂度：$O(1)$，常数级别空间大小。

---

总结

不看答案还是很难想到这种思路。还需要好好学习一下dp才行