「BZOJ3916」friends（Hash）

题目

题目描述
有三个好朋友喜欢在一起玩游戏,A君写下一个字符串S,B君将其复制一遍得到T,C君在T的任意位置(包括首尾)插入一个字符得到U.现在你得到了U,请你找出S.
输入
第一行一个数N,表示U的长度.
第二行一个字符串U,保证U由大写字母组成
输出
输出一行,若S不存在,输出"NOT POSSIBLE".若S不唯一,输出"NOT UNIQUE".否则输出S.
样例输入
7
ABCXABC
样例输出
ABC
提示
对于100%的数据 2<=N<=2000001

分析

就是求去掉一个点后使剩余部分相等，并且判断剩余部分是否有多个。暴力+Hash即可。
注意！T不一定在S后面。（我因为这个WA了）
另外在输出时还需要注意一下（详见代码）

代码

#include<bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
#define full(a,b) memset(a,b,sizeof a)
#define ll long long
#define ui unsigned int
int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9') f=ch=='-'?-1:1,ch=getchar();
	while(ch>='0'&&ch<='9') x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
	return x*f;
}
#define debug 1
#define N 2000005
#define b 31
int n,w,ans;char arr[N];
ui Hash[N],p[N],lastt;
ui hh(int S,int E)//求X~Y位的Hash值 
{
	if(!S) return Hash[E];
	return Hash[E]-Hash[S-1]*p[E-S+1];
}
int check(int x,int y)
{
	if(lastt)//保存上次的 
	{
		ui t=hh(x,y);
		if(t!=lastt) return 1;//有新答案，数量要+1 
		return 0;
	}
	lastt=hh(x,y);
	return 1;
	//注意，这里不需要数组保存之前的，因为答案多余1就不需要再统计了 
}
int main()
{
	if(debug==-1)
	{
		freopen("friends.in","r",stdin);
		freopen("friends.out","w",stdout);
	}
	p[0]=1;
	for(int i=1; i<N; i++)
		p[i]=p[i-1]*b;
	n=read();
	scanf("\n%s",arr);
	if(!(n&1))//偶数无解 
	{
		puts("NOT POSSIBLE");
		return 0;
	}
	Hash[0]=arr[0];
	for(int i=1; i<n; i++)
		Hash[i]=Hash[i-1]*b+arr[i];
	if(hh(1,n/2)==hh(n/2+1,n-1)) if(check(1,n/2)) w=0,++ans;
	if(hh(0,n/2-1)==hh(n/2,n-2)) if(check(0,n/2-1)) w=n-1,++ans;
	if(hh(0,n/2-1)==hh(n/2+1,n-1)) if(check(0,n/2-1)) w=n/2,++ans;//特判头尾和中间 
	for(int i=1; i<n/2; i++)//在左半部分枚举 
	{
		ui t1=hh(0,i-1)*p[n/2-i]+hh(i+1,n/2);
		ui t2=hh(n/2+1,n-1);
		if(t1==t2&&check(n/2+1,n-1)) ++ans,w=i;
	}
	for(int i=n/2+1; i<n-1; i++)//右半部分枚举 
	{
		ui t1=hh(0,n/2-1);
		ui t2=hh(n/2,i-1)*p[n-1-i]+hh(i+1,n-1);
		if(t1==t2&&check(0,n/2-1)) ++ans,w=i;
	}
	if(!ans) puts("NOT POSSIBLE");
	else if(ans>1) puts("NOT UNIQUE");
	else
	{
		if(w>n/2) w=n/2;//U在右半部分，不输出中点 
		for(int i=0; i<=n/2; i++)
			if(i!=w) putchar(arr[i]);//不输出U
		//当然也可以分类讨论输出 
	}
	return 0;
}