Trie树10pts求条(玄关)
查看原帖
Trie树10pts求条(玄关)
823949
Hhy882577楼主2024/7/21 20:07
#include<bits/stdc++.h>
#define N 500005
using namespace std;
int sum[N],n,m,len;
bool path[N][26];
string s,a;
void insert(){
	int p=1,cnt=0;
	len=s.size();
	for(int i=0;i<len;i++){
		int j=s[i]-'a';
		if(path[p][j]==0)
			path[p][j]=++cnt;
		p=path[p][j];
	}
	sum[p]=1;
	return ;
}
void query(){
	int p=1;
	int llen=a.size();
	for(int i=0;i<llen;i++){
		int j=a[i]-'a';
		if(path[p][j]==0){
			cout<<"WRONG\n";
			return ;
		}
		p=path[p][j];
	}
	if(sum[p]==1){
		sum[p]++;
		cout<<"OK\n";
		return ;
	}
	if(sum[p]>1){
		cout<<"REPEAT\n";
		return ;
	}
	return ;
}
int main(){
	//ios::sync_with_stdio(0);
	//cin.tie(0);
	//cout.tie(0);
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>s;
		insert();
	}
	cin>>m;
	for(int i=1;i<=m;i++){
		cin>>a;
		query();
	}
	return 0;
} 
2024/7/21 20:07
加载中...