UVA 11732 mark这个字典树模版

点击打开链接

题意：询问所有的字符串两两比较的次数和

思路：边插入边计算，节点保存的信息是有多少字符串到这个节点是相同的，每次的结果就是个数*2，然后匹配不成功的时候要+1，若相同则加2

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3fll;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=4010*1010;
ll ans;
struct Trie{
    int ch[maxn][62];
    int val[maxn],num[maxn],sz;
    Trie(){sz=1;memset(ch[0],0,sizeof(ch[0]));}
    int change(char ch){
        if(ch>='0'&&ch<='9') return ch-'0';
        if(ch>='a'&&ch<='z') return ch-'a'+10;
        if(ch>='A'&&ch<='Z') return ch-'A'+36;
    }
    void T_insert(char *s,int v){
        int u=0,n=strlen(s);
        for(int i=0;i<n;i++){
            int c=change(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);
            val[u]++;
        }
        ans+=num[u];num[u]++;
    }
}trie;
char str[1010];
int main(){
    int n,cas=1;
    while(scanf("%d",&n)!=-1){
        if(n==0) break;
        ans=0;trie.sz=1;
        memset(trie.ch[0],0,sizeof(trie.ch[0]));
        for(int i=0;i<n;i++) scanf("%s",str),trie.T_insert(str,1);
        printf("Case %d: %lld\n",cas++,ans+n*(n-1)/2);
    }
    return 0;
}