[HDOJ2846]Repository（字典树）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2846

题意：求一堆串中有多少个以0开始的子串包含目标串。

可以把所有的串符合要求的子串放入字典树统计。这时候会有一个问题，那就是adddd这样的单词：样例中已经说明了，这样的单词明显是只算一次的。所以可以在字典树中打标记，每一个节点最后更新的时候的字符串是哪个。

#include <bits/stdc++.h>
using namespace std;

typedef struct Node {
    Node* next[26];
    int cnt, id;
}Node;
int n, q, ncnt;
char tmp[22];
Node memory[350000];
Node* rt;

void insert(Node* rt, char* s, int id) {
    Node* p = rt;
    for(int i = 0; s[i]; i++) {
        int x = s[i] - 'a';
        if(p->next[x] == NULL) {
            p->next[x] = &memory[ncnt++];
            p->next[x]->id = -1;
        }
        p = p->next[x];
        if(id != p->id) {
            p->cnt++;
            p->id = id;
        }
    }
}

int query(Node* rt, char* s) {
    Node* p = rt;
    for(int i = 0; s[i]; i++) {
        int x = s[i] - 'a';
        if(p->next[x] == NULL) return 0;
        p = p->next[x];
    }
    return p->cnt;
}

int main() {
    // freopen("in", "r", stdin);
    while(~scanf("%d", &n)) {
        ncnt = 0;
        memset(memory, 0, sizeof(memory));
        rt = &memory[ncnt++];
        memset(rt->next, 0, sizeof(rt->next));
        for(int i = 0; i < n; i++) {
            scanf("%s", tmp);
            for(int j = 0; tmp[j]; j++) {
                insert(rt, tmp+j, i);
            }
        }
        scanf("%d", &q);
        while(q--) {
            scanf("%s", tmp);
            printf("%d
", query(rt, tmp));
        }
    }
    return 0;
}