poj 2001/1056 Trie树（求单词表的最短前缀/判断立即码）

题意：2001：输入若干单词组成单词表，求每个单词合理的最短前缀（通过前缀确定唯一的单词）。

1056：给定一系列01码，判断是否为立即码，即没有序列为另一个序列的前缀。

思路：使用trie树。其中的flag域用于标定通过这个节点是否有多个单词。初始为0（也就是插入第一个通过该节点的单词时），如果经过此节点有多于一个单词，则置为1。输出方法：从根开始，如果节点flag为1，则继续输出，为0则停止。版本2在trie结构体内进行初始化，运用了C++的写法。（最后是1056的代码）

#include <stdio.h>
#include <string.h>
char s[1005][22];
typedef struct trie{
int flag;
struct trie* next[26];
}trie;
trie t[20000];
trie *alloc,*root;
int n=-1;
void init(){
int i;
alloc = t;
root = alloc++;
root->flag = 0;
for(i = 0;i<26;i++)
root->next[i] = NULL;
}
void insert(char s[22]){
int i,j;
trie *p,*q;
p = root;
for(i = 0;s[i]!='\0';i++){
if(p->next[s[i]-'a']){//如果该节点已建立（和之前某单词前缀重复），flag置为1
p = p->next[s[i]-'a'];
p->flag = 1;
}else{
q = alloc++;
q->flag = 0;
for(j = 0;j<26;j++)//扫字符串用了i，此地不能再用
q->next[j] = NULL;
p->next[s[i]-'a'] = q;
p = q;
}
}
}
void findprefix(char s[22]){
int i;
trie *p = root;
for(i = 0;s[i]!='\0';i++){
putchar(s[i]);
p = p->next[s[i]-'a'];
if(p->flag == 0)
break;
}
putchar('\n');
}
int main(){
int i,j;
freopen("a.txt","r",stdin);
init();
while(scanf("%s",s[++n])!=EOF)
insert(s
);
for(i = 0;i<=n;i++){
printf("%s ",s[i]);
findprefix(s[i]);
}
return 0;
}

2001 版本2：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#define N 1040
#define INF 0x3fffffff
using namespace std;
char s[1005][22];
int num = 0;
struct node{
int flag;
struct node* next[26];
node(){
flag = 0;
memset(next,NULL,sizeof(next));
}
}t[20005],*root;
void insert(char x[22]){
int i;
struct node *p = root;
for(i = 0;x[i]!='\0';i++){
if(!p->next[x[i]-'a']){
num++;
p->next[x[i]-'a'] = t+num;
}else
p->next[x[i]-'a']->flag = 1;
p = p->next[x[i]-'a'];
}
}
void output(char x[22]){
int i;
struct node *p = root;
for(i = 0;x[i]!='\0';i++){
putchar(x[i]);
p = p->next[x[i]-'a'];
if(!p || p->flag == 0)
break;
}
putchar('\n');
}
int main(){
int i,len = 0;
root = t;
while(scanf("%s",s[++len])!=EOF){
insert(s[len]);
}
for(i = 1;i<=len;i++){
printf("%s ",s[i]);
output(s[i]);
}
return 0;
}

1056：

#include <stdio.h>
#include <string.h>
typedef struct node{
int flag;
struct node *next[2];
}Node;
Node tree[100000],*root,*alloc;
char s[1025];
int c = 1,flag;
void init(){
flag = 1;
alloc = tree;
root = alloc++;
root->flag = 0;
root->next[0] = root->next[1] = NULL;
}
int insert(char *s){
int i,flag = 1;
Node *p,*q;
p = root;
for(i = 0;s[i]!='\0';i++){
if(!p->next[s[i]-'0']){
flag = 0;
q = alloc++;
q->flag = 0;
q->next[0] = q->next[1] = NULL;
p->next[s[i]-'0'] = q;
}
p = p->next[s[i]-'0'];
if(p->flag)
return 0;
}
p->flag = 1;
return flag==0;
}
int main(){
//freopen("a.txt","r",stdin);
while(scanf("%s",s)!=EOF){
int j;
init();
do{
if(flag){
j = insert(s);
if(!j)
flag = 0;
}
scanf("%s",s);
}while(strcmp(s,"9"));
if(flag)
printf("Set %d is immediately decodable\n",c++);
else
printf("Set %d is not immediately decodable\n",c++);
}
}