社区讨论

字典树板子求条谢谢啦

P8306【模板】字典树参与者 2已保存回复 3

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
3 条
当前快照
1 份
快照标识符
@m42tkt69
此快照首次捕获于
2024/11/29 22:09
去年
此快照最后确认于
2025/11/04 13:38
4 个月前
查看原帖
CPP
#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;
}

回复

3 条回复,欢迎继续交流。

正在加载回复...