2019HDU多校第五场 HDU 6629. string matching (扩展KMP应用//Z函数)

这道题坎扩展KMPnxt数组规律看了一会 也意识到 自己对展开kmp各种应用还是不够熟练
题意：给出一个用于求在一个给定字符串中求每一位的后缀（含自己）与这个字符串的最长相同的长度的算法，问你对于给出的字符串执行这个算***进行多少次这个算法中那个字符与字符的比较过程
扩展KMP里求NXT数组的代码跑一遍，我们就可以得到每一位的最长字串的长度，然后因为问的是比较次数，那么只要这一位的位置加上它的子串长没用超过字符串的最后一位，就说明它会在统计时还要加上一次比较次数，也就是在统计它的字串长度时它和下一位比较并且匹配失败的那一次。
nxt[i]表示T[i,len-1]与T[0,len-1]的最长公共前缀。

#include<bits/stdc++.h>
#define N 1000010
#define int long long
using namespace std;

int q, nxt[N], extend[N];
string s, t;

void getnxt() {
    nxt[0] = t.size(); //nxt[0]一定是T的长度
    int now = 0;
    while(t[now] == t[1 + now] && now + 1 < (int)t.size())
        now++;//这就是从1开始暴力
    nxt[1] = now;
    int p0 = 1;
    for(int i = 2; i < (int)t.size(); i++) {
        if(i + nxt[i - p0] < nxt[p0] + p0)
            nxt[i] = nxt[i - p0]; //第一种情况
        else {
            //第二种情况
            int now = nxt[p0] + p0 - i;
            now = max(now, 0ll); //这里是为了防止i>p的情况
            while(t[now] == t[i + now] && i + now < (int)t.size())
                now++;//暴力
            nxt[i] = now;
            p0 = i; //更新p0
        }
    }
}
signed main() {
    int cas;
    ios::sync_with_stdio(false);cin.tie(0);
    cin >>cas;
    while(cas--) {
        cin >> t;
        memset(nxt, 0, sizeof(nxt));
        getnxt();
        int len = t.size();
        int ans =0;
        for(int i = 1; i < len; i++) {
            // printf("%d ", nxt[i]); //输出nxt
            // ans ++;
            if(nxt[i]) {
                ans += nxt[i];
            }
            if(nxt[i]!=len-i) ans ++; // nxt[i] + i < len 
        }
        cout << ans << endl;
    // puts("");
    }
    return 0;
}