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;
}

转载于:https://www.cnblogs.com/wangxiaodai/p/9901644.html