[字典树 + dfs] HDU 1298 - T9

题目想了好久，发现需要在查询字典树的时候，枚举每个数字上的字母。就没敢去想用dfs。

但是后来经过计算发现，w最大值是1000，每个数字串长度最大为100。

w的最大值也就是dfs枚举的组合数量。。

先建字典树，题目说的是每个字符的权值是出现过的概率和。

那么插入操作的时候直接更新就可以了。

插入完毕后，用dfs查询即可。。

#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stdlib.h>
#include <algorithm>
#include <iostream>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <assert.h>
#include <malloc.h>
#include <time.h>
typedef long long LL;
const int INF = 500000001;
const double EPS = 1e-9;
const double PI = acos(-1.0);
using namespace std;
char cc[10][10]={"", "abc", "def", "ghi", "jkl", "mno", "pqrs", "tuv", "wxyz"};
int ccflg[101];
int chlen;
char tmp[101];
char temp[101][101];
char ch[101];
struct Trie
{
    Trie *next[26];
    int rk;
};
Trie *H;
Trie *newnode()
{
    Trie *p = new Trie;
    for(int i = 0; i < 26; i++)
    {
        p->next[i] = NULL;
    }
    p->rk = 0;
}
int idx(char c)
{
    return c - 'a';
}
void insert(char *ss, int k)
{
    Trie *p = H;
    int len = strlen(ss);
    for(int i = 0; i < len; ++i)
    {
        int c = idx(ss[i]);
        if(p->next[c] == NULL)
        {
            p->next[c] = newnode();
        }
        p = p->next[c];
        p->rk += k;
    }
}
void dfs(int deep, Trie *p)
{
    if(deep < chlen)
    {
        int cclen = strlen(cc[ch[deep]-'0'-1]);
        for(int j = 0; j < cclen; ++j)
        {
            int c = idx(cc[ch[deep]-'0'-1][j]);

            if(p->next[c] == NULL)
            {
                continue;
            }
            tmp[deep] = c + 'a';
            dfs(deep+1, p->next[c]);
            if(p->next[c]->rk > ccflg[deep])
            {
                tmp[deep+1] = '\0';
                strcpy(temp[deep], tmp);
                ccflg[deep] = p->next[c]->rk;
            }
        }
    }
}
int main()
{
    //freopen("test0.in", "r", stdin);
    //freopen("test0.out", "w", stdout);
    //srand(time(NULL));
    int T, w, k, p;
    scanf("%d", &T);
    for(int ii = 1; ii <= T; ii++)
    {
        printf("Scenario #%d:\n", ii);
        H = newnode();
        scanf("%d", &w);
        for(int i = 0; i < w; ++i)
        {
            scanf("%s %d", ch, &k);
            insert(ch, k);
        }
        scanf("%d", &p);
        for(int j = 0; j < p; ++j)
        {
            memset(ccflg, 0, sizeof(ccflg));
            scanf("%s", ch);
            chlen = strlen(ch) - 1;
            dfs(0, H);
            for(int i = 0; i < chlen; i++)
            {
                if(ccflg[i])
                {
                    printf("%s\n", temp[i]);
                }
                else
                {
                    printf("MANUALLY\n");
                }
            }
            puts("");
        }
        puts("");
    }
    return 0;
}