洛谷 P2580 于是他错误的点名开始了 题解

每日一题 day10 打卡

Analysis

trie树模板题，只需用到简单的插入和查询就好了

如果想要学trie树，见信息学奥赛一本通·提高篇P82

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
int n,m,cnt=0;
string a[10001];
int ch[500001][105];
bool bo[500001],book[500001];
inline int read() 
{
    int x=0;
    bool f=1;
    char c=getchar();
    for(; !isdigit(c); c=getchar()) if(c=='-') f=0;
    for(; isdigit(c); c=getchar()) x=(x<<3)+(x<<1)+c-'0';
    if(f) return x;
    return 0-x;
}
void insert(string x)
{
    int len=x.size();
    int u=1;
    for(int i=0;i<len;i++)
    {
        int c=x[i]-'a';
        if(ch[u][c]==0)ch[u][c]=++cnt;
        u=ch[u][c];
    }
    bo[u]=true;
}
int check(string y)
{
    int len=y.size();
    int u=1;
    for(int i=0;i<len;i++)
    {
        int c=y[i]-'a';
        if(ch[u][c]==0)return 2;
        u=ch[u][c];
    }
    if(bo[u]!=true)return 2;
    else if(bo[u]==true&&book[u]==0)
    {
        book[u]=1;
        return 1;
    }
    else if(bo[u]==true&&book[u]==1)
    {
        return 3;
    }
}
int main()
{
    n=read();
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        insert(a[i]);
    }
    m=read();
    string x;
    for(int i=1;i<=m;i++)
    {
        cin>>x;
        int se=check(x);
        if(se==1)cout<<"OK"<<endl;
        if(se==2)cout<<"WRONG"<<endl;
        if(se==3)cout<<"REPEAT"<<endl;
    }
    return 0;
}

请各位大佬斧正（反正我不认识斧正是什么意思）