UVa OJ 11732

#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
const int maxnode=4000*1000+10;
const int sigma_size=26;
const int maxl=1000+10;
int n;
char word[maxl];
struct Trie{
    int head[maxnode];
    int next[maxnode];
    char ch[maxnode];
    int tot[maxnode];
    int sz;
    long long ans;
    void clear(){sz=1;tot[0]=head[0]=next[0]=0;}
    void insert(const char*s){
        int u=0,v,n=strlen(s);
        tot[0]++;
        for(int i=0;i<=n;i++){
           bool found=false;
           for(v=head[u];v!=0;v=next[v])
              if(ch[v]==s[i]){
                  found=true;
                  break;
              }
              if(!found){
                  v=sz++;
                  tot[v]=0;
                  ch[v]=s[i];
                  next[v]=head[u];
                  head[u]=v;
                  head[v]=0;
              }
              u=v;
              tot[u]++;
        }
        return;
    }
    void dfs(int depth,int u){
       if(head[u]==0)
          ans+=tot[u]*(tot[u]-1)*depth;
       else{
          int sum=0;
          for(int v=head[u];v!=0;v=next[v])
              sum+=tot[v]*(tot[u]-tot[v]);
          ans+=sum/2*(2*depth+1);
          for(int v=head[u];v!=0;v=next[v])
              dfs(depth+1,v);
       }
       return;
    }
    long long count(){
       ans=0;
       dfs(0,0);
       return ans;
    }
};
Trie trie;
int main(){
    int kase=1;
    while(scanf("%d",&n)==1&&n){
        trie.clear();
        for(int i=0;i<n;i++){
            scanf("%s",word);
            trie.insert(word);
        }
        printf("Case %d: %lld\n",kase++,trie.count());
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值