算法竞赛入门经典 6.3.2节,二叉树的宽度优先遍历
运用结构体变量。注意在定义函数的时候是环环相扣的。最开始先定义节点的数据类型,然后定义newnode,然后再定义addnode(addnode里面要用到newnode),然后在定义read_input。
在定义根节点的时候先定义了一个root1,要不然没法用指针指向它。
感觉白书上面再定义addnode的时候有一个问题。白书上面的u->v =va;u->have_value=1没有判断经过的节点是否已经赋值。因为u指针是按照顺序走过节点,如果没有判断的话,就会把u经过的每一个的节点的值全部覆盖。
另外一个问题,我不知道为什么要定义一个getv变量(书里面是failed,我把这个改了),感觉没有这个变量一样可以运行。
在bfs里面我觉得这个方法很经典,先建立一个结构体数组,然后把节点依次放到数组里面,在把一个节点放进去以后就把它的两个子树放进去。把结构体数组中节点的值依次赋给一个整型数组。而且这两个数组是同时工作的。
read_input函数中,调用addnode函数也很精彩~用一个数组指针,指向逗号后面那个字符,然后传递给形参。相当于是把输入的字符数组截断然后给addnode函数。
#include<iostream>
#include<string.h>
using namespace std;
typedef struct TNode{
int have_value;
int v;
struct TNode *left,*right;
}Node;
Node root1;
Node* root=&root1; //define the root node
int getv;
char s[100]; //used to get data
Node *newnode(){
Node *u=(Node*)malloc(sizeof(Node));
if(u!=NULL){
u->have_value=0;
u->left=u->right=NULL;
}
return u;
}
void addnode(int va,char* s){
int n=strlen(s);
Node* u =root;
for(int i=0;i<=n;++i){
if(s[i]=='L'){
if(u->left==NULL)u->left=newnode();
u=u->left;
}
if(s[i]=='R'){
if(u->right==NULL)u->right=newnode();
u=u->right;
}
if(u->have_value) getv=1;
if(!u->have_value){u->v =va;
u->have_value=1;}
}
}
int read_input(){
getv=0;
root=newnode();
for(;;){
if(scanf("%s",s)!=1)return 0;
if(!strcmp(s,"()"))break;
int va;
sscanf(&s[1],"%d",&va);
addnode(va,strchr(s,',')+1);
}
return 1;
}
int n=0,ans[10000];
int bfs(){
int front=0,rear=1;
Node *q[10000];
q[0]=root;
while(front<rear){
Node *u=q[front++];
if(!u->have_value)return 0;
ans[n++]=u->v;
if(u->left!=NULL)q[rear++]=u->left;
if(u->right!=NULL)q[rear++]=u->right;
}
return 1;
}
int main(){
read_input();
bfs();
for(int i=0;i<n;++i){
cout<<ans[i]<<' '<<i<<endl;
}
return 0;
}