HDU 2846：Repository（Trie）

http://acm.hdu.edu.cn/showproblem.php?pid=2846

题意：给出N个模式串，再给出M个文本串，问每一个文本串在多少个模式串中出现。

思路：平时都是找前缀的，这里将模式串s[1……len]的每一个[i,len]的子串都插入，这样就可以满足条件。还要注意如果两个子串都为同一个模式串的子串，不能重复计数。可以用一个id数组装上一次是哪个串的id，如果id相同就不要重复计数了。

#include <cstdio>
#include <algorithm>
#include <iostream>
#include <cstring>
#include <string>
#include <cmath>
#include <queue>
#include <vector>
using namespace std;
#define N 2000010
char s[25];
struct Trie
{
    int ch[N][26], num, cnt[N], id[N]; //空间尽量开大点

    void init()
    {
        memset(ch[0], 0, sizeof(ch[0]));
        num = 1;
    }

    void update(char *s, int index)
    {
        int u = 0;
        int len = strlen(s);
        for(int i = 0; i < len; i++) {
            int tmp = s[i] - 'a';
            if(!ch[u][tmp]) {
                memset(ch[num], 0, sizeof(ch[num]));
                cnt[num] = 0;
                id[num] = 0;
                ch[u][tmp] = num++;
            }
            u = ch[u][tmp];
            if(id[u] != index) cnt[u]++;
            id[u] = index; //标记上一个是否是相同的串,如果相同不能重复计数
        }
    }

    int query(char *s)
    {
        int u = 0;
        int len = strlen(s);
        for(int i = 0; i < len; i++) {
            int tmp = s[i] - 'a';
            if(!ch[u][tmp]) return 0;
            u = ch[u][tmp];
        }
        return cnt[u];
    }
}trie;

int main()
{
    int n;
    scanf("%d", &n);
    trie.init();
    for(int i = 1; i <= n; i++) {
        scanf("%*c%s", s);
        int len = strlen(s);
        for(int j = 0; j < len; j++) {
            trie.update(s + j, i); //每一个串的以s[j]开头的子串都要插入
        }
    }
    scanf("%d", &n);
    for(int i = 0; i < n; i++) {
        scanf("%*c%s", s);
        printf("%d
", trie.query(s));
    }
    return 0;
}