[数据结构]trie

注意一下一个字符串在trie中存在不一定代表这是一个完整的单词，要用eq判断是否是一个完整的单词。
这个trie是邻接表实现的，空间换时间。AC自动机会超时，需要使用邻接矩阵实现。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define maxn 200005
struct node{
    node() : c(0), eq(0), cnts(0) {sub.clear();}
    char c;
    vector<node*> sub;
    bool eq;
    int cnts;
}*root = new node;
void add(const char s[])
{
    node *p = root;
    int j = 0, ls = strlen(s);
    for (int i = 0; i < ls; ++i){
        ++p -> cnts;
        for (j = 0; j < p -> sub.size(); ++j) if (p -> sub[j] -> c == s[i]) break;
        if (j < p -> sub.size()) p = p -> sub[j];
        else{
            node *t = new node;
            t -> c = s[i];
            p -> sub.push_back(t);
            p = t;
        }
    }
    p -> eq = true;
}
int query(const char s[])
{
    node *p = root;
    int j = 0, ls = strlen(s);
    for (int i = 0; i < ls; ++i){
        for (j = 0; j < p -> sub.size(); ++j) if (p -> sub[j] -> c == s[i]) break;
        if (j < p -> sub.size()) p = p -> sub[j];
        else return 0;
    }
    return p -> eq ? p -> cnts + 1 : p -> cnts;
}
int main()
{
    char s[15];
    while (gets(s) && strlen(s)) add(s);
    while (~scanf("%s", s)) printf("%d\n", query(s));
    return 0;
}

这是邻接矩阵的trie，记得先new一个root，然后传进参数。

struct node{
    int cnt;
    node *nex[26];
    node()
    {
        cnt = 0;
        memset(nex, 0, sizeof(nex));
    }
};
void ins(node *r, char *s)
{
    int ls = strlen(s);
    for (int i = 0; i < ls; ++i)
    {
        int id = s[i] - 'a';
        if(r -> nex[id] == NULL) r -> nex[id] = new node();
        r = r -> nex[id];
        ++r -> cnt;//统计单词数量请把这一行放在循环外面
    }
}