uva 11732

字典树；

不过数据量很大；4000*1000；

我用的是传统的方法建树，虽然优化了一下，边插入边统计，不过还是T了；

这是我的代码：

#include<cstdio>
#include<cstring>
#define maxn 4000009
using namespace std;
int nonocount;
long long ans;
struct node
{
    int num;
    node *a[62];
} no[maxn];

node* newnode()
{
    node *p=no+nonocount++;
    p->num=0;
    memset(p->a,NULL,sizeof p->a);
    return p;
}

void insert(node *root,char *s)
{
    int l=strlen(s);
    for(int i=0; i<l; i++)
    {
        int x;
        if(s[i]>='0'&&s[i]<='9')x=s[i]-'0';
        else if(s[i]>='a'&&s[i]<='z')x=s[i]-'a'+10;
        else if(s[i]>='A'&&s[i]<='Z')x=s[i]-'A'+36;
        if(root->a[x]==NULL)
            root->a[x]=newnode();
        ans+=(root->num)-(root->a[x]->num);
        root->num++;
        root=root->a[x];
    }
}
char s[1005];
int main()
{
    int n,ca=1;
    while(scanf("%d",&n)&&n)
    {
        nonocount=0;
        node *p=newnode();
        ans=0;
        for(int i=0; i<n; i++)
        {
            scanf("%s",s);
            insert(no,s);
        }
        printf("Case %d: %lld
",ca++,ans);
    }
    return 0;
}

网上的大神的方法是采用左儿子右兄弟的方法建树；

这样节省了很多的时间，受教了！

大神的代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxnode = 4000*1000;
struct trie
{
    int left[maxnode]; // 左儿子
    int right[maxnode]; // 右兄弟
    int val[maxnode] ; //该节点上有多少串
    char ch[maxnode]; // 节点表示的字符
    int cnt ;
    long long int ans ;
    void init()
    {
        ch[0] = 0, left[0] = right[0] = 0;
        val[0] = 0;
        cnt = 1;
        ans = 0;
    }
    void insert(const char *s)
    {
        int len = strlen(s) , u = 0,j;
        for(int i = 0; i <= len ; ++i )
        {
            for(j = left[u]; j ; j = right[j])if(ch[j] == s[i]) break; //找到这个节点
            if(j==0)
            {
                j = cnt++;
                ch[j] = s[i];
                right[j] = left[u];
                left[j] = 0;
                left[u] = j;
                val[j] = 0;
            }
            ans += (val[u] - val[j])*(i << 1|1);
            if(i == len)
            {
                ans += val[j]*((i+1)<<1);    //特判结尾 Trie中不会进入结尾字符,需特判统计
                val[j]++;
            }
            val[u]++;
            u = j;
        }
    }
} Tri;
char tmp[2000];
int main()
{
    int n,kase = 1 ;
    while(scanf("%d",&n)==1 && n)
    {
        Tri.init();
        for(int k=0; k<n; ++k)
        {
            scanf("%s",tmp);
            Tri.insert(tmp);
        }
        printf("Case %d: %lld
", kase++, Tri.ans);
    }
    return 0;
}