hihocoder Trie樹 hiho題庫 #1014 : Trie樹
http://hihocoder.com/problemset/problem/1014
#1014 : Trie樹
描述
小Hi和小Ho是一對好朋友,出生在信息化社會的他們對編程產生了莫大的興趣,他們約定好互相幫助,在編程的學習道路上一同前進。
這一天,他們遇到了一本詞典,於是小Hi就向小Ho提出了那個經典的問題:“小Ho,你能不能對於每一個我給出的字符串,都在這個詞典裏面找到以這個字符串開頭的所有單詞呢?”
身經百戰的小Ho答道:“怎麼會不能呢!你每給我一個字符串,我就依次遍歷詞典裏的所有單詞,檢查你給我的字符串是不是這個單詞的前綴不就是了?”
小Hi笑道:“你啊,還是太年輕了!~假設這本詞典裏有10萬個單詞,我詢問你一萬次,你得要算到哪年哪月去?”
小Ho低頭算了一算,看着那一堆堆的0,頓時感覺自己這輩子都要花在上面了...
小Hi看着小Ho的囧樣,也是繼續笑道:“讓我來提高一下你的知識水平吧~你知道樹這樣一種數據結構麼?”
小Ho想了想,說道:“知道~它是一種基礎的數據結構,就像這裏說的一樣!”
小Hi滿意的點了點頭,說道:“那你知道我怎麼樣用一棵樹來表示整個詞典麼?”
小Ho搖搖頭表示自己不清楚。
“你看,我們現在得到了這樣一棵樹,那麼你看,如果我給你一個字符串ap,你要怎麼找到所有以ap開頭的單詞呢?”小Hi又開始考校小Ho。
“唔...一個個遍歷所有的單詞?”小Ho還是不忘自己最開始提出來的算法。
“笨!這棵樹難道就白構建了!”小Hi教訓完小Ho,繼續道:“看好了!”
“那麼現在!趕緊去用代碼實現吧!”小Hi如是說道
輸入
輸入的第一行爲一個正整數n,表示詞典的大小,其後n行,每一行一個單詞(不保證是英文單詞,也有可能是火星文單詞哦),單詞由不超過10個的小寫英文字母組成,可能存在相同的單詞,此時應將其視作不同的單詞。接下來的一行爲一個正整數m,表示小Hi詢問的次數,其後m行,每一行一個字符串,該字符串由不超過10個的小寫英文字母組成,表示小Hi的一個詢問。
在20%的數據中n, m<=10,詞典的字母表大小<=2.
在60%的數據中n, m<=1000,詞典的字母表大小<=5.
在100%的數據中n, m<=100000,詞典的字母表大小<=26.
本題按通過的數據量排名哦~
輸出
對於小Hi的每一個詢問,輸出一個整數Ans,表示詞典中以小Hi給出的字符串爲前綴的單詞的個數。
5 babaab babbbaaaa abba aaaaabaa babaababb 5 babb baabaaa bab bb bbabbaab樣例輸出
1 0 3 0 0
//建立trie書即可,這是紅書的代碼
#include<vector>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<string>
#include<cstdio>
#include<iostream>
using namespace std;
const int CHARSET=26,BASE='a',MAX_NODE=100005*3;
struct Trie
{
int tot,root,child[MAX_NODE][CHARSET],num[MAX_NODE];
bool flag[MAX_NODE];
Trie(){
memset(child[1],0,sizeof(child[1]));
memset(num,0,sizeof(num));
flag[1]=false;
root=tot=1;
}
void insertT(char *str){
int *cur=&root;
num[*cur]++;
for(char *p=str;*p;++p){
cur=&child[*cur][*p-BASE];
if(*cur==0){
*cur=++tot;
memset(child[tot],0,sizeof(child[tot]));
flag[tot]=false;
}
num[*cur]++;
}
flag[*cur]=true;
}
int queryT(char *s){
int *cur=&root;
for(char *p=s;*p&&*cur;++p){
cur=&child[*cur][*p-BASE];
}
if(!cur) return 0;
else return num[*cur];
}
}*trie;
int main()
{
int n,m;char s[12];
while(~scanf("%d",&n))
{
//memset(trie.num,0,sizeof())
trie = new Trie();
while(n--){
scanf("%s",s);
trie->insertT(s);
}
scanf("%d",&m);
while(m--){
scanf("%s",s);
printf("%d\n",trie->queryT(s));
}
}
return 0;
}
//這是大神的代碼,看上去更清晰,值得借鑑。原鏈接找不到了,還望見諒
#include<vector>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<string>
#include<cstdio>
#include<iostream>
using namespace std;
struct trie
{
trie *ch[26];
int num;
trie(){
num=0;
for(int i=0;i<26;i++) ch[i]=NULL;
}
}*root;
void insertTrie(char s[])
{
trie *p=root;
p->num++;
for(int i=0;s[i];++i){
int j=s[i]-'a';
if(p->ch[j]==NULL) p->ch[j]=new trie;
p=p->ch[j];
p->num++;
}
}
int searchTrie(char s[])
{
trie *p=root;
for(int i=0;p&&s[i];++i){
int j=s[i]-'a';
p=p->ch[j];
}
if(!p) return 0;
else return p->num;
}
int main()
{
int n,m;char s[12];
while(~scanf("%d",&n))
{
root = new trie;
while(n--){
scanf("%s",s);
insertTrie(s);
}
scanf("%d",&m);
while(m--){
scanf("%s",s);
printf("%d\n",searchTrie(s));
}
}
return 0;
}