uva 11732 strcmp() Anyone? 字典树

题意：

每次用strcmp()比较两个字符串(s和t)的时候，每个字符需要s[i]==t[i]和s[i]=='\0'共两次比较操作（不相等时一次操作），问n()个字符串两两比较时需要多少次比较操作。（答案保证在有符号64位整数内，n<=4000,字符串长不超过1000）

题解：

用字典树保存，由于插入字典树就需要4000*1000次操作了，所以基本上就要边插入边计算，即使是这样还是卡着时间过的，不知道有没什么改进的地方。

通过了解strcmp()的具体比较过程，我们发现两个字符串比较的时候需要的比较操作数为：相同前缀数*2+1（但如果两个字符串相同的话还要+1）,所以用val[i]数组保存i状态下相同前缀的个数，那么每次插入就可以计算得到ans+=val[i]*2,还要用num[i]数组记录i状态下相同字符串的个数，ans+=num[i]，插入完成后，还要加上不相同比较的ans+=n*(n-1)/2（将相同字符串也算进去了）。

水过的代码，只看思路就好，不知道哪里能改进时间的。。

代码：

#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <queue>
#include <map>
#include <vector>
using namespace std;

#define LL long long
const int maxn=4001*1001;
LL ans;
struct Trie{
    int ch[maxn][62];
    int val[maxn];
    int num[maxn];
    int sz;
    Trie(){sz=1;memset(ch[0],0,sizeof(ch[0]));}
    int idx(char c)
    {
        if(c<='9'&&c>='0')return c-'0';
        if(c<='z'&&c>='a')return c-'a'+10;
        if(c<='Z'&&c>='A')return c-'A'+36;
    }

    void insert(char *s,int v)
    {
        int i,u=0,n=strlen(s);
        for(i=0;i<n;i++)
        {
            int c=idx(s[i]);
            if(!ch[u][c])
            {
                memset(ch[sz],0,sizeof(ch[sz]));
                val[sz]=0;
                num[sz]=0;
                ch[u][c]=sz++;
            }
            u=ch[u][c];
            ans+=(val[u]<<1);
            //printf("%d\n",val[u]);
            val[u]++;
        }
        ans+=num[u];
        num[u]++;
    }
}e;
char a[1010];
int main()
{
    int tt=0,n;
    while(scanf("%d",&n)!=EOF)
    {
        if(n==0)break;
        int i,j,k,u,c;
        ans=0;
        memset(e.ch[0],0,sizeof(e.ch[0]));
        e.sz=1;
        for(i=0;i<n;i++)
        {
            scanf("%s",a);
            e.insert(a,1);
        }
        printf("Case %d: %lld\n",++tt,ans+n*(n-1)/2);
    }
    return 0;
}

/*
2
than
that
*/