【51Nod1753】相似子串

两个字符串相似定义为：
1.两个字符串长度相等
2.两个字符串对应位置上有且仅有至多一个位置所对应的字符不相同
给定一个字符串，每次询问两个子串在给定的规则下是否相似。给定的规则指每次给出一些等价关系，如‘a’=’b’，‘b’=’c’等，注意这里的等价关系具有传递性，即若‘a’=’b’，‘b’=’c’，则‘a’=’c’。
Input
第一行一个字符串s(1<=|s|<=300000)
第二行一个整数T(1<=T<=300000)
对于每次询问：
第一行5个整数k，l1，r1，l2，r2，表示有k个等价规则，询问的是子串[l1,r1],l2,r2
接下来k行每行两个连续的字符表示这两个字符等价。
此题中所有的字符均为小写字母。
Output
T行，若相似则输出“YES”否则输出“NO”
Input示例
abac
3
1 1 2 3 4
bc
1 1 2 3 4
ac
1 1 2 2 3
ac
Output示例
YES
YES
NO

题解
sum[i][j]表示字符j，从1-i的哈希值之和。
等价字符用并查集合并，累加区间哈希值。
判断同一个集合哈希值是否相同或者相差一个哈希数。

代码

#include<bits/stdc++.h>
#define mod 6999983
#define inv 499122177
#define N 1000005
#define pa pair<int,int>
typedef __float128 F;
typedef long long ll;
using namespace std;
inline int read()
{
    int x=0;char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x;
}
int n,a1[30],a2[30],f[30],sum1[300005][27],fac1[300005],fac2[300005];
int b1[300005],b2[300005],sum2[300005][27];
char s[300005];
bool Hash1[7000005],Hash2[7000005];
int find(int i)
{
    if (f[i]!=i) f[i]=find(f[i]);
    return f[i];
}
inline void Union(int a,int b)
{
    int r1=find(a),r2=find(b);
    if (r1!=r2)
    {
        f[r1]=r2;
        a1[r2]+=a1[r1];a2[r2]+=a2[r1];
        a1[r2]-=(a1[r2]>=mod)*mod;
        a2[r2]-=(a2[r2]>=mod)*mod;
        b1[r2]+=b1[r1];b2[r2]+=b2[r1];
        b1[r2]-=(b1[r2]>=mod)*mod;
        b2[r2]-=(b2[r2]>=mod)*mod;
    }
}
inline void solve()
{
    int k=read(),l1=read(),r1=read(),l2=read(),r2=read();
    if (r1-l1!=r2-l2){puts("NO");return;}
    for (int i=1;i<=26;i++)
    {
        f[i]=i;
        a1[i]=sum1[r1][i]-sum1[l1-1][i]+mod;a1[i]-=(a1[i]>=mod)*mod;
        a2[i]=sum1[r2][i]-sum1[l2-1][i]+mod;a2[i]-=(a2[i]>=mod)*mod;
        b1[i]=sum2[r1][i]-sum2[l1-1][i]+mod;b1[i]-=(b1[i]>=mod)*mod;
        b2[i]=sum2[r2][i]-sum2[l2-1][i]+mod;b2[i]-=(b2[i]>=mod)*mod;
        if (l1<l2) a1[i]=(ll)a1[i]*fac1[l2-l1]%mod;
        else a2[i]=(ll)a2[i]*fac1[l1-l2]%mod;
        if (l1<l2) b1[i]=(ll)b1[i]*fac2[l2-l1]%mod;
        else b2[i]=(ll)b2[i]*fac2[l1-l2]%mod;
    }
    char ch[5];
    while (k--)
    {
        scanf("%s",ch);
        Union(ch[0]-'a'+1,ch[1]-'a'+1);
    }
    int flag=0,ans1=0,ans2=0;
    for (int i=1;i<=26;i++)
    {
        if (f[i]!=i) continue;
        if (a1[i]!=a2[i])
        {
            flag++;
            ans1=a2[i]-a1[i];
            ans1=(ans1+mod)%mod;
            ans2=b2[i]-b1[i];
            ans2=(ans2+mod)%mod;
            if (flag>2){puts("NO");return;}
            if (!((Hash1[ans1]&&Hash2[ans2])||(Hash1[mod-ans1]&&Hash2[mod-ans2]))){puts("NO");return;}
        }
    }
    puts("YES");
}
int main()
{
    scanf("%s",s+1);
    n=strlen(s+1);fac1[0]=fac2[0]=1;
    for (int i=1;i<=n;i++)
    {
        fac1[i]=(ll)fac1[i-1]*29%mod,Hash1[fac1[i]]=1;
        fac2[i]=(ll)fac2[i-1]*47%mod,Hash2[fac2[i]]=1;
    }
    for (int i=1;i<=n;i++)
        for (int j=1;j<=26;j++)
        {
            sum1[i][j]=sum1[i-1][j]+(s[i]==(char)('a'+j-1))*fac1[i];
            sum1[i][j]-=(sum1[i][j]>=mod)*mod;
            sum2[i][j]=sum2[i-1][j]+(s[i]==(char)('a'+j-1))*fac2[i];
            sum2[i][j]-=(sum2[i][j]>=mod)*mod;
        }
    int Case=read();
    while (Case--) solve();
    return 0;
}