friends 一道要想清楚的 hash

Friends

描述
有三个好朋友喜欢在一起玩游戏，A 君写下一个字符串 S ，B 君将其复制一遍得到 T，C 君在 T 的任意位置（包括首尾）插入一个字符得到 U。现在你得到了 U，请你找出 S 。

输入
第一行一个数 N ，表示 U 的长度。 第二行一个字符串 U，保证 U 由大写字母组成。

输出
输出一行，若 S 不存在,输出 NOT POSSIBLE。若 S 不唯一，输出 NOT UNIQUE，否则输出 S 。

样例输入 [复制]
样例输入 1
7
ABXCABC
样例输出 1
ABC
样例输入 2
6
ABCDEF
样例输出 2
NOT POSSIBLE
样例输入 3
9
ABABABABA
样例输出 3
NOT UNIQUE
样例输出 [复制]
提示
2≤N≤2000001。

标签
BalticOI 2014 BOI2014

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int mod=998244353;
const int sl=33;
inline int read(){
	int w=1,data=0;char ch=0;
	ch=getchar();
	while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
	if(ch=='-') w=-1,ch=getchar();
	while(ch>='0'&&ch<='9') data=data*10+ch-'0',ch=getchar();
    return w*data;
}
int n;
const int N=2000011;
int pre[N];
char a[N];
int mul[N];
signed main(){
	int n=read();
	scanf("%s",a+1);
	mul[0]=1;
	for(int i=1;i<=n;i++){
		pre[i]=(pre[i-1]*sl%mod+a[i]-'A')%mod;
		mul[i]=mul[i-1]*sl%mod; 
	}
	if(!(n%2)){
		puts("NOT POSSIBLE");return 0;
	}          
	int flag=0,dfn;     
	int x,y,ans;    
	for(int i=1;i<=n;i++){
		if(i<=n/2) x=(pre[n/2+1]-pre[i]*mul[n/2+1-i]%mod+pre[i-1]*mul[n/2+1-i]%mod+mod)%mod;
		else x=pre[n/2];
		if(i<=n/2+1) y=(pre[n]-pre[n/2+1]*mul[n/2]%mod+mod)%mod; 
		else y=(pre[n]-pre[i]*mul[n-i]%mod+(pre[i-1]-pre[n/2]*mul[i-1-n/2]%mod+mod)*mul[n-i]%mod+mod)%mod;
		if(x==y){
			if(flag&&ans!=x){
				puts("NOT UNIQUE"); return 0;
			}
		    flag=1;ans=x;dfn=i;
		} 
	}
	if(!flag) {
		puts("NOT POSSIBLE");
		return 0;
	}
    if(dfn<=n/2){
    	for(int i=n/2+1+1;i<=n;i++){
    		cout<<a[i];
		}
	}
	else for(int i=1;i<=n/2;i++){
		cout<<a[i];
	}
	return 0;
}