首页 > 其他 > 详细

Prefix

时间:2019-07-31 11:00:45      阅读:101      评论:0      收藏:0      [点我收藏+]

Prefix

南昌邀请赛的题,字典树

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll A[26];
string s[100005];
ll trie[200008][26];
ll v[200005];
ll tot=1;
ll cnt[200005];
ll B[100005];
ll n,mod;
void insert(string s,int x)
{
    int p=1;
    int len=s.length();
    ll t=1;
    for(int i=0;i<len;i++){
        t=(t*A[s[i]-a])%mod;
        if(trie[p][s[i]-a]==0){
            trie[p][s[i]-a]=++tot;
        }
        p=trie[p][s[i]-a];
        cnt[p]++;
        v[p]=t;
    }
    B[x]=t;
}
ll ans=0;
void search(string s,int x)
{
    int p=1;
    int len=s.length();
    for(int i=0;i<len;i++){
        p=trie[p][s[i]-a];
        if(v[p]>B[x]){
            ans+=cnt[p];
        }
    }
}
int main()
{
    
    scanf("%lld%lld",&n,&mod);
    for(int i=0;i<26;i++){
        scanf("%lld",&A[i]);
    }
    for(int i=0;i<n;i++){
        cin>>s[i];
        insert(s[i],i);
    }
    for(int i=0;i<n;i++){
        search(s[i],i);
        cout<<ans<< ;
        ans=0;
    }
    
    
}

 

Prefix

原文:https://www.cnblogs.com/liulex/p/11274041.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!