HDU1075 - What Are You Talking About(Trie树)

题目大意

给定一些火星文单词以及对应的英语单词，然后给你一些火星文，要求你翻译成对应的英文

题解

第一次写Trie树！

把所有火星文单词插入到Trie树中，并且每个火星文单词结尾的节点记录相应英文单词的位置，然后进行查询即可~~~~

代码：

#include <iostream>

#include <algorithm>

#include <cstdio>

#include <cstring>

#include <string>

#include <vector>

using namespace std;

const int maxnode = 3000005;

const int sigma_size = 26;

char dic[maxnode][15];

struct Trie

{

    int ch[maxnode][sigma_size];

    int val[maxnode];

    int sz;

    Trie(){sz=1;memset(ch[0],0,sizeof(ch[0]));val[0]=-1;}

    int idx(char c){return c-'a';}

    void insert(char *s,int v)

    {

        int u=0,n=strlen(s);

        for(int i=0;i<n;i++)

        {

            int c=idx(s[i]);

            if(!ch[u][c])

            {

                memset(ch[sz],0,sizeof(ch[sz]));

                val[sz]=-1;

                ch[u][c]=sz++;

            }

            u=ch[u][c];

        }

        val[u]=v;

    }

    int find(const char *s)

    {

        int u=0;

        for(int i=0;s[i]!='\0';i++)

        {

            int c=idx(s[i]);

            if(!ch[u][c])

                return -1;

            u=ch[u][c];

        }

        return val[u];

    }



};

Trie trie;

int main()

{

    char word[15];

    int cnt=0;

    scanf("%s",word);

    while(scanf("%s",word)!=EOF)

    {

        if(word[0]=='E') break;

        strcpy(dic[cnt],word);

        scanf("%s",word);

        trie.insert(word,cnt);

        cnt++;

    }

    scanf("%s",word);

    getchar();

    char ch;

    int len=0;

    while(scanf("%c",&ch)!=EOF&&ch!='E')

    {

        if(islower(ch))

            word[len++]=ch;

        else

        {

            word[len]='\0';

            int pos=trie.find(word);

            if(pos>=0)

                printf("%s",dic[pos]);

            else

                if(word[0]!='\0')

                    printf("%s",word);

             printf("%c",ch);

             len=0;

        }

    }

    return 0;

}