二叉树层次遍历如何判断当前结点是哪层的?

来源:互联网 发布:易语言远控源码2016 编辑:程序博客网 时间:2024/05/21 09:07
         二叉树层次遍历就是按每层从左到右(一般是从左到右,若想从右到左也很简单)的次序遍历结点。下面是一个简单的例子: 

           这棵二叉树层次遍历的结果是 1,2,3,4,5。实现层次遍历一般是用队列,思路还是比较简单:

                  (1)  首先把根结点入队。

                  (2)  若队列不为空,则将队头结点出队,访问该结点,并判断出队结点的左右孩子,若左孩子非空,则左孩子入队,若右孩子非空,则右孩子也入队。

                  (3)  重复2)直到队列为空。

           那么对于上面那棵树,其层次遍历的过程我用下图来表示:

       下面给出代码:

#include <iostream>#include <queue>using namespace std;struct TreeLinkNode {int val;TreeLinkNode *left, *right, *next;TreeLinkNode(int x) : val(x), left(NULL), right(NULL), next(NULL) {}};void levelTraversal(TreeLinkNode *root) {    if(NULL==root)return;queue<TreeLinkNode*> myQueue;myQueue.push(root);while(!myQueue.empty()){TreeLinkNode *node=myQueue.front();myQueue.pop();cout<<node->val<<endl;if(NULL!=node->left)myQueue.push(node->left);if(NULL!=node->right)myQueue.push(node->right);}}int main(){TreeLinkNode *root=new TreeLinkNode(1);TreeLinkNode *left=new TreeLinkNode(2);TreeLinkNode *right=new TreeLinkNode(3);TreeLinkNode *leftRight=new TreeLinkNode(4);TreeLinkNode *rightRight=new TreeLinkNode(5);root->left=left;root->right=right;left->right=leftRight;right->left=rightRight;levelTraversal(root);system("pause");return 0;}

         下面说一个稍微麻烦一丁点儿的问题,就是上面遍历结点时,其实并不知道访问的结点是哪一层的。

         如果我在访问一个结点的时候,想知道这个结点是哪一层的,怎么办?

         我的解决方法是:在一层结束的时候,把一个特殊标记入队,用于标记一层已经结束了,那么当这个特殊标记出队时,我就知道即将遍历到开始新一层的结点了。

         我这里所用的特殊标记是NULL,下面继续先用图来阐释一下,有图有真相。。还是刚才那棵树的例子(画图还是挺累的,嘿嘿>_<):


          呼呼呼,终于画完图了(>_<)。。。啥也不说了,上代码。。。

#include <iostream>#include <queue>using namespace std;struct TreeLinkNode {int val;TreeLinkNode *left, *right, *next;TreeLinkNode(int x) : val(x), left(NULL), right(NULL), next(NULL) {}};void levelTraversal(TreeLinkNode *root) {    if(NULL==root)return;queue<TreeLinkNode*> myQueue;myQueue.push(root);myQueue.push(NULL);int level=1;while(!myQueue.empty()){TreeLinkNode *node=myQueue.front();myQueue.pop();if(NULL==node){if(myQueue.empty())break;level++;myQueue.push(NULL);continue;}cout<<"第"<<level<<"层结点:"<<node->val<<endl;if(NULL!=node->left)myQueue.push(node->left);if(NULL!=node->right)myQueue.push(node->right);}}int main(){TreeLinkNode *root=new TreeLinkNode(1);TreeLinkNode *left=new TreeLinkNode(2);TreeLinkNode *right=new TreeLinkNode(3);TreeLinkNode *leftRight=new TreeLinkNode(4);TreeLinkNode *rightRight=new TreeLinkNode(5);root->left=left;root->right=right;left->right=leftRight;right->left=rightRight;levelTraversal(root);system("pause");return 0;}
        这样就达到效果了:

        如果有更好的方法,欢迎交流~

0 0