洛谷 P3879 [TJOI2010]阅读理解（trie树）

传送门

---

解题思路

普通的trie貌似会MLE

但是我们可以偷偷数组少开个零，就A了

想知道为什么？

别问，问就是数据水。

其实正解map bitset等等……

看来以后要学学bitset……

AC代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<iomanip>
#include<cmath>
#include<set>
using namespace std;
const int maxn=500005;
int n,m,nn,cnt=1,ch[maxn][26];
set<int> vis[maxn];
string s;
void insert(int id){
    int len=s.length(),now=1;
    for(int i=0;i<len;i++){
        if(!ch[now][s[i]-'a']) ch[now][s[i]-'a']=++cnt; 
        now=ch[now][s[i]-'a'];
    }
    vis[now].insert(id);
}
void query(){
    int len=s.length(),now=1;
    for(int i=0;i<len;i++){
        if(!ch[now][s[i]-'a']){
            cout<<endl;
            return;
        }
        now=ch[now][s[i]-'a'];
    }
    if(vis[now].empty()) cout<<endl;
    else{
        for(set<int>::iterator i=vis[now].begin();i!=vis[now].end();i++) cout<<(*i)<<" ";
        cout<<endl;
    }
}
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>nn;
        for(int j=1;j<=nn;j++){
            cin>>s;
            insert(i);
        }
    }
    cin>>m;
    for(int i=1;i<=m;i++){
        cin>>s;
        query();
    }
    return 0;
}