hdoj 4287 Intelligent IME 字典树

2012天津赛区网络赛的题，当年被一个大一的学弟A了，今天重温数据结构，做了一下，水题，没多大意义！

不能用数字来匹配单词，那样复杂度太大，反过来比较简单！

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

struct node
{
    int m;
    node *next[10];
    int su;
    node()
    {
        m=0;
        su=0;
        for(int i=0;i<10;i++)
            next[i]=NULL;
    }
};

int map[26];
node *root;
int n,m;
char num[5006][7];

//这个地方是初始化map
void init()
{
    int cs=0;
    for(int i=2;i<=9;i++)
    {
        if(i==7||i==9)
        {
            for(int j=0;j<4;j++)
                map[cs++]=i;
        }
        else
        {
            for(int j=0;j<3;j++)
                map[cs++]=i;
        }
    }
}

void insert_tree(char *s)
{
    int id;
    int len=strlen(s);
    node *p,*q;
    if(root==NULL)
    {
        root = new node;
    }
    p=root;
    for(int i=0;i<len;i++)
    {
        id=s[i]-'0';
        if(p->next[id]==NULL)
        {
            q=new node;
            p->next[id]=q;
            p=p->next[id];
        }
        else
        {
            p=p->next[id];
        }
        if(i==len-1)
        {
            p->m=1;
        }
    }
}

//检查是否存在
void check(char *s)
{
    int id;
    int len=strlen(s);
    node *p,*q;
    p=root;
    for(int i=0;i<len;i++)
    {
        id=map[s[i]-'a'];
        if(p->next[id]==NULL)
            return ;
        else
            p=p->next[id];
        if(i==len-1)
        {
            if(p->m==1)
                p->su++;
        }
    }
    return ;
}

int search_tree(char *s)
{
    int id,len;
    node *p;
    p=root;
    len=strlen(s);
    for(int i=0;i<len;i++)
    {
        id=s[i]-'0';
        p=p->next[id];
        if(i==len-1)
        {
            return p->su;
        }
    }
}

void delete_tree(node *s)
{
    for(int i=0;i<10;i++)
    {
        if(s->next[i]!=NULL)
        {
            delete_tree(s->next[i]);
        }
    }
    delete s;
}

int main()
{
    init();
    int t;
    scanf("%d",&t);
    char s[10];
    int x,y;
    while(t--)
    {
        root=NULL;
        memset(num,0,sizeof(num));
        scanf("%d%d%*c",&n,&m);
        x=n,y=m;
        while(n--)
        {
            scanf("%s%*c",s);
            strcpy(num[n],s);
            insert_tree(s);
        }
        while(m--)
        {
            scanf("%s%*c",s);
            check(s);
        }
        for(int i=x-1;i>=0;i--)
        {
            printf("%d\n",search_tree(num[i]));
        }
        delete_tree(root);
    }
}