两个字符串相似定义为:
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;
}