POJ 1451 T9（字典树+bfs）

背景：为了方便九宫格手机用户发短信，希望在用户按键时，根据提供的字典（给出字符串和频数），给出各个阶段最有可能要打的单词。

题意：

首先给出的是字典，每个单词有一个出现频率。然后给出的是询问，每个询问有一个数字字符串，代表在手机上按了哪些键，以1结束。问按键的过程中最可能出现的单词分别是哪些。

注意：hell的权值为hell hello hellmoun 的权值和。
即一个字符串前缀的概率等于所有前缀之和。
还有就是一个单词在输入法中，这个单词的所有前缀也在输入法中。

解析：

字典树来搞这道题目，可以把所有的字符串都插入字典树中，然后bfs这颗字典树，找到当前层权最大的节点，并输出该路径。

my code

#include 
#include 
#include 
#include 
#include 
using namespace std;
const int maxnode = 1005 * 105;
const int sigma_size = 26;
string T9[10];

int ch[maxnode][sigma_size];
int fa[maxnode];
char alpha[maxnode];
int val[maxnode];


int idx(int c) { return c - 'a'; }

struct Trie {
    int sz;
    void clear() { sz = 1; memset(ch[0], 0, sizeof(ch[0])); }
    Trie() { clear(); }

    void insert(char *s, int p) {
        int u = 0, n = strlen(s);
        for(int i = 0; i < n; i++) {
            int c = idx(s[i]);
            if(!ch[u][c]) {
                memset(ch[sz], 0, sizeof(ch[sz]));
                val[sz] = 0;
                fa[sz] = u;
                alpha[sz] = c + 'a';
                ch[u][c] = sz++;
            }
            u = ch[u][c];
            val[u] += p;
        }
    }
} trie;

void init() {
    T9[0] = ""; T9[1] = "";
    T9[2] = "abc"; T9[3] = "def";
    T9[4] = "ghi"; T9[5] = "jkl";
    T9[6] = "mno"; T9[7] = "pqrs";
    T9[8] = "tuv"; T9[9] = "wxyz";
}

void dfs(int u) {
    if(u == 0) return ;
    dfs(fa[u]);
    putchar(alpha[u]);
}

struct Node {
    int u, step;
    Node(int u, int step) : u(u), step(step) {}
};

int bfs(char* s, int len) {
    queue que;
    que.push(Node(0, -1));

    int maxp = -1, ret = -1;
    int u, step;
    while(!que.empty()) {
        Node front = que.front();
        que.pop();

        u = front.u, step = front.step;
        if(step > len) continue;
        if(step == len && val[u] > maxp) {
            maxp = val[u];
            ret = u;
        }

        int num = s[step+1] - '0';
        for(int i = 0; i < T9[num].size(); i++) {
            int c = idx(T9[num][i]);
            if(ch[u][c]) {
                int v = ch[u][c];
                que.push(Node(v, step+1));
            }
        }
    }
    return ret;
}

int n, m, p;
char str[105];

int main() {
    init();
    int T, cas = 1;
    scanf("%d", &T);
    while(T--) {
        trie.clear();
        scanf("%d", &n);
        while(n--) {
            scanf("%s%d", str, &p);
            trie.insert(str, p);
        }
        scanf("%d", &m);
        printf("Scenario #%d:\n", cas++);
        while(m--) {
            scanf("%s", str);
            int len = strlen(str);
            for(int i = 0; str[i] != '1'; i++) {
                int u = bfs(str, i);
                if(u == -1) {
                    puts("MANUALLY");
                }else {
                    dfs(u);
                    puts("");
                }
            }puts("");
        }puts("");
    }
    return 0;
}