poj 2503 Babelfish（字典树）

Babelfish

题目链接：

http://poj.org/problem?id=2503

解题思路：

简单的一一映射关系，可以用map直接存然后再输出，也可以用字典树存再输出，找不到，就输出“eh”。。。

AC代码（map）：

#include <iostream>
#include <cstdio>
#include <string>
#include <map>
using namespace std;

map<string,string> m;

int main(){
    string str,s1,s2;
    m.clear();
    while(getline(cin,str)){
        if(str == "")
            break;
        int i,l = str.size();
        s1 = "";s2 = "";
        for(i = 0; i < l; i++){
            if(str[i] == ' ')
                break;
            s1 += str[i];
        }
        for(i++; i < l; i++)
            s2 += str[i];
        m[s2] = s1;
    }
    while(cin>>str){
        if(m[str] != "")
            cout<<m[str]<<endl;
        else
            cout<<"eh"<<endl;
    }
    return 0;
}

AC代码（字典树）：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

struct node
{
    int flag;
    char s[15];
    struct node *next[26];
    node()
    {
        flag = 0;
        memset(next,0,sizeof(next));
    }
};

node *root = NULL;

void buildtrie(char *s,char *temp)
{
    node *p = root;
    node *tmp = NULL;
    int i,l = strlen(s);
    for(i=0;i<l;i++)
    {
        if(p->next[s[i]-'a'] == NULL)
        {
            tmp = new node;
            p->next[s[i]-'a'] = tmp;
        }
        p = p->next[s[i]-'a'];
    }
    p->flag = 1;
    strcpy(p->s,temp);
}

void findtrie(char *s)
{
    node *p = root;
    int i,l = strlen(s);
    for(i=0;i<l;i++)
    {
        if(p->next[s[i]-'a'] == NULL)
        {
            printf("eh\n");
            return;
        }
        p = p->next[s[i]-'a'];
    }
    printf("%s\n",p->s);
}

int main()
{
    char str[30],s1[15],s2[15];
    root = new node;
    while(gets(str))
    {
        if(str[0] == '\0')
            break;
        sscanf(str,"%s %s",s1,s2);
        buildtrie(s2,s1);
    }
    while(scanf("%s",str)!=EOF)
        findtrie(str);
    return 0;
}