HDOJ 1305 Immediate Decodability（trie树入门）

题意：

即判断某一个字符串是否为其他字符串的前缀。

思路：

字典树。每个单词增加一个标记，表示这个字典是否是一个单词。

下次输入的时候就可以提前判断了。

 

#include <iostream>

using namespace std;



struct node {

    bool isword;

    int child[2];

} trie[1100] ;



int size = 0;

bool flag = false;



void Insert(char* word, int i, int rt)

{

    if (!word[i])

    {

        trie[rt].isword = true;

        return ;

    }



    int m = word[i] - '0';

    if (trie[rt].child[m])

    {

        if (trie[trie[rt].child[m]].isword || !word[i+1])

            flag = true;

        Insert(word, i + 1, trie[rt].child[m]);

    }

    else

    {

        trie[rt].child[m] = ++size;

        Insert(word, i + 1, size);

    }

}



int main()

{

    char word[12];

    int cases = 0;

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

    {

        if (word[0] == '9')

        {

            if (flag)

                printf("Set %d is not immediately decodable\n", ++cases);

            else

                printf("Set %d is immediately decodable\n", ++cases);

            size = 0;

            flag = false;

            memset(trie, 0, sizeof(trie));

            continue;

        }

        Insert(word, 0, 0);

    }

    return 0;

}