HDU 1251 統計難題 Trie題解

標籤：使用   io   for   問題   ar   amp   size   c   

基本上是標準的尋找首碼的問題，只需要insert和search函數就可以了。

我這裡主要是修改一下n的記錄方法，這裡的n代表的不是葉子節點的標誌，而是有多少單詞經過了這條路徑的標誌。

然後是尋找需要尋找的首碼單詞，如果沒有找到，就返回0，表示沒有單詞以這個首碼單詞為首碼，如果找到，直接返回n就是答案了。因為有n個單詞經過了這條路徑。

尋找效率是常數。

使用靜態分配空間的辦法。

#include <stdio.h>#include <string.h>const int MAX_N = 500000;const int ARR_SIZE = 26;const int MAX_WORD = 11;struct Node{int n;Node *arr[ARR_SIZE];};Node pool[MAX_N];int poolId;void clearNode(Node *p){p->n = 0;for (int i = 0; i < ARR_SIZE; i++){p->arr[i] = NULL;}}void insert(Node *trie, char w[]){int len = strlen(w);for (int i = 0; i < len; i++){int j = w[i] - 'a';if (trie->arr[j] == NULL){trie->arr[j] = &pool[poolId++];clearNode(trie->arr[j]);}trie = trie->arr[j];trie->n++;//修改成可以快速尋找首碼}}int searchPre(Node *trie, char w[]){int len = strlen(w);for (int i = 0; i < len; i++){int j = w[i] - 'a';if (!trie->arr[j]) return 0;trie = trie->arr[j];}return trie->n;}int main(){Node *trie = &pool[0];clearNode(trie);poolId = 1;char word[MAX_WORD];while (gets(word) && word[0] != '\0'){insert(trie, word);}while (gets(word)){printf("%d\n", searchPre(trie, word));}return 0;}