poj 1056 Trie树判断哈夫曼编码是否合法

理解了Trie树然后就能1A   其实估计这个题随便做做就能A掉，可能不需要高级数据。

先贴吉林大学的代码模板

/*==================================================*\| Trie树(k叉)| INIT: init();| 注: tree[i][tk]>0时表示单词存在, 当然也可赋予它更多含义;\*==================================================*/const int tk = 26, tb = 'a'; // tk叉; 起始字母为tb;int top, tree[N][tk + 1];  // N: 最大结点个数 top: 第一层有多少节点void init(){    top = 1;    memset(tree[0], 0, sizeof(tree[0]));}int sear(char*s){ // 失败返回0    for(int rt = 0; rt = tree[rt][*s - tb]; )        if(*(++s) == 0) return tree[rt][tk]; //插入单词结束的时候 有这一句 tree[rt][tk] = Rank;    return 0;}void Insert(char*s, int Rank = 1){    int rt, nxt;    for(rt = 0; *s; rt = nxt, ++s) {        nxt = tree[rt][*s - tb];        if(0 == nxt) {//没被使用时插入            tree[rt][*s - tb] = nxt = top;            memset(tree[top], 0, sizeof(tree[top]));            top++;        }    }    tree[rt][tk] = Rank;//1表示存在0表示不存在，也可以赋予其其他含义}void delt(char*s){ // 只做标记, 假定s一定存在    int rt = 0;    for( ; *s; ++s) rt = tree[rt][*s - tb];//tree的值本身指向下一个节点    tree[rt][tk]=0;}int prefix(char*s){ // 最长前缀    int rt = 0, lv;    for(lv = 0; *s; ++s, ++lv) {      rt = tree[rt][*s - tb];      if(rt == 0) break;    }    return lv;}

其实我发现这题数据实在太弱了，，，，，目测所有的编码已经被按长度排序了，所以我的可以直接AC

我把模板改掉然后A了poj1056

#include <cstdio>#include <cstring>#include <iostream>using namespace std;const int N = 2000+10;char str[N];const int tk = 26, tb = '0'-5; // tk叉; 起始字母为tb;int top, tree[N][tk + 1];  // N: 最大结点个数 top: 第一层有多少节点void init(){    top = 1;    memset(tree[0], 0, sizeof(tree[0]));}int Insert(char*s, int Rank = -2){//-2表示一个单词的结尾    int rt, nxt;    for(rt = 0; *s; rt = nxt, ++s) {        nxt = tree[rt][*s - tb];        if(tree[rt][tk]<0)return -2;        //这是我改的地方，原来的模板，在 tree[rt][tk] = Rank;表示结束（相当于空字符）        //那么每走到一步查看是不是以前的单词到这里是结尾，如果是的话，说明这之前的都匹配了，当然就不合法        if(0 == nxt) {//没被使用时插入            tree[rt][*s - tb] = nxt = top;            memset(tree[top], 0, sizeof(tree[top]));            top++;        }    }    tree[rt][tk] = Rank;//1表示存在0表示不存在，也可以赋予其其他含义    return 0;}int main(){    int ans=0,cnt=1,flag=0;    init();    while(~scanf("%s",str) && strcmp(str,"9"))    {        ans=flag=0;        ans=Insert(str);        if(ans<0)flag=1;        while(scanf("%s",str) && strcmp(str,"9"))        {            ans=Insert(str);            if(ans<0)flag=1;        }        if(flag){printf("Set %d is not immediately decodable\n",cnt++);}        else        {            printf("Set %d is immediately decodable\n",cnt++);        }        init();    }}

其实我的代码这组数据就过不了

10
010

01

0000
9

题中好像没说已经按长度排过序了，但是测试数据肯定是排序过的

应该加上判断，如果*(s+1)==0 && nxt!=0,那么说明已经有比现在正在插入的长的字符串，当前正在插的字符串是已经插的前缀。。。