UVA 11732 strcmp() Anyone?（trie）

题目大意：给你那个 strcmp 函数的内部，现在给你n个字符串，要两两比较，问你strcmp里判断总共用了几次？

思路：trie，又是Submission error，算了，先存下吧。。
代码如下（不知道对不对。。 = =）：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long lld;

const int MAXN = 4001;

const int MAX_NODE = MAXN*1001;

const int SIGMA_SIZE = 27;

int ch[MAX_NODE][SIGMA_SIZE];

int val[MAX_NODE];

struct Trie
{
int sz;

Trie()
{
sz = 1;
val[0] = 0;
memset(ch[0],0,sizeof(ch[0]));
}

int idx(char c)
{
if(c == '\0')
return 0;
else return c - 'a' + 1;
}

void insert(char *s)
{
int u = 0;
int len = strlen(s);
for(int i = 0;i<=len;i++)
{
int c = idx(s[i]);
if(!ch[u][c])
{
memset(ch[sz],0,sizeof(ch[sz]));
val[sz] = 0;
ch[u][c] = sz++;
}
val[u]++;
u = ch[u][c];
}
}

int count(char *s)
{
int u = 0;
int len = strlen(s);
int cnt = 0;
for(int i = 0;i<=len;i++)
{
cnt += val[u];
int c = idx(s[i]);
if(!ch[u][c])
return cnt;
cnt += val[u];
u = ch[u][c];
}
return cnt;
}
};

char str[MAXN];

int main()
{
int cas = 0;
int n;
while(~scanf("%d",&n)&&n)
{
Trie tr;
lld ans = 0;
for(int i = 1;i<=n;i++)
{
scanf("%s",str);
if(i > 1) ans += tr.count(str);
//printf("ans = %d\n",ans);
tr.insert(str);
}
printf("Case %d: %lld\n",++cas,ans);
}
return 0;
}