字典树模板求调,TLE 52分

P8306 【模板】字典树

ncwzdlsd @ 2023-02-14 14:14:40

蒟蒻复健字典树又双叒叕写挂了/kk/kk/kk

求调 谢谢www

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

const int maxn=2e6+5;
int t[maxn][65],cnt[maxn],tot;
char s[maxn];

int getn(char x)
{
    if(x<='Z'&&x>='A') return x-'A';
    else if(x<='z'&&x>='a') return x-'a'+26;
    else return x-'0'+52;
}

void insert(char s[])
{
    int p=0,len=strlen(s);//根节点为0
    for(int i=0;i<len;i++)
    {
        int u=getn(s[i]);//当前字母子节点
        if(!t[p][u]) t[p][u]=++tot;//如果当前子节点不存在就创造一个点来存储子节点
        p=t[p][u];//让p走到子节点的位置
        cnt[p]++;//结尾是p的字符串个数增加
    }
}

int ask(char s[])
{
    int p=0,len=strlen(s);
    for(int i=0;i<len;i++)
    {
        int u=getn(s[i]);
        if(!t[p][u]) return 0;
        p=t[p][u];
    }
    return cnt[p];
}

int main()
{
    int T;cin>>T;
    while(T--)
    {
        for(int i=0;i<tot;i++)
            for(int j=0;j<65;j++) t[i][j]=0;
        for(int i=0;i<tot;i++) cnt[i]=0;
        int n,q;cin>>n>>q;
        for(int i=1;i<=n;i++)
            cin>>s,insert(s);
        for(int i=1;i<=q;i++)
            cin>>s,cout<<ask(s)<<endl;
    }
    return 0;
}

by Steven24 @ 2023-02-14 14:16:50

@ncwzdlsd 您是不是忘了每次循环把 tot 置零了/fad


by ncwzdlsd @ 2023-02-14 14:19:36

@Steven24 是这样的,谢谢 QwQ


|