題目:
時間限制: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 <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int MAX = 26;//小寫英文字母;
struct Trie {
Trie *next[MAX];//每層有多少種類的數;
int v;//一個字典樹到此為止有多少相同前綴的數目;
char ch;//當前的字母;
};
Trie *root;
void createTrie(string s) {//生成字典樹;
int len = s.length();//該字符串的長度;
Trie *p = root, *q;//每次從根節點開始一次搜索;
for (int i = 0;i < len;++i) {//取得單詞的每個字母;
int id = s[i] - 'a';//不同的小寫字母對應不同的下標;
if (p->next[id] == NULL) {//表明該坐標下沒有子樹, 之前沒有以它為準的前綴;
q = new Trie();//創建一個新節點;
q->v = 1;//初始v==1;
for (int j = 0;j < MAX;++j) {
q->next[j] = NULL;//該節點之后暫時沒有分支;
}
p->next[id] = q;
p->ch = s[i];
p = p->next[id];//移到下一個分支;
}
else {
p->ch = s[i];
p->next[id]->v++;//到目前位置有相同前綴;
p = p->next[id];
}
}
}
int findTrie(string s) {
int len = s.length();
Trie *p = root;
for (int i = 0;i < len;++i) {
int id = s[i] - 'a';
int cnt = p->v;
char ch = p->ch;
p = p->next[id];
if (p == NULL) {//為空集, 表示不存在以此為前綴的串;
if (i != len-1) return 0;
else if (ch != s[i]) {
return 0;
}
else return cnt;
}
}
return p->v;
}
int main() {
ios::sync_with_stdio(false);
cin.tie(NULL);
int n;
cin >> n;
root = new Trie();
//////初始化;//////////
for (int i = 0;i < MAX;++i) {
//cout << "xyz" << endl;
root->next[i] = NULL;//初始情況;
}
root->v = 0;
//////////////////////
for (int i = 1;i <= n;++i) {
string s;
cin >> s;
//cout << s << endl;
createTrie(s);
}
int m;
cin >> m;
for (int i = 1;i <= m;++i) {
string s;
cin >> s;
int ans = findTrie(s);
cout << ans << endl;
}
delete root;
return 0;
}
這道題就是用字典樹解決,但是在處理時需要注意:
這道題是統計以該查找項為前綴的單詞的個數,注意有可能出現字典樹中有單詞是該查找項的前綴。
注意一下樣例:
4
pp
ppppp
p
ppp
4
p
pp
pppp
ppppp