1. 程式人生 > >hihocoder--hiho一下第二周(Trie樹)

hihocoder--hiho一下第二周(Trie樹)

英文 -h 字符 modal data- clas 接下來 小寫 nbsp

時間限制:10000ms 單點時限:1000ms 內存限制:256MB

描述

小Hi和小Ho是一對好朋友,出生在信息化社會的他們對編程產生了莫大的興趣,他們約定好互相幫助,在編程的學習道路上一同前進。

這一天,他們遇到了一本詞典,於是小Hi就向小Ho提出了那個經典的問題:“小Ho,你能不能對於每一個我給出的字符串,都在這個詞典裏面找到以這個字符串開頭的所有單詞呢?”

身經百戰的小Ho答道:“怎麽會不能呢!你每給我一個字符串,我就依次遍歷詞典裏的所有單詞,檢查你給我的字符串是不是這個單詞的前綴不就是了?”

小Hi笑道:“你啊,還是太年輕了!~假設這本詞典裏有10萬個單詞,我詢問你一萬次,你得要算到哪年哪月去?”

小Ho低頭算了一算,看著那一堆堆的0,頓時感覺自己這輩子都要花在上面了...

小Hi看著小Ho的囧樣,也是繼續笑道:“讓我來提高一下你的知識水平吧~你知道樹這樣一種數據結構麽?”

小Ho想了想,說道:“知道~它是一種基礎的數據結構,就像這裏說的一樣!”

小Hi滿意的點了點頭,說道:“那你知道我怎麽樣用一棵樹來表示整個詞典麽?”

小Ho搖搖頭表示自己不清楚。

提示一:Trie樹的建立

“你看,我們現在得到了這樣一棵樹,那麽你看,如果我給你一個字符串ap,你要怎麽找到所有以ap開頭的單詞呢?”小Hi又開始考校小Ho。

“唔...一個個遍歷所有的單詞?”小Ho還是不忘自己最開始提出來的算法。

“笨!這棵樹難道就白構建了!”小Hi教訓完小Ho,繼續道:“看好了!”

提示二:如何使用Trie樹

提示三:在建立Trie樹時同時進行統計!

“那麽現在!趕緊去用代碼實現吧!”小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

hint:之前記得自己做過一個字典樹的題目。。。好像又給忘記了orz 腦子不夠用(其實是之前沒理解它。。。)
好像大家的字典樹都用結構體實現的,結構體中一個記錄元素--用來記錄以該單詞為前綴的單詞個數(這個可以在初始化的時候就得到,字典樹中的節點標號就是數組下標,用next數組來指向它)
 1 #include<cstdio>
 2 #include<cstring>
 3 #include<cstdlib>
 4 #include<algorithm>
 5 #include<iostream>
 6 using namespace std;
 7 const int MAXN = 100005;
 8 char s[MAXN], q[MAXN];
 9 struct node{
10     int cnt;
11     int next[30];
12     void init(){
13         cnt=0;
14         memset(next, -1, sizeof(next));
15     }
16 }T[MAXN*26];
17 int id=1;
18 void build_tree(char *s){
19     int p=0, i=0;
20     while(s[i]){
21         int x=s[i]-a;
22         if(T[p].next[x]==-1){
23             T[id].init();
24             T[p].next[x]=id++;
25         }
26         p=T[p].next[x];
27         T[p].cnt++;
28         i++;
29     }
30 }
31 int query(char *q){
32     int i=0, p=0;
33     while(q[i]){
34         int x=q[i]-a;
35         if(T[p].next[x]==-1){
36             return 0;
37         }
38         p=T[p].next[x];
39         i++;
40     }
41     return T[p].cnt;
42 }
43 int main(){
44     int n, m;
45     scanf("%d", &n);
46     T[0].init();
47     while(n--){
48         scanf("%s", s);
49         build_tree(s);
50     }
51     scanf("%d", &m);
52     while(m--){
53         scanf("%s", q);
54         printf("%d\n", query(q));
55     }
56     return 0;
57 }

hihocoder--hiho一下第二周(Trie樹)