Java实现 LeetCode 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”。.

PS：

唯一的子字符串的数量就等于：分别以a，b， c， d。。。z结尾的最长子字符串的长度之后；注意， 不能统计重复的， 那么， 以z为例， 以z结尾的最长子字符串的长度， 就是， 以z结尾的“唯一”的子字符串的数量；

class Solution {
    public int findSubstringInWraproundString(String p) {

        int n = p.length();
        if(n < 1) return 0;
        int ret = 0;
        int[] count = new int[26];
        char[] str = p.toCharArray();
        int curMaxLen = 1;
        for(int i = 0; i < n; ++i) {
            if(i > 0 && (str[i]-str[i-1] == 1 || str[i-1]-str[i] == 25))
                curMaxLen++;
            else
                curMaxLen = 1;
            count[str[i]-'a'] = Math.max(count[str[i]-'a'], curMaxLen);
        }
        for(int temp : count)
            ret += temp;
        return ret;
    }
}