前言:烦死了,这个题目一看就是用字典树来做,但是空间不知道开多大,烦死了
后来发现其实tree的第一维空间直接开极端的情况就行,就好像这一题,最多有 1e4 个字符串,每个字符串最长为 50,那我们假设所有的字符都是a,那我们必须要开 50 * 1e4 的空间
#define _CRT_SECURE_NO_WARNINGS
#include<bits/stdc++.h>
using namespace std;
int tree[55*27][27];
int record[55*27];
int idx = 0;
int n,m;
void insert(char *a){
int p = 0;
for(int i=0;a[i];i++){
int u = a[i]-'a';
if(!tree[p][u]) tree[p][u] = ++idx;
p = tree[p][u];
}
record[p] = 1;
}
int query(char *a){
int p = 0;
for(int i=0;a[i];i++){
int u = a[i] - 'a';
if(!tree[p][u]) return 0;
p = tree[p][u];
}
if(record[p]==1){
record[p] ++; return 1;
}
return record[p];
}
int main(){
cin >> n;
char a[60];
for(int i=1;i<=n;i++){
cin >> a;
insert(a);
}
cin >> m;
for(int i=1;i<=m;i++){
cin >> a;
int t = query(a);
if(t==0) cout << "WRONG" << endl;
if(t==1) cout << "OK" << endl;
if(t>=2) cout << "REPEAT" << endl;
}
return 0;
}