听取WA声一片,0分,求dalao帮助
查看原帖
听取WA声一片,0分,求dalao帮助
970743
I_LOVE_XYN楼主2024/5/22 20:32
#include <iostream>
using namespace std;
inline int read(){
	char x = getchar();
	int l = 1,d = 0;
	while(!isdigit(x)) l = (x == '-' ? -l : l),x = getchar();
	while(isdigit(x)) d = (d << 1) + (d << 3) + x - 48,x = getchar();
	return l * d;
}
inline void write(int x){
	if(x < 0) putchar('-'),x = -x;
	if(x > 9) write(x / 10);
	putchar(x % 10 + '0');
}
int n,m,ind = 1,trie[30][2000006],b[2000006];
string s;
void insert(string s){
	int p = 1;
	for(int i = 0;i < s.size(); i++){
		char c = s[i] - 'a';
		if(!trie[c][p]) trie[c][p] = ++ind;
		p = trie[c][p];
	}
	b[p] = 1;
}
int query(string s){
	int p = 1,cnt = 0;
	for(int i = 0;i < s.size(); i++){
		char c = s[i] - 'a';
		if(!trie[c][p]) break;
		else cnt += b[trie[c][p]];
		p = trie[c][p];
	}
	return cnt;
}
int main(){
	n = read(),m = read();
	for(int i = 1;i <= n; i++){
		cin >> s;
		insert(s);
	}
	for(int i = 1;i <= m; i++){
		cin >> s;
		write(query(s)),putchar('\n');
	}
	return 0;
}
2024/5/22 20:32
加载中...