bzoj3916: [Baltic2014]friends

链接

　　http://www.lydsy.com/JudgeOnline/problem.php?id=3916

题解

　　枚举删除位置，哈希判断
　　我查了一上午的错，就因为题没读好。长得一样的串，算一个。

代码

//哈希
#include <cstdio>
#include <algorithm>
#define maxn 3000000
#define mod 1000000007
#define base 4894651ll
#define ll long long
using namespace std;
int N, h[maxn], inv[maxn], b[maxn];
char s[maxn];
void init()
{
    int i;
    scanf("%d%s",&N,s+1);
    for(b[0]=i=1;i<=N;i++)b[i]=(ll)b[i-1]*base%mod;
    for(inv[0]=1,i=1;i<=N;i++)inv[i]=(inv[i-1]*612582597ll)%mod;
    for(i=1;i<=N;i++)h[i]=((ll)(s[i]-'A')*b[i-1]+h[i-1])%mod;
}
inline int geth(int l, int r)
{
    if(l>r)return 0;
    return ((ll)h[r]-h[l-1])*inv[l-1]%mod;
}
inline int check(int pos)
{
    int h1, h2;
    if(pos<=(N>>1))
    {
        h1=(geth(1,pos-1)+(ll)geth(pos+1,(N>>1)+1)*b[pos-1]%mod)%mod;
        h2=geth((N>>1)+2,N);
    }
    else
    {
        h1=geth(1,N>>1);
        h2=(geth((N>>1)+1,pos-1)+(ll)geth(pos+1,N)*b[pos-(N>>1)-1]%mod)%mod;
    }
    return (h1+mod)%mod == (h2+mod)%mod ? (h1+mod)%mod : -1;
}
void show(int pos)
{
    int i;
    for(i=pos;i<N;i++)s[i]=s[i+1];
    for(i=1;i<=(N>>1);i++)putchar(s[i]);
}
int main()
{
    int i, pos, cnt=0, H=-1, t;
    init();
    if((~N&1) or N<3){printf("NOT POSSIBLE");return 0;}
    for(i=1;i<=N;i++)
    {
        t=check(i);
        if(t^-1 and t^H)
        {
            pos=i;cnt++;H=t;
            if(cnt>1)break;
        }
    }
    if(cnt==0){printf("NOT POSSIBLE");return 0;}
    if(cnt>1){printf("NOT UNIQUE");return 0;}
    show(pos);
    return 0;
}