UVA 11732 链表+字典树

因为字符集比较大，所以就不能用简单字典树，在字典树里面，用链表进行存储。这个倒是不难，练了下手

统计的时候还是有点难搞，因为要算所有的两两比较的次数之和，对分叉处进行计算，注意细节

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int N = *+;
struct Trie
{
    int head[N];
    int next[N];
    char ch[N];
    int tot[N];
    int sz;
    long long ans;
    void clear()
    {
        sz=;
        head[]=next[]=tot[]=;
        ans=;
    }
    void insert(const char* s)
    {
        int u=,v,len;
        len=strlen(s);
        tot[]++;
        for (int i=; i<=len; i++)
        {
            bool found=false;
            for (v=head[u]; v!=; v=next[v])
            {
                if (ch[v]==s[i])
                {
                    found=true;
                    break;
                }
            }
            if (!found)
            {
                v=sz++;
                ch[v]=s[i];
                tot[v]=;
                next[v]=head[u];
                head[u]=v;
                head[v]=;
            }
            u=v;
            tot[u]++;
        }
    }

    void dfs(int depth, int u) //计算总数
    {
        if(head[u] == )
            ans += tot[u] * (tot[u] - ) * depth;
        else
        {
            int sum = ;
            for(int v = head[u]; v != ; v = next[v])
                sum += tot[v] * (tot[u] - tot[v]);
            ans += sum /  * ( * depth + );
            for(int v = head[u]; v != ; v = next[v])
                dfs(depth+, v);
        }
    }

    long long count()
    {
        ans = ;
        dfs(, );
        return ans;
    }
} T;
char str[];
int main()
{
    int n;

    int kase=;
    while (scanf("%d",&n) && n)
    {
        T.clear();
        for (int i=; i<=n; i++)
        {
            scanf("%s",str);
            T.insert(str);
        }
        T.count();
        printf("Case %d: %lld\n",++kase,T.ans);
    }
    return ;
}