P3879 [TJOI2010]阅读理解
STL
Trie+bitset
想练习tire的我被卡的一愣愣的qaq。数组开的小了是WA#11.开成1000就是前面十个点随机MLE。?
具体怎么用tire过这个题就是把标记出现的bool数组改成bitset，节省32倍空间，你，值得拥有。?
定义一个bitset后当二维数组用就好啦。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e6+6;
char s[25];
struct node
{
int son[26];
bool mark;
} trie[maxn];
int root=0,num=0;
bitset<1005>bt[maxn];
int n,m;
bool insert_check(char *s,int id)
{
int pos=root;
for(int i=0; s[i]; i++)
{
int now=s[i]-'a';
if(!trie[pos].son[now])
trie[pos].son[now]=++num;
pos=trie[pos].son[now];
}
bt[pos][id]=1;
int tmp=trie[pos].mark;
trie[pos].mark=1;
return tmp!=0;
}
void find_check(char *s)
{
int pos=root;
for(int i=0; s[i]; i++)
{
int now=s[i]-'a';
if(!trie[pos].son[now])
{
printf("\n");
return ;
}
pos=trie[pos].son[now];
}
for(int i=1; i<=n;i++)
{
if(bt[pos][i])
printf("%d ",i);
}
printf("\n");
}
int main()
{
scanf("%d",&n);
for(int i=1; i<=n; i++)
{
scanf("%d",&m);
for(int j=1; j<=m; j++)
{
scanf("%s",s);
insert_check(s,i);
}
}
scanf("%d",&m);
for(int i=1; i<=m; i++)
{
scanf("%s",s);
find_check(s);
}
}