求条
查看原帖
求条
1428495
KMYC楼主2024/10/7 10:49
#include<bits/stdc++.h>
using namespace std;
const int N=3000010;
struct node{ int child[128],cnt; }trie[N];
int tot;
void insert(string s){
	int len=s.size();
	int cur=0;
	for(int i=0;i<len;i++){
		if(trie[cur].child[s[i]]==0) trie[cur].child[s[i]]=++tot;
		trie[cur].cnt++;
		cur=trie[cur].child[s[i]];
	}
}
int find(string s){
	int len=s.size();
	int cur=0;
	for(int i=0;i<len;i++){
		if(trie[cur].child[s[i]]==0) return 0;
		cur=trie[cur].child[s[i]];
	}
	return trie[cur].cnt;
}
int t,n,q;
int main(){
	cin>>t;
	while(t--){
		cin>>n>>q;
		for(int i=1;i<=tot;i++){
			trie[i].cnt=0;
			for(int j=0;j<128;j++) trie[i].child[j]=0;
		}
		tot=0;
		string str;
		for(int i=1;i<=n;i++){
			cin>>str;
			insert(str);
		}
		while(q--){
			cin>>str;
			cout<<find(str)<<"\n";
		}
	} 
	return 0;
}
2024/10/7 10:49
加载中...