Codeforces Round #578 (Div. 2) E. Compress Words（KMP优化）

题目链接

题意：给定n个字符串，现在要合并这n个字符串，如果第i个字符串的某个前缀和第i-1个字符串的某个后缀相等的话，合并的时候只能并一个，如want to合并后就是wanto，输出合并后的字符串。
思路：KMP模板题，每次匹配与前面已经合并好的字符串ans进行kmp合并，不过这里好优化一下匹配位置，要从ans的起点max（0，ans,size()-strlen(s))到终点ans.size()匹配，否则会tle。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+1;
char s[maxn];
int n,len,Next[maxn];
void getNext()
{
	int j=-1;
	Next[0]=-1;
	for(int i=1;i<len;++i)
	{
		while(j!=-1&&s[i]!=s[j+1]) j=Next[j];
		if(s[i]==s[j+1]) j++;
		Next[i]=j;
	}
}
int main()
{
	string ans;
	scanf("%d",&n);
	scanf("%s",s);
	ans=(s);
	for(int j=2;j<=n;++j)
	{
		scanf("%s",s);
		len=strlen(s);
		getNext();
		int size=-1,len1=ans.size();
		for(int i=max(0,len1-len);i<len1;++i)
		{
			while(size!=-1&&ans[i]!=s[size+1]) size=Next[size];
			if(ans[i]==s[size+1]) size++; 
		}
		for(int i=size+1;i<len;++i) ans+=s[i];
	}
	cout<<ans<<endl;
}