字典树模板题洛谷P2580

最新推荐文章于 2024-08-03 15:53:55 发布

SCUEC__zhazhahui

最新推荐文章于 2024-08-03 15:53:55 发布

阅读量220

点赞数

CC 4.0 BY-SA版权

分类专栏：高级数据结构

本文链接：https://blog.youkuaiyun.com/SCUEC__zhazhahui/article/details/109123029

高级数据结构专栏收录该内容

12 篇文章

订阅专栏

该程序实现了一个字符串查询系统，检查输入字符串是否存在于已添加的字符串中，并判断是否重复。利用字典树（Trie）数据结构进行高效查找，0表示字符串不存在，1表示存在且唯一，2表示存在但重复。

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

洛谷P2580

查询字符串是否出现过，以及是否重复出现。

#include<bits/stdc++.h>

using namespace std;
char s[55];
const int maxn=5e6+10;
int tree[maxn][26];
int vis[maxn];//标记第几次访问
int tot=0;
void addc(char *s)
{
    int idx=0;
    for (int i = 0; i < strlen(s); i++){
        int x=s[i]-'a';
        if(!tree[idx][x]){
            tree[idx][x]=++tot;
        }
        idx=tree[idx][x];
    }
    vis[idx]=1;
}
int check(char *s)
{
    int idx=0;
    for(int i=0;i<strlen(s);i++){
        int x=s[i]-'a';
        idx=tree[idx][x];
        if(!idx)
            return 0;
    }
    if(vis[idx]==1){
        vis[idx]=2;
        return 1;
    }
    else if(!vis[idx])
        return 0;
    return 2;
}
int main()
{
    //freopen("p2580.in","r",stdin);
    int n;
    cin>>n;
    while (n--){
        cin>>s;
        addc(s);
    }
    int m;
    cin>>m;
    while (m--){
        cin>>s;
        switch(check(s)){
            case 0:
                cout<<"WRONG";break;
            case 1:
                cout<<"OK";break;
            case 2:
                cout<<"REPEAT";
        }
        cout<<'\n';
    }
    return 0;
}