hdu 1298 字段树+DFS

#include <iostream>
#include <stdio.h>
#include <cstring>
#include <string>
#include<malloc.h>
#include<stdlib.h>
using namespace std;
#define M 105
char s2[M],ans[M],end[M]; //单词
int p;
//模拟按键，比如说0对应2按键，2有三种可能0,1,2（a,b,c）
int 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 re[8]={3,3,3,3,3,4,3,4};
struct Trie  //字典树结构
{
    Trie *child[26];
    int num;   //频率
    Trie()
    {
        num=0;
        memset(child,0,sizeof(child)); //初始化~
    }
};
Trie *root,*s;
void Create(char *str,int k)  //构造字典树，k为频率
{
    s=root;
    int i=0;
    while(str[i])
    {
        int id=str[i]-'a';
        if(s->child[id]==0)
        {
        s->child[id]=new Trie;
        s=s->child[id];
        s->num+=k;
        }
        else
        {
            s=s->child[id];
            (s->num)+=k;
        }
        i++;
    }
}
/*void _free(Trie *a)
{
    for(int i=0;i<26;++i)
    {
        if(a->child[i])
        _free(a->child[i]);
    }
    free(a);
}*/
void dfs(Trie *you,int now,int len)  //深搜~
{
    if(now==len)  //递归出口
    {
        //p记录当前最大的频率，如果遇到比它还大的，更新end[i]
        if(p<you->num)
        {
            p=you->num;
            for(int i=0;i<len;++i)
            end[i]=ans[i];
            end[len]=0;
        }
        return;
    }
    int t=s2[now]-'2';
    for(int i=0;i<re[t];i++)
    {
        int call=phone[t][i];
        if(you->child[call]!=0)
        {
            ans[now]='a'+call;  //当前的字符
            dfs(you->child[call],now+1,len); //递归寻找下一个字符
        }
    }
   return;
}
int main()
{
    int t,n,m,y,len;
    scanf("%d",&t);
    for(int tt=1;tt<=t;tt++)
    {
        printf("Scenario #%d:\n",tt);
        scanf("%d",&n);
        root=new Trie;
        while(n--)  //构造字典树
        {
            scanf("%s %d",s2,&y);
            Create(s2,y);
        }
        scanf("%d",&m);
        while(m--)
        {
            scanf("%s",s2);
            len=strlen(s2);


            /*每种长度递归寻找符合要求的联想字符串,其实还有个
            可以优化的方法，当第一次深搜后p等于0后，以后出现
            的都是MANUALLY了
            */
            for(int i=1;i<len;i++)  //题目说了最后的字符1是无效的，不需要递归
            {
                p=0;
                dfs(root,0,i);
                if(p>0) printf("%s\n",end);
                else puts("MANUALLY");
            }
            if(m!=0) puts("");//每两个单词间空一行
        }
        printf("\n\n");  //每个案例后空两行
    }
    //_free(root);
    //_free(s);
    return 0;
}