HDU - 1075 What Are You Talking About(字典树)

点我看题

题意:给你一个英文火星文对应字典,让把火星文翻译为英文.

分析:用一棵字典树把字典建立起来,然后查询.

参考代码:

#include<cstdio>
#include<cmath>
#include<cstring>
#include<cctype>//islower(char ch)的头文件
#include<algorithm>
#include<iostream>

using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
const int maxn = 1e6+10;//树的结点,不要开太小
const int maxm = 3e3+10;//字符个数
//TireNode
int tn[maxn][26];//结点
int node;//结点个数
char w[maxn][12];//结点i代表的字符串
bool exist[maxn];//判断到当前为止是否存在从根到这里表示的火星文
char sen[maxm];//输入的**

//初始化,很重要
void init()
{
    mem(tn[0],0);
    node = 1;
    mem(exist,false);
}

//单词插入
void TireInsert( char e[], char m[])
{
    int p = 0;//火星文的下标
    int root = 0;//对应树的根
    int id;//字符的数字值
    while( m[p] != '\0')
    {
        id = m[p]-'a';
        if( tn[root][id] == 0)//不存在子树,新建一个结点
        {
            mem(tn[node],0);//结点初始化
            tn[root][id] = node;
            node++;//节点数加1
        }
        root = tn[root][id];//往下走
        p++;
    }
    strcpy(w[root],e);//把翻译值存下来
    exist[root] = true;//有
}

//单词查找
void TireSearch( char m[])
{
    int p = 0;
    int root = 0;
    int id;
    while( m[p] != '\0')
    {
        id = m[p]-'a';
        if( tn[root][id] == 0)
        {
            printf("%s",m);
            return;
        }
        root = tn[root][id];
        p++;
    }
    if( exist[root])//存在
        printf("%s",w[root]);
    else//不存在
        printf("%s",m);
}

int main()
{
    init();
    char word[25];
    while( fgets(word,25,stdin) && strcmp(word,"START\n"))
        ;
    while( fgets(word,25,stdin) && strcmp(word,"END\n"))
    {
        char a[15],b[15];
        int len = strlen(word);
        int flag = 0;
        int p = 0;
        for( int i = 0; i < len; i++)//把word分成英文火星文两个串
        {
            if( word[i] == ' ')
            {
                a[p] = '\0';
                flag = 1;
                p = 0;
                continue;
            }
            if( islower(word[i]))
            {
                if( !flag)
                    a[p++] = word[i];
                    else
                    b[p++] = word[i];
            }
        }
        b[p] = '\0';
//        printf("%d %s\n",(int)strlen(a),a);
//        printf("%d %s\n",(int)strlen(b),b);
        TireInsert(a,b);
    }

    while( fgets(sen,maxm,stdin) && strcmp(sen,"START\n"))
        ;
    while( fgets(sen,maxm,stdin) && strcmp(sen,"END\n"))
    {
        int len = strlen(sen);
        char a[15];
        int p = 0;
        for( int i = 0; i < len; i++)
        {
            if( islower(sen[i]))
                a[p++] = sen[i];
            else
            {
                a[p] = '\0';
                if( strcmp(a,""))
                    TireSearch(a);
                printf("%c",sen[i]);
                p = 0;
            }
        }
    }
    return 0;
}