Prefix

Prefix

南昌邀请赛的题，字典树

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

}