HDU 2846 Trie查询

给出若干模式串，再给出若干询问串，求每个询问串作为多少个模式串的子串出现。

如果一个串是另一个串的子串，则一定是另一个串某个前缀的后缀或者某个后缀的前缀。根据字典树的性质，将模式串的每一个后缀插入字典树中，同时更新字典树中节点的cnt值。这里需要注意不要重复累加贡献，可以在字典树中新增一个num的信息值来实现这一点。

#include <iostream>
#include <vector>
#include <algorithm>
#include <string>
#include <string.h>
#include <stdio.h>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <cmath>
#include <ctime>
#include <cassert>
#include <sstream>
using namespace std;

const int N=223456;
const int MOD=1e9+7;


const int CHARSET=26,BASE='a',MAX_NODE=10000*20*26+5000;
struct Trie {
    int tot,root,child[MAX_NODE][CHARSET];
    int flag[MAX_NODE],num[MAX_NODE];
    Trie(){
        init();
    }
    void init(){
        tot=0;
        root=newNode();
    }
    int newNode() {
        ++tot;
        memset(child[tot],0,sizeof(child[tot]));
        flag[tot]=0;
        num[tot]=-1;
        return tot;
    }
    void insert(const char *str,int id){
        int *cur=&root;
        for (const char *p=str;*p;++p){
            cur=&child[*cur][*p-BASE];
            if (*cur==0){
                *cur=newNode();
            }
            if (num[*cur]!=id)
                flag[*cur]++;
            num[*cur]=id;
        }
    }
    int query(const char *str){
        int *cur=&root;
        for (const char *p=str;*p&&(*cur);++p){
            cur=&child[*cur][*p-BASE];
        }
        return flag[*cur];
    }
}trie;
int main () {
    int n;
    scanf("%d",&n);
    char s[120];
    for (int i=1;i<=n;i++) {
        scanf("%s",s);
        int len=strlen(s);
        for (int j=0;j<len;j++)
            trie.insert(s+j,i);
    }
    int Q;
    scanf("%d",&Q);
    while (Q--) {
        scanf("%s",s);
        int ret=trie.query(s);
        printf("%d
",ret);
    }
    return 0;
}