HDU 6629 string matching（扩展kmp）

题意：给你一个字符串 问从第2位 每一位和字符串相等前缀多长。 （比较多少次 直到完全匹配 或 失败）

分析：扩展kmp板子题，注意完全匹配时不需要加一次失败匹配，很久不摸都忘了。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e6 + 10;
char s[N];
int nt[N];
ll ans;
void solve(){
    int l = strlen(s);
    nt[0] = l;
    int mx = 0,id;
    for(int i = 1;i < l;i++){
        if(i < mx) nt[i] = min(mx-i,nt[i-id]);
        else nt[i] = 0;
        while(s[i+nt[i]] == s[nt[i]]) nt[i]++;
        if(mx < i + nt[i]) {
            id = i;
            mx = i + nt[i];
        }
        ans += (ll)min(nt[i]+1,l-i);
    }
}

int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%s",s);
        ans = 0;
        solve();
        ll n = strlen(s);
        printf("%lld\n",ans);
    }
    return 0;
}