hdu 1305 Immediate Decodability（Trie树）

小记：1A，在将code插入trie树里，判断插入时是否有其它单词也是这样插的路径上耽搁了下，

思路：trie树，在插入一个code时，如果一路一个一个字符的存入trie树时是没有在哪个字符时是被标记为一个code的时候，那么就返回插入成功。

这里有三种情况，

第一：从第一个就到最后一个字符都是new开辟出来的空间，也就是没有出现过已经被new过的字符

第二：前面的字符都已经被开辟了，到某一个时还没开辟，那么之后的就都开辟

第三，一路都是已经开辟过了的

对于第一种情况，肯定返回插入成功

第二种，如果前面的开辟的字符里，没有一个字符的标记是标记为是一个code的，那么就返回成功，否则返回插入失败，即不是immediate decodability

第三种，那就是当没开辟过，那么肯定是插入失败了，即是前面某个code的prefix前缀

如果插入失败那么之后的code就不需要处理了，直接读完即可

代码：

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stdlib.h>
#include <map>
#include <set>
#include <vector>
#include <stack>
#include <queue>
#include <algorithm>

using namespace std;

#define mst(a,b) memset(a,b,sizeof(a))
#define eps 10e-8

const int MAX_ = 10010;
const int MAX = 10;
const int N = 100010;
const int INF = 0x7fffffff;

typedef struct Node{
    int isStr;

    struct Node *next[MAX];

    Node():isStr(0){
        memset(next, NULL, sizeof(next));
    }
    ~Node(){
        for(int i = 0;i < MAX; ++i)
            if(next[i] != NULL)
                delete next[i];
    }
}TrieNode,*Trie;

Trie root;



int Insert(char *s){
    bool flag = false;
    TrieNode *p = root,*q;
    while(*s){
        if(p ->next[*s-'0'] == NULL){
            q = new TrieNode;
            p ->next[*s-'0'] = q;
            flag = true;
        }
        if(p->isStr == 1){
            return 0;
        }
        p = p ->next[*s-'0'];
        s++;
    }
    p->isStr = 1;
    if(!flag)return 0;
    return 1;
}

int main() {
    char s[MAX*10], str[MAX*10];
    int ans, cnt, len;
    bool flag;
    flag = true;cnt = 1;
    root = new TrieNode;
    //ans = 0;flag = false;cnt = 0;
    while(scanf("%s",s)!=EOF){
        if(s[0] == '9'){
            if(!flag){
                printf("Set %d is not immediately decodable\n",cnt++);
            }
            else printf("Set %d is immediately decodable\n",cnt++);

            flag = true;
            delete root;
            root = new TrieNode;
            continue;
        }
        if(flag)
            flag = Insert(s);
    }

}