环绕字符串中唯一的子字符串(leetcode)

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

题目描述：

把字符串 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”。.

题目分析：

示例三： p = "zab"

在字符串 S 中有六个子串“z”、“a”、“b”、“za”、“ab”、“zab”。
经过分析可以看出————在分析串连续的情况下b结尾的字串有三个，以a结尾的字串有两个。
故只需要遍历分析串，计算以p[i]结尾的字串的最大长度是多少，最后累加即可得出结果》

源码：

#include<iostream>
#include<vector>
#include<cstring>
using namespace std;
int main()
{
    string p("zab");
    int dp[26],ans = 0;//dp表示以p[i]结尾的长度最长的字串的长度
    memset(dp, 0, sizeof(dp));
    int k = 0;
    for (int i = 0;i < p.length();i++)
    {
        if (i == 0 || (p[i] - 97 + 25) % 26 == (p[i - 1] - 97) % 26)
            k++;
        else k = 1;
        dp[p[i] - 97] = max(k, dp[p[i] - 97]);
    }
    for (int i : dp)
        ans += i;
    cout << ans;
}