UVAlive - 5792 Diccionário Portuñol (Trie)

转自：cxlove
题意：有两个字符串集合，从第一个集合中取某个串的非空前缀，从第二个集合中取某个串的非空后缀，拼接成一个串，问有多少个不同的新串。
思路：首先把第一个集合中的串加入到Trie中，然后就可以枚举前缀了，然后就是统计后缀，麻烦的是可能出现重复的。

当我们按长度枚举前缀的时候，可以考虑如果后缀的前缀出现在前缀的后继中，那便不用考虑，后面总会统计到。

这样的话，其实只要比较第一个字母。

如果当前结点P表示一个前缀串，那么如果p -> next[i]为空，说明前缀中不含字母i为后继，那么我们就可以统计以i开头的后缀了。否则总能在p -> next[i]为前缀中统计到。

有一点需要注意的是，如果 p -> next[i]便是我们最终的串，即以p 为前缀时，后缀只有一个字母的话，那就不能把这一个字母拉入到前缀中，因为后缀是非空的。所以再处理下end[i]，表示是否存在以单个字母i为后缀的。

至于统计以i开头的不重复后缀的话，可以反向加入到Trie中统计。

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<algorithm>
using namespace std;
typedef long long LL;
const int maxn=100010;
const int maxm=1010;
const int SIGMA_SIZE=26;
int P,S;
char a[maxm][maxm],b[maxm][maxm];
int tail[maxn];
int cnt[30];
LL ans;
struct Trie
{
    int ch[maxn][SIGMA_SIZE];
    int val[maxn];
    int sz;
    void clear(){memset(ch[0],0,sizeof(ch[0]));sz=1;}
    int idx(char x){return x-'a';}
    void insert(char *s)
    {
        int u=0;
        for(int i=0;s[i];i++)
        {
            int c=idx(s[i]);
            if(!ch[u][c])
            {
                memset(ch[sz],0,sizeof(ch[sz]));
                val[sz]=0;
                ch[u][c]=sz++;
            }
            u=ch[u][c];
        }
        val[u]=1;
    }
    void cal(int u)
    {
        for(int i=0;i<SIGMA_SIZE;i++)
        {
            if(ch[u][i])
            {
                cnt[i]++;
                cal(ch[u][i]);
            }
        }
    }
    void solve(int u)
    {
        for(int i=0;i<SIGMA_SIZE;i++)
        {
            if(!ch[u][i])
            {
                if(u)
                ans+=cnt[i];
            }
            else
            {
                if(u&&tail[i])ans++;
                solve(ch[u][i]);
            }
        }
    }
}prefix,suffix;
int main()
{
    while(scanf("%d%d",&P,&S)!=EOF,P||S)
    {
        prefix.clear();
        suffix.clear();
        memset(cnt,0,sizeof(cnt));
        memset(tail,0,sizeof(tail));
        for(int i=0;i<P;i++)
        {
            scanf("%s",a[i]);
            prefix.insert(a[i]);
        }
        for(int i=0;i<S;i++)
        {
            scanf("%s",b[i]);
            reverse(b[i],b[i]+strlen(b[i]));
            tail[b[i][0]-'a']=1;
            suffix.insert(b[i]);
        }
        ans=0;
        suffix.cal(0);
        prefix.solve(0);
        cout<<ans<<endl;
    }
    return 0;
}