字典树简单入门题(居然是蓝题?)

[TJOI2010] 阅读理解

题目描述

英语老师留了 $N$ 篇阅读理解作业，但是每篇英文短文都有很多生词需要查字典，为了节约时间，现在要做个统计，算一算某些生词都在哪几篇短文中出现过。

输入格式

第一行为整数 $N$ ，表示短文篇数，其中每篇短文只含空格和小写字母。

按下来的 $N$ 行，每行描述一篇短文。每行的开头是一个整数 $L$ ，表示这篇短文由 $L$ 个单词组成。

接下来是 $L$ 个单词，单词之间用一个空格分隔。

然后为一个整数 $M$ ，表示要做几次询问。后面有 $M$ 行，每行表示一个要统计的生词。

输出格式

对于每个生词输出一行，统计其在哪几篇短文中出现过，并按从小到大输出短文的序号，序号不应有重复，序号之间用一个空格隔开（注意第一个序号的前面和最后一个序号的后面不应有空格）。如果该单词一直没出现过，则输出一个空行。

样例 #1

样例输入 #1

3
9 you are a good boy ha ha o yeah
13 o my god you like bleach naruto one piece and so do i
11 but i do not think you will get all the points
5
you
i
o
all
naruto

样例输出 #1

1 2 3
2 3
1 2
3
2

提示

对于 $30\%$ 的数据， $1\le M\le 10^3$ 。

对于 $100\%$ 的数据，$1\le M\le 10^4$，$1\le N\le 10^3$ 。

每篇短文长度（含相邻单词之间的空格）$\le 5\times 10^3$ 字符，每个单词长度 $\le 20$ 字符。

每个测试点时限 $2$ 秒。

所有过程都在注释里

#include<bits/stdc++.h>
using namespace std; 
const int N=5e5+5;
typedef long long ll;
int son[N][30],idx,n,l,m;
char a[N];
bitset<1010> sign[N];
// 用bool会爆空间
// 用bool一个值占1字节,1个字节是8比特
// 用bitset则每个bool值只占用一个比特,大大节省空间
void insert(char str[],int k){
     //插入字符串,同时告知是哪行 
	int p=0; //从根节点开始遍历插入
	int l=strlen(str);
	for(int i=0;i<l;i++){
    
		int u=str[i]-'a'; // 方便当作下标 
		if(!son[p][u]) son[p][u]=++idx; 
		//查看当前节点p的子节点u是否已给予编号,没有则标记给予编号
		p=son[p][u];//移至下一节点转换位置
	}
	sign[p][k]=1;//标记该字符串存在于第k行
}
void find(char str[]){
    //查找该字符串
	int p=0,signsign=1;//初始化,回到根节点继续查
	int l=strlen(str);
	for(int i=0;i<l;i++){
    //遍历节点进行查询
		int u=str[i]-'a';
		if(!son[p][u]){
    //一旦发现未被标记，则不存在该字符串，退出遍历查询
			signsign=0;//同时标记不存在
			break;
		}
		p=son[p][u];//移至下一节点转换位置
	}
	if(signsign){
    //通过标记判断是否查询成功
		for(int i=1;i<=n;i++){
    
			if(sign[p][i]){
    //通过标记查看此行是否存在此字符串
				printf("%d ",i);
			}
		}printf("\n");
	}
	else{
    
		puts("");
	}
}
int main(){
    
	cin>>n;
	for(int i=1;i<=n;i++){
    
		cin>>l;
		for(int j=1;j<=l;j++){
    
			scanf("%s",a);
			insert(a,i);
		}
	}
	cin>>m;
	for(int i=1;i<=m;i++){
    
		scanf("%s",a);
		find(a);
	}
	return 0;
}