题意:
给出n个字符串, 计算两两比较的次数. 每次比较都需要比较(str1[i] == str2[i])和 (str1[i]== ‘\0’各一次).
分析:
边插入边统计,这样就刚好是两两比较一次的结果。如果先建树再深搜,那么就多比较了一次,/2就可以。
#include<cstdio>
#include<cstring>
typedef long long ll;
const int N=4000*1000+10;
ll ans;
struct Trie
{
int son[N],bro[N],val[N];
char ch[N];
int sz;
void clear(){sz=1;ch[0]=val[0]=bro[0]=son[0]=0;}
void insert(char *s)
{
int u=0,v,n=strlen(s);
for(int i=0;i<=n;i++){
for(v=son[u];v;v=bro[v]){
if(ch[v]==s[i])break;
}
if(!v){
v=sz++;
ch[v]=s[i];
bro[v]=son[u];
son[v]=0;
son[u]=v;
val[v]=0;
}
ans+=(val[u]-val[v])*(2*i+1);
if(i==n){
ans+=val[v]*(2*i+2);
val[v]++;
}
val[u]++;
u=v;
}
}
};
Trie trie;
char s[1009];
int main()
{
int n,cas=0;
//freopen("f.txt","r",stdin);
while(~scanf("%d",&n)&&n){
trie.clear();
ans=0;
for(int i=0;i<n;i++){
scanf("%s",s);
trie.insert(s);
}
printf("Case %d: %lld\n",++cas,ans);
}
return 0;
}