字典树板子求条谢谢啦
查看原帖
字典树板子求条谢谢啦
1118614
I_Love_DS楼主2024/11/29 22:07
#include <bits/stdc++.h>

using namespace std;

const int N = 3000001, S = 65;

int n, q;

int nxt[N][S], cnt;
int is[N];
char a[N];

int check(char x) {
	if (isupper(x)) return x - 'A';
	if (islower(x)) return 26 + x - 'a';
	return 52 + x - '0';
}

void insert(char s[], int len) {
	int now = 0;
	for (int i = 1; i <= len; i++) {
		int x = check(s[i]);
		if (!nxt[now][x]) nxt[now][x] = ++cnt;
		now = nxt[now][x];
		++is[now];
	}
}

int search(char s[], int len) {
	int now = 0;
	for (int i = 1; i <= len; i++) {
		int x = check(s[i]);
		if (!nxt[now][x]) return 0;
		now = nxt[now][x];
	}
	return is[now];
}

void solve() {
	for (int i = 1; i <= cnt; i++) 
		for (int j = 1; j <= 64; j++) 
			nxt[i][j] = 0;
	for (int i = 1; i <= cnt; i++) is[i] = 0;
	cnt = 0;
	scanf("%d%d", &n, &q);
	for (int i = 1; i <= n; i++) {
		scanf("%s", a + 1);
		int m = strlen(a + 1);
		insert(a, m);
	}
	for (int i = 1; i <= q; i++) {
		scanf("%s", a + 1);
		int m = strlen(a + 1);
		printf("%d\n", search(a, m));
	}
}

int main() {
	int t;
	scanf("%d", &t);
	while (t--) solve();
	return 0;
}
2024/11/29 22:07
加载中...