hihocoder trie 樹

ACM_e發表於2017-12-26

#1014 : Trie樹

時間限制: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




#include<bits/stdc++.h>
using namespace std;
struct ac{
   int a[27]={0},n;
}a[1000001];
int t=0;
int main(){
    int n;
    cin>>n;
    while(n--){
      string s;
      cin>>s;
      int len=s.size();
      int i=0;
      for(int j=0;j<len;j++){
          int k=s[j]-'a';
          if(a[i].a[k]){
             i=a[i].a[k];
             a[i].n++;
          }
          else{
             a[i].a[k]=++t;
             i=t;
             a[i].n=1;
          }
      }
    }
    int m;
    cin>>m;
    while(m--){
       string s;
       cin>>s;
       int len=s.size();
       int i=0,sum=0;
       for(int j=0;j<len;j++){
          int k=s[j]-'a';
          if(a[i].a[k]){
             i=a[i].a[k];
            // cout<<k<<" "<<i<<" ";
          }
          else break;
          if(j==len-1){
            sum=a[i].n;
          }
       }
       cout<<sum<<endl;
    }

    return 0;
}
/*
babb
1 1 0 2 1 3 1 7 1
baabaaa
1 1 0 2 0 15 0
bab
1 1 0 2 1 3 4
bb
1 1 1 14 1
bbabbaab
1 1 1 14 0
*/


相關文章