trie字符串统计

原题链接： 835. Trie字符串统计 - AcWing题库

不管是链表、trie树还是堆，他们的基本单元都是一个个节点连接构成的

都可以成为一个由本身的值e 和指向下一个节点的指针 ne 构成的链式结构。 idx ++ 保障了不同的节点总是对应不同的idx值，

因此可以用，idx将这两个属性联系起来。

链表：

链表会使用以下四件套：

h, e[N] , ne[N], idx;

h 表示头节点指针，一开始初始化指向 - 1， 每次插入的操作是 idx ++ 。

单链表
void add_to_head (int x)
{

   e[idx] = x;
   ne[idx] = h;
   h = idx ++;// h 始终指向头节点处，idx 始终指向下一个空位

}

双链表
void insert(int a, int x)
{

   e[idx] = x;
   
   l[idx] = a;
   r[idx] = r[a];
  
   l[r[a]] = idx;
   r[a] = idx ++; // 整个就是一个连线的过程

}

trie 树

trie 树中有个二维数组son[N][26], 相当于next数组

void insert(char str[])
{
    int p = 0; // 从根节点开始遍历
    for(int i = 0; str[i]; i++)
    {
        int u = str[i] - 'a';
        if(!son[p][u]) son[p][u] = ++idx;
        p = son[p][u]; // son 相当于 ne 数组
    }
    cnt[p] ++; // 记录字符串结尾位置
}

完整代码：

#include<iostream>
using namespace std;

const int N = 1e5 + 10;

int son[N][26]; 相当于ne指针
int cnt[N]; 记录字符串结尾位置
int idx；
char str[N];

void insert(char *str)
{
    int p = 0;
    for(int i = 0; str[i]; i++)
    {
        int u = str[i] - 'a';
        if(!son[p][u]) son[p][u] = ++idx;
        p = son[p][u];
    }
    cnt[p] ++;
}

int query(char *str)
{
    int p = 0; 
    for(int i = 0; str[i]; i++)
    {
        int u = str[i] - 'a';
        if(!son[p][u]) return 0;
        p = son[p][u];
    }
    return cnt[p];
}

int main()
{
    int n;
    scanf("%d", &n);
    char op[2];
    while(n--)
    {
        scanf("%s%s", op,str);
        if(op[0] == 'I') insert(str);
        else printf("%d\n", query(str));
    }
    return 0;
}