KMP 牛客 Just A String

链接：https://ac.nowcoder.com/acm/problem/13821
来源：牛客网

题目描述

何老师手中有一个字符串S，他发现这个字符串有一个神奇的性质，取出一个长为i的前缀（就是由S的前i个字符顺序构成的字符串）prei和一个长为j的后缀（就是由S的后j个字符顺序构成的字符串）sufj之后，总是存在三个字符串A,B,C（可能为空）使得prei=A+B,sufj=B+C， 虽然这听起来像是一句废话。
显然三元组A,B,C不总是唯一的，何老师从所有可能的三元组中找到B最长的，很容易知道这样的三元组是唯一的，并且认为prei和sufj的契合度就是f(i,j)=|A||B|2|C|，现在你需要帮何老师算出所有f(i,j)(0 ≤ i,j ≤ n)的异或和。
这里|X|表示字符串X的长度，X+Y表示将两个字符串X和Y顺序拼接起来后得到的新字符串。

输入描述:

第一行是一个正整数T(≤ 500)，表示测试数据的组数， 每组测试数据，包含一个仅由小写字母构成的非空字符串S(|S| ≤ 2000)， 保证满足|S|>200的数据不超过5组。

输出描述:

对于每组测试数据，输出所有f(i,j)(0 ≤ i,j ≤ n)的异或和。
示例1
输入
复制
1
abcab
输出
复制
13

分析：

枚举s的所有后缀，对每个后缀计算next数组，在s中匹配字符串（后缀），v[i]表示前缀s(1,i)与当前的后缀能找到的最大B。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 2e3+10;
char a[N];
int v[N],ne[N];
void get_ne(char s[])
{
	int n=strlen(s);
	int j=-1;
	ne[0]=-1;
	for(int i=1;i<n;i++)
	{
		while(j!=-1&&s[i]!=s[j+1]) j=ne[j];
		if(s[j+1]==s[i]) j++;
		ne[i]=j;
	}
}
void kmp(char s1[],char s2[])
{
	get_ne(s2);
	int j=-1,i=0;
	int l=strlen(s1);
	int r=strlen(s2);
	for(int i=0;i<l;i++)
	{
		while(j!=-1&&s1[i]!=s2[j+1]) j=ne[j];
		if(s1[i]==s2[j+1]) j++;
		v[i]=j+1; 
	}
}
int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%s",a+1);
		int n=strlen(a+1);
		ll ans=0;
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
			{
				v[i]=0;
			}
			kmp(a+1,a+n-i+1);
			for(int j=0;j<n;j++)
			{
				ll A=j+1-v[j];
				ll B=v[j];
				ll C=i-v[j];
				ll sum=A*B*B*C;
				ans^=sum;
			} 
		}
		printf("%lld\n",ans);
	}
	return 0;
 }