洛谷 P2580 于是他错误的点名开始了 字典树

本来是计划写一篇kmp的，但学了一天有些地方还是有些没搞懂，然后就碰到一道字典树的题就想着复习一下，kmp…往后拖一拖吧。

于是他错误的点名开始了

其实就是道板子题，在板子的基础上多了个判重，所以只需要另外设置一个used数组，来记录一下树上的路径，如果一个字符串在遍历的过程中每一个点都被标记过，那就说明是重复的。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<iostream>
#include<iomanip>
#include<algorithm>
#include<stack>
#include<queue>
#include<cstdlib>
#include<string>
#include<vector>
#include<list>
#include<map>
#include<set>
#define fi first
#define se second
#define pb push_back
#define me(a,b) memset(a,b,sizeof(a))
#define INIT() std::ios::sync_with_stdio(false)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int>P;
const int MAX_N=1000000+5;
const int INF=0x7fffffff;
const int inf=100000;
const double EPS=1e-6;
const ull base=123;
const ll mod=1e9+7;
const double pi=4*atan(1.0);
int trie[MAX_N][30];
int color[MAX_N];
char s[MAX_N];
int k=1;
int used[MAX_N];
void build()
{
    int len=strlen(s);
    int p=0;
    int i;
    for(i=0;i<len;i++)
    {
        int a=s[i]-'a';
        if(!trie[p][a])
        {
            trie[p][a]=k;
            k++;
        }
        p=trie[p][a];
    }
    color[p]=1;
}
int Find()
{
    int len=strlen(s);
    int p=0;
    int flag=0;
    for(int i=0;i<len;i++)
    {
        int a=s[i]-'a';
        if(!trie[p][a])
            return 0;
        p=trie[p][a];
        if(!used[p])
        {
            used[p]=1;
            flag=1;
        }
    }
    if(!flag)
        return 2;
    if(color[p])
        return 1;
}
int main()
{
    int n;
    cin>>n;
    while(n--)
    {
        scanf("%s",s);
        build();
    }
    cin>>n;
    while(n--)
    {
        scanf("%s",s);
        int ans=Find();
        if(ans==1)
            cout<<"OK"<<endl;
        else if(ans==0)
            cout<<"WRONG"<<endl;
        else
            cout<<"REPEAT"<<endl;
    }
}