POJ 1451 T9

 Trie树

POJ 1451 T9

题意：模拟用9个键的手机进行输入的，每输一个字母，输出最可能匹配的单词。

一开始想用链表储存所有的单词，写的时候才发现连链表维护，实在太麻烦了，而且效率又不高。

对于固定的输入，只用输出一个最可能匹配的词，所以不用保存所有的单词，只用保存频率最高的词就可以了。

两个单词的前缀可能重叠，所以要预处理，使任何单词或前缀只插入一次。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define MAXN 1010
using namespace std;

struct Word
{
    char s[111];
    int t;
};

struct Trie
{
    Trie *num[8];
    Word wo;
};

Trie root;
Word w[1010];
int c[1010][103];
char key[]="22233344455566677778889999";

void Add(Trie *node, char *s, int i, int j)
{
    if(j>0&&c[i][j-1])
    {
        s[j]='\0';
        if(node->wo.t<c[i][j-1])
        {
            strcpy(node->wo.s,s);
            node->wo.t=c[i][j-1];
        }
    }
    s[j]=w[i].s[j];
    if(s[j]=='\0')
        return ;
        
    int d,k=key[w[i].s[j]-'a']-'2';
    if(!node->num[k])
    {
        node->num[k] = new Trie[8];
            for(d=0;d<8;d++)
        node->num[k]->num[d]=NULL;
        node->num[k]->wo.t=0;
    }

    Add(node->num[k],s,i,j+1);
}

void Construct()
{
    int n,i,k,j;
    char s[103];
    memset(c,0,sizeof(c));
    for(i=0;i<8;i++)
        root.num[i]=NULL;
    root.wo.t=0;
    
    scanf("%d",&n);
    for(i=0;i<n;i++)
        scanf("%s%d",w[i].s,&w[i].t);
    for(i=0;i<n;i++)
        for(j=0,k=(int)strlen(w[i].s);j<k;j++)
            c[i][j]=w[i].t;
    for(i=n-2;i>=0;i--)
    {
        for(j=0;w[i].s[j]!='\0';j++)
            if(w[i].s[j]==w[i+1].s[j])
            {
                c[i][j]+=c[i+1][j];
                c[i+1][j]=0;
            }
            else
                break;
    }
    for(i=0;i<n;i++)
        Add(&root,s,i,0);
}

void Query(Trie *node, char *s, int j)
{
    if(node==NULL||node->wo.t==0)
    {
        for(;s[j]!='1';j++)
            puts("MANUALLY");
        return;
    }
    puts(node->wo.s);
    j++;
    if(s[j]=='1')
        return ;
    Query(node->num[s[j]-'2'],s,j);
}

void Solve()
{
    int m,i,j;
    char s[201];
    scanf("%d",&m);
    while(m--)
    {
        scanf("%s",s);
        if(s[0]!='1')
            Query(root.num[s[0]-'2'],s,0);
        puts("");
    }
}

void Trans(Trie *node)
{
    int i;
    for(i=0;i<8;i++)
        if(node->num[i])
        {
            Trans(node->num[i]);
            free(node->num[i]);
        }
}

void Destroy()
{
    int i;
    for(i=0;i<8;i++)
        if(root.num[i])
        {
            Trans(root.num[i]);
            free(root.num[i]);
        }
}

int main()
{
    int T,k;
    scanf("%d",&T);
    for(k=1;k<=T;k++)
    {
        Construct();
        printf("Scenario #%d:\n",k);
        Solve();
        puts("");
        Destroy();
    }
    return 0;
}