friends 一道要想清楚的 hash

2024-02-04 09:58

文章标签 一道 hash 清楚 friends

本文主要是介绍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;
}