HDU_1305 Immediate Decodability 【字典树】

题目链接

题目描述

不断输入字符串直到字符为”9”结束，如果在”9”之前，一个字符的代码不是另一个字符的代码的前缀，则这种情况为立即可解就输出”is immediately decodable”否则输出”is not immediately decodable”

解题思路

字典树求前缀是否出现过，每插入一个字符就对字符进行查找。
插入时给字符最后的节点加1表示该节点存在。查找的过程：如果在向下扩展的过程中发现节点的cnt有值，就说明该字符已经存在，即为当前查找的字符的前缀，就让标记为假然后按格式输出就可以了。

代码部分

#include <bits/stdc++.h>
using namespace std;
const int en = 10;
const int maxn = 1e6;
bool flag;
struct Trie
{
    int cnt;
    Trie *nex[en];
} T[maxn];
int top;
inline int idx(char c)
{
    return c - '0';
}
inline Trie * CreatTrie()
{
    Trie *p = &T[top ++];
    p -> cnt = 0;
    for(int i = 0; i< en; ++ i) p -> nex[i] = NULL;
    return p;
}
inline Trie * Init()
{
    top = 0;
    return CreatTrie();
}
void Insert(Trie *root, string s)
{
    Trie *p = root;
    for(int i = 0; i < s.size(); ++ i)
    {
        int temp = idx(s[i]);
        if(p -> nex[temp] == NULL) p -> nex[temp] = CreatTrie();
        p = p -> nex[temp];
    }
    p -> cnt ++;
}
int Search(Trie *root, string s)
{
    Trie *p = root;
    for(int i = 0; i < s.size(); ++ i)
    {
        int temp = idx(s[i]);
        if(p -> nex[temp] -> cnt && i != s.size() - 1) flag = false;
        p = p -> nex[temp];
    }
}
int main()
{
    ios::sync_with_stdio(false);
    int cas = 0;
    Trie *root = Init();
    flag = true;
    string s;
    while(cin >> s)
    {
        if(s != "9")
        {
            Insert(root, s);
            Search(root, s);
        }
        else
        {
            if(flag)
                cout << "Set " << ++ cas <<  " is immediately decodable" << endl;
            else
                cout << "Set " << ++ cas <<  " is not immediately decodable" << endl;
            Trie *root = Init();
            flag = true;
        }
    }
    return 0;
}