poj2503

trie树。

这题学会的：

1.strtok（char * st1, char*st2）用st2分割st1，损坏原串，返回分割后的第一个串的指针，想获得被分割的第二个串则需要调用第二次，并且第一个参数给NULL

2.unget(ch, stdin);可以把读到的字符ch放回到输入文件中去。相当于getchar()的你过程。

3.树状结构频繁开辟指针空间浪费时间，可以直接开辟节点数组，并让指针指向数组中的未使用位。

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

#define maxn 100005
#define L 11

struct Node
{
    Node *next[26];
    char *word;
}*trie;

Node *insert(Node *proot, char *a)
{
    if (a[0] == '\0')
        return proot;
    if (proot->next[a[0] - 'a'] == NULL)
    {
        proot->next[a[0] - 'a'] = new Node;
        memset(proot->next[a[0] - 'a']->next, 0, sizeof(proot->next[a[0] - 'a']->next));
        proot->next[a[0] - 'a']->word = NULL;
    }
    return insert(proot->next[a[0]-'a'], a + 1);
}

Node *query(Node *proot, char *a)
{
    if (a[0] == '\0')
    {
        if (proot->word)
            return proot;
        return NULL;
    }
    if (proot->next[a[0]-'a'])
        return query(proot->next[a[0]-'a'], a + 1);
    return NULL;
}


int main()
{
    freopen("t.txt", "r", stdin);
    char st[L];
    trie = new Node;
    memset(trie->next, 0, sizeof(trie->next));
    trie->word = NULL;
    while (gets(st) && strcmp(st, "") != 0)
    {
        strtok(st, " ");
        char *ps = strtok(NULL, " ");
        if (ps == NULL)
            break;
        Node *p = insert(trie, ps);
        p->word = new char[10];
        strcpy(p->word, st);
    }
    while (gets(st) && strcmp(st, "") != 0)
    {
        Node *p = query(trie, st);
        if (p)
            printf("%s\n", p->word);
        else
            printf("eh\n");
    }
    return 0;
}