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

2019年11月4日 235次阅读

         二叉树层次遍历就是按每层从左到右（一般是从左到右，若想从右到左也很简单）的次序遍历结点。下面是一个简单的例子： 
《二叉树层次遍历如何判断当前结点是哪层的？》

           这棵二叉树层次遍历的结果是 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;
}

        这样就达到效果了：

《二叉树层次遍历如何判断当前结点是哪层的？》         如果有更好的方法，欢迎交流~

点赞