HDU——1298 T9

/*
题目意思：
相信大家都知道搜狗拼音的词语联想功能。这题的大致意思就
是根据你的字典里的词出现的频率，把词语联想的功能模拟一
遍。每多一个按键，你只需输出第一联想到的东西，如果没有，
就输出MANULLY。一个按键对应一个字母，但是一个按键有3或
者4个选择的字母。注意：在你查询过程中，是不会额外增加单
词的。
这题真的花了我不少时间~T T，过去用的字典树模板实在是
通用性太差劲了。终于给我百度到一个挺好的模板。字典树的
最大复杂度为Q(n),n代表字符串的长度。我真心没有想到这题
可以用字典树+深搜做~，思路也是百度来的。经过抄袭了N个大
神的模板和学习后，整理如下：
AC状态：0ms 660KB
*/
#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;
}