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

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

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

现在我们有了另一个字符串 p 。你需要的是找出 s 中有多少个唯一的 p 的非空子串，尤其是当你的输入是字符串 p ，你需要输出字符串 s 中 p 的不同的非空子串的数目。

注意: p 仅由小写的英文字母组成，p 的大小可能超过 10000。

示例 1:

输入: “a”
输出: 1
解释: 字符串 S 中只有一个"a"子字符。

示例 2:

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

示例 3:

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/unique-substrings-in-wraparound-string

问题分析：

• 注意是 p 的不同的子串
• p 的子串在 z 中存在且连续

一个字符串的字串数量是以每个字符结尾的最长子串的长度之和

abcd：a的最长子串长度为1，b的最长字串长度为2，同理c为3，d为4
所以子串数目是1 + 2 + 3 + 4 = 10
子串：a、ab、b、abc、bc、c、abcd、bcd、cd、d

代码如下：

class Solution {
public:
    int findSubstringInWraproundString(string p) {
        vector<int> dp(26, 0);
        int cnt = 0;
        int n = p.size();
        for(int i = 0; i < n; i++){
            if(i > 0 && isContinue(p[i - 1], p[i])){
                cnt++;
            }
            else{
                cnt = 1;
            }
            dp[p[i] - 'a'] = max(cnt, dp[p[i] - 'a']);
        }
        return accumulate(dp.begin(), dp.end(), 0);
    }
    //判断是否连续
    bool isContinue(char prev, char curr){
        if(curr == 'a'){
            return prev == 'z';
        }
        return prev + 1 == curr;
    }
};