Hdu 1075 - What Are You Talking About（字典树）

题目链接：
http://acm.hdu.edu.cn/showproblem.php?pid=1075

题目大意：
先给出一些单词的英文翻译，然后给出一篇文章，存在标点，若文章中单词存在翻译，则转换成英文单词，若没有则不变，然后输出翻译后的文章

分析：
建立一个译文数组，在字典树中插入单词时，即可在单词结点处添加对应译文在数组中的下标，然后遍历文章，遇到标点与空格切分单词，特判最后末尾不是标点的情况，在树中查找是否出现过当前单词即可，存在即将单词的译文输出，否则输出原文，然后将当前遍历的标点直接输出

代码：

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cmath>
#include<map>
using namespace std;

typedef long long ll;


struct Trie{
    Trie *next[26];
    int val;
    Trie(){
        val = -1;
        for (int i = 0 ; i < 26 ; i ++)
            next[i] = NULL;
    }
};

void addword(char *str,Trie* node,int i)
{
    if (node->next[str[0]-'a']==NULL)
    {
        node->next[str[0]-'a'] = new Trie;
        node = node->next[str[0]-'a'];
    }
    else
    {
        node = node->next[str[0]-'a'];
    }
    str++;
    if (*str)
        addword(str,node,i);
    else
    {
        node->val = i;
        return;
    }
}


int query(char *str ,Trie *node)
{
    if (node->next[*str-'a']==NULL)
        return -1;
    else
        node= node->next[*str-'a'];
    ++str;
    if (*str)
        return query(str,node);
    else
        return node->val;
}

char str[4000];
char temp[4000],tw[4000];
char word[1200000][20];


int main()
{
    int T,n,i;
    int cnt = 0;
    Trie* head = new Trie;
    scanf("%s",str);
    getchar();
    gets(str);
    while (strcmp(str,"END")!=0)
    {
        for (i = 0 ; ; i ++)
        {
            if (str[i]==' ')
            {
                word[cnt][i++] = 0;
                break;
            }
            else
                word[cnt][i] = str[i];
        }
        cnt++;
        addword(str+i,head,cnt-1);
        gets(str);
    }
    scanf("%s",str);
    getchar();
    gets(str);
    int lw;
    while (strcmp(str,"END")!=0)
    {
        cnt =0;
        int tp = 0;
        memset(temp,0,sizeof(temp));
        for (i = 0; str[i]; i ++)
        {
            if (str[i]>='a'&&str[i]<='z')
            {
                tw[cnt++] = str[i];
            }
            else
            {
                if (cnt!=0)
                {
                    tw[cnt] = 0;
                    int pos = query(tw,head);
                    if (pos!=-1)
                    {
                        strcat(temp,word[pos]);
                        tp += strlen(word[pos]);
                    }
                    else
                    {
                        strcat(temp,tw);
                        tp += strlen(tw);
                    }
                    cnt = 0;
                }
                temp[tp++] = str[i];
            }
        }
        if (cnt!=0)
        {
            tw[cnt] = 0;
            int pos = query(tw,head);
            if (pos!=-1)
            {
                strcat(temp,word[pos]);
                tp += strlen(word[pos]);
            }
            else
            {
                strcat(temp,tw);
                tp += strlen(tw);
            }
            cnt = 0;
        }
        temp[tp] = 0;
        printf("%s\n",temp);
        gets(str);
    }

    return 0;
}