新學了字典樹算法,這個可以算是一個模板吧
在有n個單詞的字典中 對m個單詞詢問是否在此字典裏。對於這個問題,我們可以通過構建字典樹來解決。
trie樹的思路就是對於n個單詞,我們開一個數組to[maxn][26],代表編號爲maxn、字母編號爲x的編號(我在說啥。。。);具體思路就是構建一棵長成這樣的樹並記錄其編號
然後對於讀入的每個單詞,我們對於其末尾編號開一個b數組記錄下來代表一個單詞的終結。
接着我們讀入要查詢的m個單詞,對於每個單詞的每一位來說,如果它的編號不在這棵樹裏,就讓flag爲false並退出循環。否則就繼續讓now遞歸下去。如果到了最後flag是false並且最後一位的編號是true,就說明這個單詞在字典中。
然後對於這道題目,我的做法是構建一棵字典樹,不過對判斷數組進行記錄他是那個句子裏的,就解決了。
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<iomanip>
using namespace std;
const int maxn=500010;
int n,m;
int tot,now;
char s[10010];
int to[maxn][26];
bool b[maxn][1010];
//int g[1010];
void insert(int x)
{
scanf("%s",s);
int now=0;
for(int i=0;i<strlen(s);i++)
{
if(to[now][s[i]-'a']==0)
{
to[now][s[i]-'a']=++tot;
}
now=to[now][s[i]-'a'];
}
b[now][x]=true;
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
int num;
scanf("%d",&num);
for(int j=1;j<=num;j++)
{
insert(i);
}
}
scanf("%d",&m);
for(int i=1;i<=m;i++)
{
string s;
cin>>s;
now=0;
bool flag=false;
for(int j=0;j<s.size();j++)
{
if(to[now][s[j]-'a']==0)
{
flag=true;
break;
}
now=to[now][s[j]-'a'];
}
if(flag==false)
{
for(int j=1;j<=n;j++)
{
if(b[now][j]==true)
{
printf("%d ",j);
}
}
}
printf("\n");
}
return 0;
}