前言
自己打的hash有点乱还总出错,以这道题为例,转了个模板(用了ull,省的取模)。
https://blog.csdn.net/sdz20172133/article/details/99182258?utm_medium=distribute.pc_relevant.none-task-blog-utm_term-1&spm=1001.2101.3001.4242
题目描述
有三个 好朋友在一起玩游戏,A君一个字符串S,B君将其复制一遍得到T,C君在T的任意位置(包括首尾)插入一个字符得到字符串U。现在你得到了U,请找出S。
输入
第一行一个数N,表示字符串U的长度。
第二行一个字符串U,保证U由大写字母组成。
输出
输出一行,若S不存在,则输出"NOT POSSIBLE"。若S不唯一,则输出"NOT UNIQUE",否则输出S。
样例输入
7
ABXCABC
样例输出
ABC
#include<bits/stdc++.h>
const int N = 2e6+5;
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
const int P=131;
ULL hashv[N],power[N];
char str[N];
int n;
void calHash()
{
power[0]=1;
for(int i=1; i<=n; i++)
{
hashv[i]=hashv[i-1]*P+(str[i]-'A'+1);
power[i]=power[i-1]*P;
}
}
ULL getHash(int l,int r)
{
if(r>=l)
return hashv[r]-hashv[l-1]*power[r-l+1];
else
return 0;
}
set<ULL> s;
int main()
{
scanf("%d",&n);
scanf("%s",str+1);
calHash();
if(n%2==0)
{
printf(("NOT POSSIBLE\n"));
return 0;
}
int len=(n-1)/2;
int pos;
for(int i=1; i<=len+1; i++)
{
//cout<<getHash(len+2,n)<<" "<<getHash(1,i-1)*power[len-(i-1)]<<" "<<getHash(i+1,1+len)<<endl;
if(getHash(len+2,n)==getHash(1,i-1)*power[len-(i-1)]+getHash(i+1,len+1))
{
pos=i;
s.insert(getHash(len+2,n));
}
if(s.size()>=2)
break;
}
if(s.size()>=2)
{
printf("NOT UNIQUE\n");
return 0;
}
for(int i=len+2; i<=n; i++)
{
//cout<<getHash(len+2,n)<<" "<<getHash(1,i-1)*power[len-(i-1)]<<" "<<getHash(i+1,1+len)<<endl;
if(getHash(1,len)==getHash(len+1,i-1)*power[len-(i-(len+1))]+getHash(i+1,n))
{
pos=i;
s.insert(getHash(1,len));
}
if(s.size()>=2)
break;
}
if(s.size()==1)
{
int t=0;
for(int i=1;i<=len+1;i++)
{
if(pos!=i)
{
t++;
printf("%c",str[i]);
if(t==len)
break;
}
}
return 0;
}
else if(s.size()==0)
{
printf(("NOT POSSIBLE\n"));
return 0;
}
else
{
printf("NOT UNIQUE\n");
return 0;
}
return 0;
}