二叉树宽度优先遍历

来源:互联网 发布:淘宝手机装修在哪里 编辑:程序博客网 时间:2024/05/18 02:57

算法竞赛入门经典  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 nodeint 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;}