Codeforces 985F-Isomorphic Strings

Codeforces 985F-Isomorphic Strings

题意

给定一个字符串\(s\)，长为\(n\)。\(m\)组询问，每次询问给出\(x\)，\(y\)，\(len\)，问是否存在一一映射，使得\(s\)中起点为\(x\)长度为\(len\)的子串能映射成起点为\(y\)，长度为\(len\)的子串。

题解

是否存在映射取决于字母的排列情况是否一致。我们可以以26个字母分别为关键字将整个字符串\(s\)表示成一个0，1串（如字符串“aabcc”，以a为关键字“11000”，b为关键字“00100”，c为关键字“00011”，其余字母为关键字均为“00000”），那么，如果两个字串相等，他们26个0，1串排好序后就应该一致。我们可以用hash来比较0，1串是否一致。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N=2e5+10;
const int base=131;
const ll mod=1e9+7;
char s[N];
ll h[30][N];
ll sum[N];
ll a[30],b[30];
int len,n,t;

void Hash(){
    for(int i=0;i<26;i++){
        for(int j=1;j<=len;j++){
            ll x;
            if(s[j]=='a'+i) x=1;
            else x=0;
            h[i][j]=(h[i][j-1]*base%mod+x)%mod;
        }
    }
}

void init(){
    sum[0]=1;
    for(int i=1;i<=len;i++){
        sum[i]=(sum[i-1]*base)%mod;
    } 
}

bool solve(int l,int r,int d){
    for(int i=0;i<26;i++){
        a[i]=(h[i][l+d-1]-h[i][l-1]*sum[d]%mod+mod)%mod;
        b[i]=(h[i][r+d-1]-h[i][r-1]*sum[d]%mod+mod)%mod;
    }
    sort(a,a+26);sort(b,b+26);
    for(int i=0;i<26;i++){
        if(a[i]!=b[i]) return 0;
    }
    return 1;
}


int main(){
    scanf("%d%d",&n,&t);
    scanf("%s",s+1);
    len=strlen(s+1);
    Hash();
    init();
    for(int i=1;i<=t;i++){
        int l,r,d;
        scanf("%d%d%d",&l,&r,&d);
        if(solve(l,r,d)) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}