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

点击查看代码

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

const int N=5e5+10;
int tr[N][26],tag[N],idx;
int n,m;

void insert(const string& s)
{
    int p=0;
    for(char c : s){
        int u=c-'a';
        if(!tr[p][u]) tr[p][u]=++idx;
        p=tr[p][u];
    }
    tag[p]=1;
}

int query(const string& s)
{
    int p=0;
    for(char c:s){
        int u=c-'a';
        if(!tr[p][u]) return 0;
        p=tr[p][u];
    }

    if(tag[p]==1){
        tag[p]=2;
        return 1;
    }else if(tag[p]==2){
        return 2;
    }else return 0;
}

int main()
{
    ios::sync_with_stdio(0),cin.tie(0);
    cin>>n;
    while(n--){
        string s;
        cin>>s;
        insert(s);
    }
    cin>>m;
    while(m--){
        string s;
        cin>>s;
        int ans=query(s);
        if(ans==1){
            cout<<"OK"<<"\n";
        }else if(ans==2){
            cout<<"REPEAT"<<"\n";
        }else{
            cout<<"WRONG"<<"\n";
        }
    }

    return 0;
}