P8306 【模板】字典樹

原題連結

題解

1.建議去B站上看看動畫演示，你就明白怎麼回事了
2.如何用程式碼實現呢？看完你就明白了

code

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

int num=0;
int tree[3000006][75]={0};
int cnt[3000006]={0};
char s[3000006];
int ans;
int t,n,q;//放全域性變數是為了加快速度，但是碼風不好看
int now,son;

int inshe(char a)
{
    if(a<='z'&&a>='a') ans=a-'a'+1;
    else if(a<='Z'&&a>='A') ans=a-'A'+1+26;
    else ans=a-'0'+1+52;
    return ans;
}
void add()
{
    now=0;
    for(int i=0;s[i];i++)
    {
        son=inshe(s[i]);
        if(!tree[now][son]) tree[now][son]=++num;
        now=tree[now][son];
        cnt[now]++;//代表這個點的出現次數加一
    }
}

int searchs()
{
    now=0;
    for(int i=0;s[i];i++)
    {
        son=inshe(s[i]);
        if(!tree[now][son]) return 0;
        now=tree[now][son];
    }
    return cnt[now];
}

int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&q);
        while(n--)
        {
            scanf("%s",s);
            add();
        }
        while(q--)
        {
            scanf("%s",s);
            printf("%d\n",searchs());
        }
        for(int i=0;i<=num;i++)//從0開始，代表要把根節點也清零，但是不能用memset，太噁心了
        {
            for(int j=1;j<=70;j++) tree[i][j]=0;
            cnt[i]=0;
        }
        num=0;
    }
    return 0;
}