1. 程式人生 > >467.環繞字串中唯一的子字串

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”。.

class Solution {
public:
    int findSubstringInWraproundString(string p) {
        vector<int> cnt(26, 0);
        int len = 0;
        for (int i = 0; i < p.size(); ++i) {
            if (i > 0 && (p[i] == p[i - 1] + 1 || p[i - 1] - p[i] == 25)) {
                ++len;
            } else {
                len = 1;
            }
            cnt[p[i] - 'a'] = max(cnt[p[i] - 'a'], len);
        }
        return accumulate(cnt.begin(), cnt.end(), 0);
    }
};