HDU 1298 T9 字典树+dfs

题目地址： http://acm.hdu.edu.cn/showproblem.php?pid=1298

这道题做了一天了，调试各种错误，最后看大牛的代码对照才找出错误。

哎~~~~~， 技术不行啊~~~~~

题意：

给出几个单词频数，然后给出一个几个数字串，确定对于给定的一个数字串，每输入一个数字，将显示什么字符。

多个数字可能代表一个字母，首先可将给出的单词即频度记录到字典树中，对于数字串进行DFS，查找其可能表示的频度最大的字符串。

代码如下：

#include <iostream>
#include <vector>
#include <List>
#include <deque>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <algorithm>
#include <cctype>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <cmath>
using namespace std;

const int N=101;
typedef long long LL;

char xh[N],cnt[N],ans[N];
char phone[8][4]={{0,1,2},{3,4,5},{6,7,8},{9,10,11},{12,13,14},{15,16,17,18},{19,20,21},{22,23,24,25}};
int  num[8]={3,3,3,3,3,4,3,4};
int flag;//记录字符出现的频度

struct Trie
{
    Trie *next[N];
    int v;
    Trie()
    {
        v=0;
        for(int i=0;i<N;i++)
            next[i]=NULL;
    }
}*root;

void createTrie(char *str,int k)
{
    int i,id,len;
    len=strlen(str);
    Trie *p=root;
    for(i=0;i<len;i++)
    {
        id=str[i]-'a';
        if(p->next[id]==NULL)
        {
            p->next[id]=new Trie();
            p=p->next[id];
            p->v+=k;
        }
        else
        {
            p=p->next[id];
            p->v+=k;
        }
    }
}

int findTrie(char *str)
{
    int i,id,j,len;
    len=strlen(str);
    Trie *p=root;
    for(i=0;i<len;i++)
    {
        id=str[i]-'a';
        p=p->next[id];
        if(p==NULL)
            return 0;
    }
    return p->v;
}

void dealTrie(Trie *T)
{
    if(T==NULL)
        return ;
    for(int i=0;i<N;i++)
        if(T->next[i]!=NULL)
            dealTrie(T->next[i]);
    free(T);
}

void dfs(int l,int r,Trie *Tr)
{
    int i,j,t;
    if(l==r)
    {
        if(Tr->v > flag)
        {
            flag=Tr->v;
            for(i=0;i<r;i++)
                cnt[i]=ans[i];
            cnt[i]='\0';
        }
        return ;
    }
    t=xh[l]-'2';
    for(i=0;i<num[t];i++)
    {
        int k=phone[t][i];
        if(Tr->next[k]==NULL)
            continue;
        ans[l]='a'+k;
        dfs(l+1,r,Tr->next[k]);
    }
}

int main()
{
    int i,j,T,casei=0;
    cin>>T;
    while(T--)
    {
        printf("Scenario #%d:\n",++casei);
        int m,t;
        cin>>m;
        root=new Trie();
        while(m--)
        {
            scanf("%s%d",xh,&t);
            createTrie(xh,t);
        }
        cin>>m;
        while(m--)
        {
            scanf("%s",xh);
            int len=strlen(xh);
            for(i=1;i<len;i++)
            {
                flag=0;
                dfs(0,i,root);//寻找最大频率数
                if(flag>0)
                    printf("%s\n",cnt);
                else
                    printf("MANUALLY\n");
            }
            printf("\n");
        }
        printf("\n");
        dealTrie(root);//这个地方如果有多个trie树的话，最好要释放。
    }
    return 0;
}