2020牛客多校 2A.All with Pairs(kmp+hash)

题意:

在这里插入图片描述
定义两个字符串的权值为满足s的前缀=t的后缀的最大长度
问所有串两两组成的字符串对的权值平方和
答案对998244343取模

数据范围:n<=1e5,所有s串的总长度不超过1e6

解法:

题解:
2020牛客多校 2A.All with Pairs(kmp+hash)_第1张图片

参考了其他队伍的代码

code:

#include
using namespace std;
#define ll long long
const int maxm=1e6+5;
const int mod=998244353;
//
const int p1=131,p2=1331;
const int mod1=998244353,mod2=1e9+7;
ll b1[maxm],b2[maxm];
void H_init(){
    b1[0]=b2[0]=1;
    for(int i=1;i<maxm;i++){
        b1[i]=b1[i-1]*p1%mod1;
        b2[i]=b2[i-1]*p2%mod2;
    }
}
//
map<pair<int,int>,int>mark;
vector<ll>h1[maxm],h2[maxm];
vector<int>nt[maxm];
int cnt[maxm];
char s[maxm];
int n;
signed main(){
    H_init();
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%s",s+1);
        int len=strlen(s+1);
        //hash
        h1[i].resize(len+1);
        h2[i].resize(len+1);
        for(int j=1;j<=len;j++){
            h1[i][j]=(h1[i][j-1]*p1+(s[j]-'a'+1))%mod1;
            h2[i][j]=(h2[i][j-1]*p2+(s[j]-'a'+1))%mod2;
        }
        //next
        nt[i].resize(len+2);
        for(int k=1,j=nt[i][1]=0;k<=len;){
            if(!j||s[k]==s[j])k++,j++,nt[i][k]=j;
            else j=nt[i][j];
        }
        //mark
        for(int j=1;j<=len;j++){
            int a=((h1[i][len]-h1[i][j-1]*b1[len-j+1])%mod1+mod1)%mod1;
            int b=((h2[i][len]-h2[i][j-1]*b2[len-j+1])%mod2+mod2)%mod2;
            mark[{a,b}]++;
        }
    }
    int ans=0;
    for(int i=1;i<=n;i++){
        int len=h1[i].size()-1;
        for(int j=1;j<=len;j++){
            int a=h1[i][j],b=h2[i][j];
            cnt[j]=mark[{a,b}];
        }
        //我的nt数组是失配下标,而字符串从1开始的,所以公共前后缀长度=失配下标-1
        for(int j=2;j<=len+1;j++){//nt数组的有效范围是[2,len+1]
            if(nt[i][j]-1)cnt[nt[i][j]-1]-=cnt[j-1];
        }
        for(int j=1;j<=len;j++){
            ans=(ans+1LL*cnt[j]*j*j%mod)%mod;
        }
    }
    printf("%d\n",ans);
    return 0;
}

你可能感兴趣的:(2020牛客多校 2A.All with Pairs(kmp+hash))