Trie树【洛谷P3879】 [TJOI2010]阅读理解

P3879 [TJOI2010]阅读理解

题目描述

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

输入输出格式

输入格式：

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

按下来的N行，每行描述一篇短文。每行的开头是一个整数L，表示这篇短文由L个单词组成。接下来是L个单词，单词之间用一个空格分隔。

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

输出格式：

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

卡空间的毒瘤题。

怒开bitset过掉。

code：

#include <iostream>
#include <cstdio>
#include <bitset>

using namespace std;

const int wx=500017;

inline int read(){
	int sum=0,f=1; char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
	while(ch>='0'&&ch<='9'){sum=(sum<<1)+(sum<<3)+ch-'0'; ch=getchar();}
	return sum*f;
}

int n,m;
char c[wx];

struct Trie{
	int tr[wx][29];
	bitset<1010>flag[wx];
	int cnt;
	
	void insert(char *s,int fl){
		int p=0;
		for(int i=0;s[i];i++){
			int k=s[i]-'a';
			if(!tr[p][k])tr[p][k]=++cnt;
			p=tr[p][k];
		}
		flag[p][fl]=1;
	}
	
	void query(char *s){
		int p=0;
		for(int i=0;s[i];i++){
			int k=s[i]-'a';
			if(!tr[p][k]){
				puts("");
				return ;
			}
			p=tr[p][k];
		}
		for(int i=1;i<=n;i++){
			if(flag[p][i])printf("%d ",i);
		}puts("");
	}
	
}Trie;

int main(){
	n=read();
	for(int i=1;i<=n;i++){
		int x; x=read();
		for(int j=1;j<=x;j++){
			scanf("%s",c);
			Trie.insert(c,i);
		}
	}
	m=read();
	for(int i=1;i<=m;i++){
		scanf("%s",c);
		Trie.query(c);
	}
	return 0;
}