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

         二叉树层次遍历就是按每层从左到右(一般是从左到右,若想从右到左也很简单)的次序遍历结点。下面是一个简单的例子: 

           这棵二叉树层次遍历的结果是 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;
}
        这样就达到效果了:

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

  • 30
    点赞
  • 34
    收藏
    觉得还不错? 一键收藏
  • 6
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值