HDU - 1305 Immediate Decodability Trie树 检验是否以某个字符串为前缀

HDU - 1305 Immediate Decodability

题意：

给你一些有 0和1 组成的的字符串，检验这些字符串中是否存在 某个字符串是另一个字符串的前缀

思路：

按长度排完序后，除第一个字符串外，其余的字符串一便输入一边检验

此题的数据太水了，即使不用排序也能过。。。

#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<sstream>
#include<cstring>
#include<bitset>
#include<cstdio>
#include<string>
#include<deque>
#include<stack>
#include<cmath>
#include<queue>
#include<set>
#include<map>
#define mod 1000000007
using namespace std;
typedef long long ll;
const int maxn = 1e5+10;
int tr[maxn][20];
//int sum[maxn];  //统计前缀
int tot,root,cnt;
bool vis[10*maxn];  //标记节点是否为字符串结尾
int ans;
struct node
{
    char s[20];
    int len;
}ss[maxn];
bool cmp(node a,node b)  //按长度从小到大排序
{
    return a.len<b.len;
}
void init()
{
    ans = tot = cnt = 0;
    memset(tr,0,sizeof(tr));
    memset(vis,false,sizeof(vis));
}
void insert(char *p)
{
    int len = strlen(p);
    root = 0;   //根节点编号为0
    for(int i=0;i<len;i++)
    {
        int id = p[i]-'0';  //第二种编码
        if(!tr[root][id])   //如果之前没有从root到id的前缀
            tr[root][id] = ++tot;  //插入,tot即为第一种编号
        //sum[tr[root][id]]++;   //前缀保存
        root = tr[root][id];   //顺着字典树往下走
        if(i==len-1)
            vis[root] = true;  //将从0到root点为一个字符串
    }
}
void find(char *p)
{
    root = 0;  //从根节点开始查找
    for(int i=0;p[i];i++)
    {
        int x = p[i]-'0';
        if(!tr[root][x])  //以root为头节点的x的字母不存在
            return ;
        root = tr[root][x];  //为查询下个字母做准备，往下走
        if(vis[root]==true)  //检验从0到root是否为某一字符串
            ans++;
    }

    //return sum[root];  //返回出现的次数
}

int main()
{
    init();
    int k = 1;
    while(gets(ss[cnt].s))
    {
        ss[cnt].len = strlen(ss[cnt].s);
        if(ss[cnt].s[0]=='9')
        {
            sort(ss,ss+cnt,cmp);
            for(int i=0;i<cnt;i++)
            {
                if(i)
                    find(ss[i].s);
                insert(ss[i].s);
            }
            if(ans==0)
                printf("Set %d is immediately decodable\n",k);
            else
                printf("Set %d is not immediately decodable\n",k);
            init();
            k++;
        }
        else
            cnt++;
    }
    return 0;
}