(uva 11732) "strcmp()" Anyone? (trie+左儿子右兄弟表示法)

题目：http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=28438

题意：定义了一个strcmp()函数，如图：

然后给出n个字符串s（n<=4000,|s|<=1000），问字符串两两调用strcmp函数（总共调用函数n(n-1)/2次）需要执行多少次判断。比如”aaa”和”aaa“需要8次，”that“和”than”需要7次。
分析：可以发现，对于不同的两个字符串比较的次数=最长公共前缀的长度*2+1，对于相同的两个字符串比较的次数=(字符串长度+1)*2。很容易想到用trie，但是最多可能有4000*1000个节点,用普通的trie存的话可能内存不够，需要用左儿子-右兄弟表示法来建树。。。。然而UVA的内存给的很大我开的4000 000*80的int型数组都可以(试了一下10 000 000*80的好像不行),居然可以水过~这里给出两种代码。

普通trie代码：

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
const int maxn = 4e6+5;
const int kd = 80;
char s[10000];
struct trie
{
    int son[maxn][kd],end[maxn],L,root,dend[maxn];//end表示覆盖,dend表示结尾 
    long long ans;
    int newnode()
    {
        dend[L]=end[L]=0;
        fill(son[L],son[L]+kd,-1);
        return L++;
    }
    void Init()
    {
        L=ans=0;
        root=newnode();
    }
    void Insert(char str[])
    {
        ans+=end[root];
        end[root]++;
        int now=root,index,i;
        for(i=0;str[i];i++)
        {
            index=str[i]-'0';
            if(son[now][index]==-1)
                son[now][index]=newnode();
            now=son[now][index];
            ans+=(end[now]<<1);
            end[now]++;
        }
        ans+=dend[now];
        dend[now]++;
    }
}T;

int main()
{
    int N,ncase=1;
    while(scanf("%d",&N),N)
    {
        T.Init();
        while(N--)
        {
            scanf("%s",s);
            T.Insert(s);
        }
        printf("Case %d: %lld\n",ncase++,T.ans);
    } 
    return 0;
}

左孩子-右兄弟表示代码：

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn = 4e6+6;
struct trie
{
    int lson[maxn];//lson[i]表示第i个节点的第一个左孩子的位置 
    int rbro[maxn];//rbro[i]表示第i个节点的第一个右兄弟的位置
    char ch[maxn];//ch[i]表示第i个节点的字符 
    int cover[maxn];
    int end[maxn];
    int cnt,root;//节点数目
    long long ans;
    inline int newnode(const int &fa,const char &c) //父亲节点的下标,及新节点的字符 
    {//产生新节点,头插法,将每次产生的新节点作为父亲节点的第一个孩子 
        rbro[cnt]=lson[fa];
        lson[cnt]=-1;
        ch[cnt]=c;
        cover[cnt]=end[cnt]=0;
        lson[fa]=cnt;
        return cnt++;
    }
    void Init() //0为虚节点(根)
    {
        ans=0;
        cnt=0;
        root=0;
        rbro[cnt]=lson[cnt]=-1;
        end[cnt]=cover[cnt]=ch[cnt]=0;
        cnt++;
    }
    void Insert(char str[])
    {
        ans+=cover[root];
        cover[root]++;
        int now=root,i,j;
        for(i=0;str[i];i++)
        {
            for(j=lson[now];~j;j=rbro[j])
                if(str[i]==ch[j])
                    break;
            if(j==-1) //没找到 
                j=newnode(now,str[i]);
            now=j;
            ans+=(cover[now]<<1);
            ++cover[now];
        }
        ans+=end[now];
        ++end[now];
    }
}T; 
char s[100000];
int main()
{
    int N,ncase=1;
    while(scanf("%d",&N),N)
    {
        T.Init();
        while(N--)
        {
            scanf("%s",s);
            T.Insert(s);
        }
        printf("Case %d: %lld\n",ncase++,T.ans);
    } 
    return 0;
}