首页 > 代码库 > POJ 1451 T9 字典树+优先队列
POJ 1451 T9 字典树+优先队列
题目来源:POJ 1451 T9
题意:给你一些单词 和优先值 然后当你按下数字的时候首先会出现哪个单词 就是我们平时手机的按键
思路:建一颗字典树 因为按一个数字对应多个字母 那么就有多种情况 我们要输出权值最大的一个 我用了优先队列 这里每个前缀的优先值是所有单词优先值的和
例如abc 5 abd 6 acd 7 那么a这个前缀的优先值是18 ab的优先值是11
#include <cstdio> #include <cstring> #include <queue> using namespace std; const int maxnode = 100010; const int sigma_size = 26; const int maxn = 1010; int ch[maxnode][sigma_size]; int val[maxnode]; int sz; char s[maxn][110]; char id[12][6] = {"0", "1", "abc", "def", "ghi", "jkl", "mno", "pqrs", "tuv", "wxyz"}; void init() { sz = 1; memset(ch[0], 0, sizeof(ch[0])); } struct node { int u, v, x; char str[110]; node(){} node(int _u, int _v, int _x, char* s) { u = _u; v = _v; x = _x; strcpy(str, s); } bool operator < (const node& rhs) const { if(v != rhs.v) return v > rhs.v; return x < rhs.x; } }; int idx(char c) { if(c >= 'a' && c <= 'c') return 2; if(c >= 'd' && c <= 'f') return 3; if(c >= 'g' && c <= 'i') return 4; if(c >= 'j' && c <= 'l') return 5; if(c >= 'm' && c <= 'o') return 6; if(c >= 'p' && c >= 's') return 7; if(c >= 't' && c <= 'v') return 8; if(c >= 'w' && c <= 'z') return 9; } void insert(char *s, int v) { int u = 0, n = strlen(s); for(int i = 0; i < n; i++) { int c = s[i] - 'a'; if(!ch[u][c]) { memset(ch[sz], 0, sizeof(ch[sz])); val[sz] = 0; ch[u][c] = sz++; } u = ch[u][c]; val[u] += v; } //val[u] += v; } void find(char *s) { int u = 0, n = strlen(s); priority_queue <node> Q; Q.push(node(0, 0, 0, "")); for(int i = 0; i < n-1; i++) { int c = s[i]-'0'; while(!Q.empty()) { node x = Q.top(); if(x.v != i) break; Q.pop(); u = x.u; for(int j = 0; id[c][j]; j++) { int v = id[c][j] - 'a'; //printf("%d\n", v); if(!ch[u][v]) continue; v = ch[u][v]; x.str[i] = id[c][j]; x.str[i+1] = 0; //printf("%c\n", id[c][j]); Q.push(node(v, i+1, val[v], x.str)); } } if(Q.empty()) { puts("MANUALLY"); } else { node x = Q.top(); printf("%s\n", x.str); } } } int main() { int cas = 1; int T; scanf("%d", &T); while(T--) { int n, m; scanf("%d", &n); init(); for(int i = 0; i < n; i++) { //char s[maxn]; int x; scanf("%s %d", s[i], &x); insert(s[i], x); } printf("Scenario #%d:\n", cas++); scanf("%d", &m); while(m--) { char str[210]; scanf("%s", str); find(str); puts(""); } puts(""); } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。