洛谷-3879 [TJOI2010]阅读理解

题目描述
英语老师留了N篇阅读理解作业，但是每篇英文短文都有很多生词需要查字典，为了节约时间，现在要做个统计，算一算某些生词都在哪几篇短文中出现过。
输入格式
第一行为整数N，表示短文篇数，其中每篇短文只含空格和小写字母。
按下来的N行，每行描述一篇短文。每行的开头是一个整数L，表示这篇短文由L个单词组成。接下来是L个单词，单词之间用一个空格分隔。
然后为一个整数M，表示要做几次询问。后面有M行，每行表示一个要统计的生词。
输出格式
对于每个生词输出一行，统计其在哪几篇短文中出现过，并按从小到大输出短文的序号，序号不应有重复，序号之间用一个空格隔开（注意第一个序号的前面和最后一个序号的后面不应有空格）。如果该单词一直没出现过，则输出一个空行。

输入输出样例
输入 #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 ≤ M ≤ 1,000
对于100%的数据，1 ≤ M ≤ 10,000，1 ≤ N ≤ 1000
每篇短文长度（含相邻单词之间的空格） ≤ 5,000 字符，每个单词长度 ≤ 20 字符
每个测试点时限2秒
感谢@钟梓俊添加的一组数据

解释:对每一篇文章，我们建立一颗字典树，这样查询其实就是在字典树上进行查询,直接暴力

#include<iostream>
using namespace std;
int n=0;
int rev(char a){
    if(a<='z'&&a>='a') return a-'a';
    return a-'A'+26;
}
struct node{
    node *tree[26];
    int v;
};
void free(node *rt){
    if(!rt) return;
    for(int i=0;i<26;i++) free(rt->tree[i]);
    delete rt;
}
void init(node *rt){
    for(int i=0;i<26;i++) rt->tree[i]=NULL;
    rt->v=0;
}
void insert(node *rt,char *str){
    node *temp=rt;
    for(int i=0;str[i];i++){
        int y=rev(str[i]);
        if(temp->tree[y]==NULL){
            temp->tree[y]=new node;
            init(temp->tree[y]);
        }
        temp=temp->tree[y];
    }
    temp->v=1;
}
bool query(node *rt,char *str){
    node *temp=rt;
    for(int i=0;str[i];i++){
        int y=rev(str[i]);
        if(temp->tree[y]==NULL) return 0;
        temp=temp->tree[y];
    }
    return temp->v;
}
node *root[1005];
int main(){
    ios::sync_with_stdio(false);
    cin>>n;
    for(int i=1;i<=n;i++){
        root[i]=new node;
        int m=0;cin>>m;
        char str[33];
        for(int j=1;j<=m;j++){
            cin>>str;
            insert(root[i],str);
        }
    }
    int q=0;cin>>q;
    while(q--){
        char str[33];cin>>str;
        for(int i=1;i<=n;i++){
            if(query(root[i],str)) cout<<i<<" ";
        }
        cout<<endl;
    }
    for(int i=1;i<=n;i++) free(root[i]);
    return 0;
}