RE+TLE求助

P8306 【模板】字典树

Jerry_Money @ 2024-08-07 14:56:43

#include<bits/stdc++.h>
using namespace std;
int n,Q;
string a[1000086];
string sb;
struct sbb{
    int tot,n=0;
}e[1000086][26+26+10];
int tot;
int maxx;
int now,ans;
int t;
int c;
int cti(char x){
    if(x>='a'&&x<='z')c=x-'a';
    else if(x>='A'&&x<='Z')c=x-'A'+26;
    else c=x-'0'+26+26;
    return c;
}
void add(string x){
    now=0;
    for(int i=0;i<x.size();i++){
        t=cti(x[i]);
        if(!e[now][t].tot)e[now][t].tot=++tot;
        e[now][t].n++;
        now=e[now][t].tot;
    }
    return;
}
int ask(string x){
    ans=0,now=0;
    for(int j=0;j<x.size();j++){
        t=cti(x[j]);
        if(!e[now][t].tot)break;
        if(j=x[x.size()-1]){
            ans+=e[now][t].n;
            break;
        }
        now=e[now][t].tot;
    }
    return ans;
}
void clean(){
    for(int i=0;i<=1000086;i++)for(int j=0;j<26+26+10;j++)e[i][j].n=0;
}
int _u,_v;
void run(){
    cin>>n>>Q;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        add(a[i]);
    }
    while(Q--){
        cin>>sb;
        cout<<ask(sb)<<endl;
    }
    clean();
    return;
}
int main(){
    int t1;
    cin>>t1;
    while(t1--)run();
}

by hytallenxu @ 2024-08-07 14:58:59

@Jerry_Money runtime error: index 1000086 out of bounds for type 'sbb [1000086][62]'


by Jerry_Money @ 2024-08-07 15:00:44

@hytallenxu 啥意思啊佬大看不太懂


by masonxiong @ 2024-08-07 15:01:24

@Jerry_Money

您的 clean 函数令人忍俊不禁

for(int i=0;i<=1000086;i++)for(int j=0;j<26+26+10;j++)e[i][j].n=0;

这不保证越界吗


by Jerry_Money @ 2024-08-07 15:11:25

@masonxiong 啊?那应该咋办


by masonxiong @ 2024-08-07 15:16:23

@Jerry_Money 您一个橙名绿勾不会改这个?

改成 memset 不就好了


by yqz1005 @ 2024-08-07 15:46:29

"且输入字符串的总长度不超过 3×10^6"
你开1000086不够。


by yqz1005 @ 2024-08-07 15:53:11

而且不应

for(int i=0;i<=1000086;i++)for(int j=0;j<26+26+10;j++)e[i][j].n=0;

应该

for(int i=0;i<1000086;i++)for(int j=0;j<26+26+10;j++)e[i][j].n=0;

by yqz1005 @ 2024-08-07 15:56:17

更准确的来说,应该

for(int i=0;i<3000009;i++)for(int j=0;j<26+26+10;j++)e[i][j].n=0;

(数组开3000009后)


by wintorsd @ 2024-08-12 10:53:25

这里用memset要MLE


|