【Baltic2014】【BZOJ3916】friends

Description

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

第一行一个数N,表示U的长度.
第二行一个字符串U,保证U由大写字母组成
Output

输出一行,若S不存在,输出”NOT POSSIBLE”.若S不唯一,输出”NOT UNIQUE”.否则输出S.
Sample Input

Sample Input1:

7

ABXCABC

Sample Input2:

6

ABCDEF

Sample Input3:

9

ABABABABA
Sample Output

Sample Output1:

ABC

Sample Output2:

NOT POSSIBLE

Sample Output3:

NOT UNIQUE
HINT

对于100%的数据 2<=N<=2000001

Source

总串串长如果是偶数肯定不符合条件
要找的串串长必定n/2,所以如果知道删除哪个字符可以判断串左右端点
枚举删除的字符位置 hash一下判断
特殊判断删除最中间那个字符

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXN 2000010
#define LL long long
#define U unsigned
#define base 41
using namespace std;
int n,top;
U Pow[MAXN]={1},hash[MAXN];
char ch[MAXN];
int ans;
U last,L,R;
U Hash(int l,int r)
{
    U ret=hash[r]-hash[l-1]*Pow[r-l+1];return ret;
}
bool check(int x)
{
    if (!ans||last==L)  {ans=x;last=L;return 1;}
    else    return 0;
}
int main()
{
    scanf("%d%s",&n,ch+1);
    if (!(n&1)) return puts("NOT POSSIBLE"),0;
    for (int i=1;i<=n;i++)  Pow[i]=Pow[i-1]*base;
    for (int i=1;i<=n;i++)  hash[i]=hash[i-1]*base+(U)(ch[i]-'A'+1);
    int mid=n>>1;
    for (int i=1;i<=mid;i++)//delete  1 to n/2
    {
        L=Hash(1,i-1)*Pow[mid-i+1]+Hash(i+1,mid+1);
        R=Hash(mid+2,n);
        if (L==R)   if (!check(i))  return puts("NOT UNIQUE"),0;
    }
    L=hash[mid];R=hash[n]-hash[mid+1]*Pow[mid];//delete n/2+1
    if (L==R)   if (!check(mid+1))  return puts("NOT UNIQUE"),0;
    for (int i=mid+2;i<=n;i++)//delete n/2+2 to n
    {
        L=hash[mid];
        R=Hash(mid+1,i-1)*Pow[n-i]+Hash(i+1,n);
        if (L==R)   if (!check(i))  return puts("NOT UNIQUE"),0;
    }
    if (!ans)   return puts("NOT POSSIBLE"),0;
    for (int i=1;top<mid;i++)   if (ans!=i) putchar(ch[i]),++top;
}