UVA11732 左儿子右兄弟 Trie

tot[i]记录每个以i为根的树的叶子节点的数目

注意插入的时候要插入‘\0’   为了分辨 前缀和相同的串

对于叶子节点  ans+=tot[u]*(tot[u]-1)/2*2*depth;

即相同的串两两一对的对数 *（2*深度）

对于非叶子节点 统计以其为根节点的串 两两一对的对数

           for(int v=head[u];v;v=next[v]){
                sum+=tot[v]*(tot[u]-tot[v]);//该节点和其他兄弟节点的叶子节点数（串数）
            }
            ans+=sum/2*(2*depth+1);///重复计算 除2

#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn=4010*1000+10;
typedef long long ll;
struct Trie
{
    int head[maxn];
    int next[maxn];
    int tot[maxn];
    char ch[maxn];
    int sz;
    ll ans;
    void clear() {
        head[0]=next[0]=tot[0]=0;
        ans=0;
        sz=1;
    }
    void insert(char *s) {
        int l=strlen(s),v,u=0;
        tot[0]++;
        for(int i = 0; i <= l; ++i) {
            bool found=0;
            for(v=head[u];v!=0;v=next[v]) {
                if(ch[v]==s[i]) {
                    found=1;
                    break;
                }
            }
            if(!found) {
                v=sz++;
                ch[v]=s[i];
                next[v]=head[u];
                head[u]=v;
                head[v]=0;
                tot[v]=0;
            }
            u=v;
            tot[u]++;
        }
    }
    void dfs(int depth,int u) {
        if(head[u]==0) {
            ans+=tot[u]*(tot[u]-1)/2*2*depth;
        }
        else {
            ll sum=0;
            for(int v=head[u];v;v=next[v]){
                sum+=tot[v]*(tot[u]-tot[v]);
            }
            ans+=sum/2*(2*depth+1);
            for(int v=head[u];v;v=next[v]) {
                dfs(depth+1,v);
            }
        }
    }
    ll count()
    {
        dfs(0,0);
        return ans;
    }
};
Trie trie;
int n;
char c[4010];
int main()
{
    int Kase=0;
    while(~scanf("%d",&n)&&n) {
        trie.clear();
        for(int i = 0; i < n; ++i) {
            scanf("%s",c);
            trie.insert(c);
        }
        printf("Case %d: %lld\n",++Kase,trie.count());
    }
    return 0;
}