HDU - 3336 扩展KMP：问所有前缀在字符串出现的次数

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3336
题目大意：问所有前缀出现的次数。

思路：
求出扩展kmp的Next数组，然后把Next的值全部加起来就是答案．
扩展kmp的Next[i]表示的是，以i为开头的后缀，与整个字符串的lcp的长度，那么显而易见，整个字符串长度为1,2,…,Next[i]的前缀在这里都出现了一次，所以此时答案加上Next[i];

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

char s[200005];
int Next[200005];

//预处理计算Next数组
void getNext(char str[])
{
    int i=0,j,po,len=strlen(str);
    Next[0]=len; //初始化Next[0]
    while(str[i]==str[i+1] && i+1<len) i++; Next[1]=i; //计算Next[1]
    po=1; //初始化po的位置
    for(i=2;i<len;i++)
    {
        if(Next[i-po]+i < Next[po]+po) //第一种情况，可以直接得到Next[i]的值
            Next[i]=Next[i-po];
        else //第二种情况，要继续匹配才能得到Next[i]的值
        {
            j = Next[po]+po-i;
            if(j<0) j=0; //如果i>po+Next[po],则要从头开始匹配
            while(i+j<len && str[j]==str[j+i]) j++; Next[i]=j;
            po=i; //更新po的位置
        }
    }
}

int main()
{
    int t;
    scanf("%d", &t);
    while(t--){
        int n;
        scanf("%d", &n);
        scanf("%s", s);
        getNext(s);
        LL sum=0;
        for(int i=0; i<n; i++){
            sum=(sum+Next[i])%10007;
        }
        printf("%lld\n", sum);
    }

    return 0;
}