【題解】洛谷P3879 閱讀理解(trie樹)

新學了字典樹算法,這個可以算是一個模板吧

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

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章