UVAlive 5792 Diccionário Portuñol (Trie)

转载请注明出处，谢谢http://blog.csdn.net/ACM_cxlove?viewmode=contents    by---cxlove

题目：有两个字符串集合，从第一个集合中取某个串的非空前缀，从第二个集合中取某个串的非空后缀，拼接成一个串，问有多少个不同的新串。

https://icpcarchive.ecs.baylor.edu/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=3803

大概是N年前训练的题。。。然后不会做，就没管了

最近波波好同学问了御坂姐姐，然后就会做了，其实很水。。。

为何这么弱。。。。

首先把第一个集合中的串加入到Trie中，然后就可以枚举前缀了，然后就是统计后缀，麻烦的是可能出现重复的。

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

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

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

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

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

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <cstring>
using namespace std;
typedef long long LL;
const int M = 100005;
const int N = 1005;
int n , m , cnt[26] , end[26];
char a[N][N] , b[N][N];
LL ans = 0;
struct Trie {
    int tot;
    struct trie {
        trie *next[26];
    }s[M] , *root;
    trie *newnode () {
        trie *p = &s[tot ++];
        memset (p -> next , 0 , sizeof(p -> next));
        return p;
    }
    void init () {
        memset (cnt , 0 , sizeof(cnt));
        tot = 0;
        root = newnode ();
    }
    void insert (char *s) {
        trie *p = root;
        for (int i = 0 ; s[i] ; i ++) {
            int c = s[i] - 'a';
            if (p -> next[c] == NULL) {
                p -> next[c] = newnode ();
            }
            p = p -> next[c];
        }
    }
    void cal (trie *p) {
        for (int i = 0 ; i < 26 ; i ++) {
            if (p -> next[i]) {
                cnt[i] ++;
                cal (p -> next[i]);
            }
        }
    }
    void gao (trie *p) {
        for (int i = 0 ; i < 26 ; i ++) {
            if (p -> next[i] == NULL) {
                if (p != root) ans += cnt[i];
            }
            else {
                if (p != root && end[i]) ans ++;
                gao (p -> next[i]);
            }
        }
    }
}suffix , prefix;
int main () {
    while (scanf ("%d %d" , &n , &m) != EOF && n + m) {
        prefix.init ();suffix.init ();
        ans = 0;
        memset (cnt , 0 , sizeof (cnt));
        memset (end , 0 , sizeof (end));
        for (int i = 0 ; i < n ; i ++) {
            scanf ("%s" , a[i]);
            prefix.insert (a[i]);
        }
        for (int i = 0 ; i < m ; i ++) {
            scanf ("%s" , b[i]);
            reverse (b[i] , b[i] + strlen(b[i]));
            end[b[i][0] - 'a'] = 1;
            suffix.insert (b[i]);
        }
        suffix.cal (suffix.root);
        prefix.gao (prefix.root);
        printf ("%lld\n" , ans);
    }
    return 0;
}