字典树+DFS T9 HDU 1298

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
int c;
char a[105],b[105];
int fr[10],ed[10];
int flag;
int max;
struct trie
{
trie *next[30];
char s;
int level;
};
trie memory[100010];
void build(char *str,int l)                          //建立字典树
{
int len=strlen(str);
trie *p=&memory[0];
for(int i=0;i<len;i++)
{
int id=str[i]-'a';
if(p->next[id]==NULL)
p->next[id]=&memory[++c];
p=p->next[id];
p->s=str[i];
p->level+=l;                             //每个字母出现的频率为其加和
}
}
void dfs(char *str,int n,trie *p)
{
if(c==n)
{
a[c]='\0';
if(p->level>max)                    //找到的单词如果出现频率最高，更新b数组
{
max=p->level;
strcpy(b,a);
flag=1;                           //表示已经找到了单词
}
return ;
}
int id=str[c]-'0';
for(int i=fr[id];i<=ed[id];i++)
{
if(p->next[i]!=NULL)
{
trie *q;
q=new trie;
q=p->next[i];
a[c++]=q->s;                //存下找到的字母，方便更新b数组
dfs(str,n,q);
c--;
}
}
return ;
}
int main()
{
int t,i;
int w,j;
char str[105];
int level;
int p;
char num[105];
for(i=2;i<7;i++)
{
fr[i]=(i-2)*3;
ed[i]=(i-1)*3-1;
}
fr[7]=15;ed[7]=18;fr[8]=19;ed[8]=21;fr[9]=22;ed[9]=25;
while(scanf("%d",&t)!=-1)
{
for(i=1;i<=t;i++)
{
memset(memory,0,sizeof(memory));
printf("Scenario #%d:\n",i);
scanf("%d",&w);
c=0;
for(j=0;j<w;j++)
{
scanf("%s%d",str,&level);
build(str,level);
}
scanf("%d",&p);
for(j=0;j<p;j++)
{
scanf("%s",num);
int len=strlen(num);
for(int k=1;k<len;k++)
{
flag=0;c=0;max=0;
memset(a,0,sizeof(a));
dfs(num,k,&memory[0]);
if(flag==1)
printf("%s\n",b);
else
printf("MANUALLY\n");
}
printf("\n");
}
printf("\n");                     //两组数之间需输出空行
}
}
return 0;
}