1. 程式人生 > >P3879 [TJOI2010]閱讀理解

P3879 [TJOI2010]閱讀理解

\(\color{#0066ff}{ 題目描述 }\)

英語老師留了N篇閱讀理解作業,但是每篇英文短文都有很多生詞需要查字典,為了節約時間,現在要做個統計,算一算某些生詞都在哪幾篇短文中出現過。

\(\color{#0066ff}{輸入格式}\)

第一行為整數N,表示短文篇數,其中每篇短文只含空格和小寫字母。

按下來的N行,每行描述一篇短文。每行的開頭是一個整數L,表示這篇短文由L個單片語成。接下來是L個單詞,單詞之間用一個空格分隔。

然後為一個整數M,表示要做幾次詢問。後面有M行,每行表示一個要統計的生詞。

\(\color{#0066ff}{輸出格式}\)

對於每個生詞輸出一行,統計其在哪幾篇短文中出現過,並按從小到大輸出短文的序號,序號不應有重複,序號之間用一個空格隔開(注意第一個序號的前面和最後一個序號的後面不應有空格)。如果該單詞一直沒出現過,則輸出一個空行。

\(\color{#0066ff}{輸入樣例}\)

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

\(\color{#0066ff}{輸出樣例}\)

1 2 3
2 3
1 2
3
2

\(\color{#0066ff}{資料範圍與提示}\)

對於30%的資料,1 ≤ M ≤ 1,000

對於100%的資料,1 ≤ M ≤ 10,000,1 ≤ N ≤ 1000

每篇短文長度(含相鄰單詞之間的空格) ≤ 5,000 字元,每個單詞長度 ≤ 20 字元

\(\color{#0066ff}{ 題解 }\)

對所有單詞建立Trie樹,每個點維護一個set,代表屬於哪一行

匹配到的時候掃一遍set就行了

#include<bits/stdc++.h>
#define LL long long
LL in() {
    char ch; LL x = 0, f = 1;
    while(!isdigit(ch = getchar()))(ch == '-') && (f = -f);
    for(x = ch ^ 48; isdigit(ch = getchar()); x = (x << 1) + (x << 3) + (ch ^ 48));
    return x * f;
}
using std::set;
struct Trie {
protected:
    struct node {
        node *ch[26];
        set<int> s;
        node() {
            memset(ch, 0, sizeof ch);
            s.clear();
        }
        void *operator new (size_t) {
            static node *S = NULL, *T = NULL;
            return (S == T) && (T = (S = new node[1024]) + 1024), S++;
        }
    };
    node *root;
public:
    Trie() { root = new node(); }
    void ins(char *s, int id) {
        node *o = root;
        for(char *p = s; *p; p++) {
            int pos = *p - 'a';
            if(!o->ch[pos]) o->ch[pos] = new node();
            o = o->ch[pos];
        }
        o->s.insert(id);
    }
    void query(char *s) {
        node *o = root;
        for(char *p = s; *p; p++) {
            int pos = *p - 'a';
            if(o->ch[pos]) o = o->ch[pos];
            else goto noans;
        }
        for(auto &k : o->s) printf("%d ", k);
        noans:;
        puts("");
    }
}T;
char s[500];
int main() {
    int n = in();
    for(int i = 1; i <= n; i++) {
        int k = in();
        for(int j = 1; j <= k; j++) {
            scanf("%s", s);
            T.ins(s, i);
        }
    }
    for(int m = in(); m --> 0;) {
        scanf("%s", s);
        T.query(s);
    }
    return 0;
}