poj 2001 Shortest Prefixes

第一道字典树题目,忘记释放内存,RE了无数回。。。。

#include <stdlib.h>

#include <string.h>
#include <stdio.h>
#include <ctype.h>
#include <math.h>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;
#define ll long long
#define ls rt<<1
#define rs ls1
#define lson l,mid,ls
#define rson mid+1,r,rs
#define middle (l+r)>>1
#define eps (1e-9)
#define clr_all(x,c) memset(x,c,sizeof(x))
#define clr(x,c,n) memset(x,c,sizeof(x[0])*(n+1))
#define MOD 1000000007
#define inf 100000007
#define pi acos(-1.0)
#define M 10000+5
#define N 26
typedef struct Trie{
int v;
Trie *son[N];
};
Trie *root;
char word[M][33];
int n;


Trie *createNode(){
Trie *p;
int i;
p=(Trie *)malloc(sizeof(Trie));
p->v=1;
for(i=0;i<N;i++)p->son[i]=NULL;
return p;

void release(Trie *p){
int i;
for(i=0;i<N;i++)
if(p->son[i]!=NULL)release(p->son[i]);
free(p);
}
void insert(char *st){
int i,j;
int len=strlen(st);
Trie *p=root,*q;
for(i=0;i<len;i++){
int pos=st[i]-'a';
if(p->son[pos]==NULL){
q=createNode();
p->son[pos]=q;
}else{
p->son[pos]->v++;
}
p=p->son[pos];
}
}
void search(char *st){
int i;
int len=strlen(st);
Trie *p=root;
for(i=0;i<len;i++){
int id=st[i]-'a';
printf("%c",st[i]);
p=p->son[id];
if(p==NULL||p->v==1)break;
}
puts("");
}
int main(){
    int i,j;
    root=createNode();
    i=0;
    while(gets(word[i])!=NULL){
insert(word[i]);
i++;
}
    for(j=0;j<i;j++){
printf("%s ",word[j]);
search(word[j]);
}
release(root);
system("pause");
    return 0;
}

你可能感兴趣的:(poj,字典树trie树)