[SCOI2016] 背单词 （Trie树）

$pdf\space solution$      link

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<vector>
#define ll long long
using namespace std;
inline int read(){
    int f=,ans=;char c=getchar();
    while(c<''||c>''){if(c=='-')f=-;c=getchar();}
    while(c>=''&&c<=''){ans=ans*+c-'';c=getchar();}
    return f*ans;
}
const int MAXN=;
vector<int> ve[MAXN];
struct node{
    int u,v,nex;
}x[MAXN<<];
int head[MAXN],cnt;
void add(int u,int v){
    x[cnt].u=u,x[cnt].v=v,x[cnt].nex=head[u],head[u]=cnt++;
}
int ch[MAXN][],bo[MAXN],siz[MAXN],tot=;
struct Trie{
    void init(){memset(ch,-,sizeof(ch));}
    void insert(char *s){
        int len=strlen(s);
        int u=;
        for(int i=len-;i>=;i--){
            int c=s[i]-'a';
            if(ch[u][c]==-) {
                ch[u][c]=++tot;
                add(u,ch[u][c]);
            }
            u=ch[u][c];
        }
        bo[u]++;
        return;
    }
}trie;
void build(int u,int spe){
    if(bo[u]) ve[spe].push_back(u),spe=u;
    for(int i=head[u];i!=-;i=x[i].nex) build(x[i].v,spe);
}
int n;
void init_siz(int u){
    siz[u]=;
    int Siz=ve[u].size();
    for(int i=;i<Siz;i++){
        init_siz(ve[u][i]);
        siz[u]+=siz[ve[u][i]];
    }return;
}ll Tot,ans;
bool cmp(int xx,int yy){
    return siz[xx]<siz[yy];
}
void query(int u,int time){
//    printf("u:%d time:%d\n",u,time);
    int p=++Tot;
    ans+=(ll)p-time;
    sort(ve[u].begin(),ve[u].end(),cmp);
    int siz=ve[u].size();
    for(int i=;i<siz;i++) query(ve[u][i],p);
    return;
}
char str[MAXN],str1[MAXN];
signed main(){
//    freopen("word.in","r",stdin);
    memset(head,-,sizeof(head));
    trie.init();
    n=read();
    for(int i=;i<=n;i++){
        scanf("%s",str);
        trie.insert(str);
    }
    build(,);
    init_siz();
    query(,);
    printf("%lld\n",ans);
}/*
2
aa
aaa
*/