P2580 于是他错误的点名开始了

最新推荐文章于 2022-10-12 17:06:10 发布

Object_S

最新推荐文章于 2022-10-12 17:06:10 发布

阅读量184

点赞数

CC 4.0 BY-SA版权

分类专栏： A -- 题解字符串算法 -- Trie树

本文链接：https://blog.youkuaiyun.com/u012972031/article/details/95799785

A -- 题解同时被 2 个专栏收录

90 篇文章

订阅专栏

字符串算法 -- Trie树

3 篇文章

订阅专栏

博客提及题目地址，并指出search方法因需返回多种状态，应使用int类型，同时要保存重复情况，这属于信息技术中算法实现的关键注意点。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目地址

注意点：

search方法由于需要返回多种状态，应当使用int并保存重复情况.

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int SIZE=1000010,MAXN=55;
int trie[SIZE][26],tot=1;
bool ed[SIZE],vis[SIZE];
void insert(char* str){
    int len=strlen(str),p=1;
    for(int k=0;k<len;k++){
        int ch=str[k]-'a';
        if(trie[p][ch]==0)trie[p][ch]=++tot;
        p=trie[p][ch];
    }
    ed[p]=true;
}
int search(char* str){
    int len=strlen(str),p=1;
    for(int k=0;k<len;k++){
        p=trie[p][str[k]-'a'];
        if(!p)return 0;
    }
    if(ed[p]&&vis[p])return 2;
    vis[p]=1;
    return ed[p];
}
char str[SIZE];
int main(){
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%s",&str);
        insert(str);
    }
    int m;
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        scanf("%s",&str);
        int tmp=search(str);
        if(!tmp)printf("WRONG\n");
        else if(tmp==1)printf("OK\n");
        else if(tmp==2)printf("REPEAT\n");
    }
    return 0;
}