字符串哈希-friends-HYSBZ - 3916

字符串哈希-friends-HYSBZ - 3916

题意：

$假设有字符串s_0,先经过复制得到2s_0,再向2s_0中插入一个字符x，得到字符串s。$

$现在给定s，判断能否复原到s_0。$

$$\begin{gathered} 若能，输出s_0； \\ 若有多种方案，输出：“NOTUNIQUE”; \\ 若不能，输出：“NOTPOSSIBLE”。 \end{gathered}$$

样例:
Sample Input:
$$\begin{gathered} SampleInput1: \\ 7 \\ ABXCABC \\ SampleInput2: \\ 6 \\ ABCDEF \\ SampleInput3: \\ 9 \\ ABABABABA \end{gathered}$$

Sample Output:

$$\begin{gathered} SampleOutput1: \\ ABC \\ SampleOutput2: \\ NOTPOSSIBLE \\ SampleOutput3: \\ NOTUNIQUE \end{gathered}$$

数据范围
$s的长度N\in [2,2000001]。$

题解:

$$\begin{gathered} ①、若N为偶数，必然无法复原。 \\ ②、先删除一个字符，枚举删除的位置，再判断中点两侧的子串是否相同。 \\ 需要注意的是，若用string类函数，结果会RE，所以只能考虑哈希来做。 \end{gathered}$$

具体落实：

$$\begin{gathered} ①、若s长度为偶数，直接输出。 \\ ②、预处理整个s的哈希值。 \\ ③、枚举删除的位置，重点在于求删除后，计算中点两侧的子串的哈希值s_1、s_2。 \\ 设s的长度为n，则分别分成两种情况： \\ (1)、删除的位置i<=\lfloor \frac{n}{2}\rfloor ,也就是左侧的子串需要分成两段来计算哈希值。 \\ 左侧的哈希值=区间[1,i-1]的哈希值+区间[i+1,\lfloor \frac{n}{2}\rfloor +1]的哈希值。 \\ 抽象地,s1=h[1,i-1]+h[i+1,\lfloor \frac{n}{2}\rfloor +1]。 \\ 还要注意，h[1,i-1]需要向高位移动\lfloor \frac{n}{2}\rfloor -i+1位，即在i右侧的字符的数量。 \end{gathered}$$

$$\begin{gathered} 否则的话，左侧哈希值为区间[1,\lfloor \frac{n}{2}\rfloor ]的哈希值，即s_1=h[1,\lfloor \frac{n}{2}\rfloor ]。 \\ (2)、删除位置i<=\lfloor \frac{n}{2}\rfloor +1,这时右侧区间的长度>=\lfloor \frac{n}{2}\rfloor ,故右侧子串的哈希值可直接计算。 \\ {s}_2=h[n-\lfloor \frac{n}{2}\rfloor +1,n]，这里左端点通过右端点来确定比较容易。 \\ 否则,右侧子串的哈希值需要分成两端计算。 \\ 左侧子串哈希值=区间[\lfloor \frac{n}{2}\rfloor +1,i-1]的哈希值+区间[i+1,n]的哈希值。 \\ 即s2=h[\lfloor \frac{n}{2}\rfloor +1,i-1]+[i+1,n]。 \\ 同样的，h[\lfloor \frac{n}{2}\rfloor +1,i-1]需要向高位移动n-(i+1)+1位，即i右侧的字符的数量。 \end{gathered}$$

$向高位移动i位就是乘基数base的i次方，即bas{e}^i。$

$$\begin{gathered} ④、若两侧哈希值相等，需要记录第一次哈希值以及删除的位置i。 \\ (一开始以为只要符合条件的位置找到2个，就判断方案不唯一，事实上是不符合题意的， \\ 因为删除不同位置的字符，两侧的哈希值可能都为同一值。)因此需要记录哈希值， \\ 若再次相等时，哈希值与第一次的哈希值不同，说明方案不唯一。 \end{gathered}$$

---

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<map>
#include<set>
#define ull unsigned long long
using namespace std;
const int base=131;
const int N=2e6+10;
int n,pos;
ull p[N],h[N],ans;
bool flag;
char s[N];

ull get(int l ,int r)
{
    return h[r]-h[l-1]*p[r-l+1];
}

int main()
{
        scanf("%d",&n);
        scanf("%s",s+1);
        if(n%2==0) {puts("NOT POSSIBLE");return 0;}

        p[0]=1;
        for(int i=1;i<=n;i++) p[i]=p[i-1]*base;

        for(int i=1;i<=n;i++) h[i]=h[i-1]*base+s[i]-'A';

        for(int i=1;i<=n;i++)
        {
            ull s1,s2;
            if(i<=n/2) s1=get(i+1,n/2+1)+h[i-1]*p[n/2-i+1];
            else s1 = h[n/2];
            if(i<=n/2+1) s2=get(n-n/2+1,n);
            else s2=get(n/2+1,i-1)*p[n - i]+get(i+1,n);
            if(s1==s2)
            {
                if(flag && ans != s1)
                {
                    printf("NOT UNIQUE\n");
                    return 0;
                }
                flag=true,ans=s1,pos=i;
            }
        }
        
        if(flag)
        {
            for(int i=1,j=0;j<n/2;i++)
                if(i!=pos)
                    printf("%c",s[i]),j++;
            printf("\n");
        }
        else puts("NOT POSSIBLE");

    return 0;
}