UVA11732(Trie树)

鸣谢https://blog.csdn.net/Baoli1008/article/details/4441936，基本都是抄的代码

#pragma GCC optimize(2)
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int MAXN = 4e6+;
int idx(char c)
{
    return c-'';
}
struct node{
    ll num;
    char ch;
    node(){
        num=;
    };
};
vector<int>G[MAXN];
node N[MAXN];
int cn;
void init()
{
    for(int i=;i<cn;i++)
    {
        G[i].clear();
        N[i].num=;
    }
    cn=;
}
void input(char s[])
{
    int cur=;
    int len=strlen(s);
    N[cur].num++;
    for(int i=;i<=len;i++)
    {
        int nxt=;
        int m=G[cur].size();
        for(int j=;j<m;j++)
        {
            int tmp=G[cur][j];
            if(N[tmp].ch==s[i])
            {
                nxt=tmp;
                break;
            }
        }
            if(!nxt)
            {
                N[cn].ch=s[i];
                G[cur].push_back(cn);
                nxt=cn;
                cn++;
            }
            N[nxt].num++;
            cur=nxt;

    }
}
ll solve(int n)
{
    ll res=;
    ll tmp=;
    if(n)
        res+=N[n].num*(N[n].num-);
    int m=G[n].size();
    for(int i=;i<m;i++)
    {
        int t=G[n][i];
        res+=solve(t);
        tmp+=(N[n].num-N[t].num)*N[t].num;
    }
    res+=tmp/;
    return res;
}
int main()
{
    int T;
    int kase=;
    while(cin>>T&&T)
    {
        init();
        for(int i=;i<T;i++)
        {
            char s[];
            scanf("%s",s);
            input(s);
        }
        printf("Case %d: %lld\n",++kase,solve());
    }

  return ;
}