uva-10282 - Babelfish

这道题目用的是字典树，复习了一下字典树。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string.h>
using namespace std;
struct list
{
    int leap;
    char str[100];
    struct list *next[26];
};
struct list *tree;
struct list *code()
{
    int i;
    struct list *p;
    p=new list;
    for(i=0;i<26;i++)
    {
        p->next[i]=NULL;
    }
    p->leap=0;
    return p;
}
void insert(char str1[],char str2[])
{
    int n,i,w;
    n=strlen(str1);
    struct list *p;
    p=tree;
    for(i=0;i<n;i++)
    {
        w=str1[i]-'a';
        if(p->next[w]==NULL)
        {
            p->next[w]=code();
        }
        p=p->next[w];
    }
    p->leap=1;
    strcpy(p->str,str2);
}
void search(char str[])
{
    int i,n,w;
    struct list *p;
    p=tree;
    n=strlen(str);
    for(i=0;i<n;i++)
    {
        w=str[i]-'a';
        if(p->next[w]==NULL)break;
        p=p->next[w];
    }
    if(i<n||p->leap==0)
    cout<<"eh"<<endl;
    else
    {
        cout<<p->str<<endl;
    }
}
int main()
{
    int i,len;
    char strr[100],str[100],str1[100],str2[100];
    tree=code();
    while(gets(strr))
    {
        len=strlen(strr);
        if(len==0)break;
        int w=0;
        for(i=0;i<len;i++)
        {
            if(strr[i]==' ')w=i+1;
            else
            {
                if(w==0)
                {
                    str1[i]=strr[i];
                }
                else
                {
                    str2[i-w]=strr[i];
                }
            }
        }
        str1[w-1]='\0';
        str2[len-w]='\0';
        insert(str2,str1);
    }
    while(gets(str))
    {
        search(str);
    }
    return 0;
}