POJ 1451 字典樹
阿新 • • 發佈:2019-01-29
這題的意思有點類似於一個輸入法,就是按了一些鍵,蹦出單詞庫中頻率最高的那個詞彙
首先給出的是單詞庫,每個單詞有一個出現頻率。
然後給出的是一些詢問,每個詢問有一個字串,代表在手機上按了哪些鍵,最後以1結束。問進行這些按鍵的過程中出現的單詞分別是哪些。
思路就是字典樹了。
以手機的8個鍵做8個指標來建立字典樹。
每個結點存的除了指標外,還有出現頻率最高的單詞。
建樹的時候需要注意,每個相同的字首只能插入一次,否則就沒法處理。
所以插入之前,把所有相同字首的頻率值都加起來 ,插入的時候插一次就好了,每次更新最優值就行。
題目給的輸入是已經排好序的,也就省了我們排序的時間。因為字首相同的肯定在一塊,所以挨個同相鄰的比較就行了、
#include <iostream> #include <algorithm> #include <cstring> #include <string> #include <cstdio> #include <cmath> #include <queue> #include <map> #include <set> #define eps 1e-5 #define MAXN 1111 #define MAXM 55555 #define INF 1000000000 using namespace std; int c[1005][105], e; char s[1005][105]; char tmp[105]; char dic[] = "22233344455566677778889999"; struct Trie { int next[8]; int num; char s[105]; void init () { memset(next, 0, sizeof(next)); num = 0; } } trie[MAXM]; void make_trie (char *str, int id) { int i = 0, index, u = 0; while(str[i]) { index = dic[str[i] - 'a'] - '2'; if(!trie[u].next[index]) trie[u].next[index] = ++e; u = trie[u].next[index]; if(c[id][i] > trie[u].num) { trie[u].num = c[id][i]; for(int j = 0; j <= i; j++) trie[u].s[j] = s[id][j]; trie[u].s[i + 1] = '\0'; } i++; } } void match(char *str) { int i = 0, index, u = 0, flag = 0;; while(str[i] && str[i] != '1') { index = str[i] - '2'; u = trie[u].next[index]; if(trie[u].num == 0 || u == 0 || flag) printf("MANUALLY\n"), flag = 1; else printf("%s\n", trie[u].s); i++; } } int main() { int T, n, p, m, cas = 0; scanf("%d", &T); while(T--) { scanf("%d", &n); memset(c, 0, sizeof(c)); for(int i = 0; i < n; i++) { scanf("%s%d", s[i], &p); int len = strlen(s[i]); for(int j = 0; j < len; j++) c[i][j] = p; } for(int i = 1; i < n; i++) for(int j = 0; s[i][j] && s[i - 1][j]; j++) { if(s[i][j] == s[i - 1][j]) { c[i][j] += c[i - 1][j]; c[i - 1][j] = 0; } else break; } for(int i = 0; i < MAXM; i++) trie[i].init(); e = 0; for(int i = 0; i < n; i++) make_trie(s[i], i); scanf("%d", &m); printf("Scenario #%d:\n", ++cas); while(m--) { scanf("%s", tmp); match(tmp); printf("\n"); } printf("\n"); } return 0; }