[编程之美] PSet3.10 分层遍历二叉树

问题描述:

        1.给定一棵二叉树,要求分层遍历二叉树(从上到下按层次访问二叉树,每一层单独输出一行),访问顺序为由左至右。

        2.打印二叉树某层次节点(从左至右),其中根节点为第0层,函数原型为int PrintNodeAtLevel(Node *root , int level),成功返回1,失败返回0

思路与解答:

         二叉树由于本身结构的递归性,常常用递归来解决。这两个问题其实是一个问题,只要解决了问题二,然后按不同层遍历一遍就可以得到问题一的结果。

         那么为什么不直接用类似于先序、中序、后序遍历的递归算法直接求解问题一呢?这是因为层序遍历一棵树本身不能直接由递归算法解出,可以用反证法证明。如果能实现对 A 节点的层序递归,在对 A 节点处理的过程中,应该递归的对两个儿子 B 和 C 分别调用了层序遍历。在这种情况下,我们无法让 B 和 C 的同一个层级的儿子在集中的时间中被遍历到,换言之,B 的第一层儿子在对 B 的调用中被遍历,而 C 的第一层儿子,则在对 C 的调用中遍历,这是分离开的。不成立,得证。

//方法:输出从根节点开始算起的第k层相当于输出从根节点左右孩子算起的k-1层
void printNodeAtLevel(Node *root , int level)
{
	if(root == NULL || level<0)
		return ;//失败
	if(level == 0)//叶子节点
		cout<<root->data<<" ";
	if(root->pLChild)
		printNodeAtLevel(root->pLChild , level-1);
	if(root->pRChild)
		printNodeAtLevel(root->pRChild,level-1);
}

       
下面如果知道该二叉树的深度n,可以直接经过n次调用得到结果:

void printNodeByLevel(Node *root , int nDepth)
{
	for(int i=0 ; i<nDepth ; i++){
		printNodeAtLevel(root , i);
		cout<<endl;
	}
}

      
如果事先不知道二叉树的深度,依旧可以用递归的思想求二叉树的深度,如下:

int Max(int a , int b)
{
	return a>b?a:b;
}
int calcTreeDepth(Node *root)
{
	if(!root)
		return -1;//依本例定义,根节点为第0层,空树的深度为-1
	int maxDepth = Max(calcTreeDepth(root->pLChild) , calcTreeDepth(root->pRChild));
	return maxDepth+1;
}

     
但是求解深度需要遍历一次二叉树,与问题二岂不是同等复杂度的问题?能不能不求深度呢?可以,只需要访问二叉树某层次失败时候返回就行了。下面是代码:

int printNodeAtLevel(Node *root , int level)
{
	if(!root || level<0)
		return 0;//未打印节点
	if( level == 0){
		cout<<root->data<<" ";
		return 1;//打印一个节点
	}
	/*
	加上if语句和不加if语句:
	加上if语句,不会递归到叶子的左右孩子,仅仅判断左右孩子如果都为空就返回了,搭配递归截止条件为叶子处返回。
	不加if语句,将会递归到叶子的左右孩子,此时需要在递归截止条件进行判定是否为NULL进行判断或赋值。
	*/
	//if(root->pLChild)
	//if(root->pRChild)仅当所有该层遍历结果为0时才返回0
	return printNodeAtLevel(root->pRChild,level-1)+printNodeAtLevel(root->pLChild,level-1);
}
void printNodeByLevel(Node *root )
{
	for(int i=0 ;  ; i++){
		if(printNodeAtLevel(root , i) == 0)//整层都未打印节点,返回
			break;
		cout<<endl;
	}
}

       
事实上这种方法的弊端在于每一层的访问都需要重新从根节点开始,而我们在访问第k层的时候,只需要知道它的前面一层(k-1)层的节点信息就够了。依照这种思路,考虑设定一个数组来存放每一层的信息,一旦遍历完该层,就开始遍历下一层,而下一层的信息可以通过压入当前层(k-1层)左右孩子节点(k层)来构造,而不要重新从根节点开始。下面是代码:

void printNodeByLevel(Node *root)
{
	vector<Node*> vec;
	int cur , last;//设定该层的起始点和结束点
	vec.push_back(root);
	cur = 0;last = vec.size();
	while(cur<vec.size()){//循环截止条件:没有更多层了
		last = vec.size();
		while(cur<last){//遍历一层
			cout<<vec[cur]->data<<" ";
			if(vec[cur]->pLChild)//有左孩子
				vec.push_back(vec[cur]->pLChild);
			if(vec[cur]->pRChild)//有右孩子
				vec.push_back(vec[cur]->pRChild);
			cur++;
		}
		cout<<endl;//该层访问结束,输出换行
	}
}

扩展问题:
       1.如果要求深度从下到上访问二叉树,每层的顺序仍然是从左到右,应该如何改进算法?(提示:可考虑左右节点的访问顺序)
       答:比较容易想到的方法是先按照层序弄好一个待遍历数组,层与层之间考虑用一个标识NULL作为分割,这样只需要反向遍历这个数组就行了,不过要注意的是弄这个遍历数组一定要先压入右节点,再压入左节点。
代码如下:

void printNodeByLevel(Node *root)
{
	vector<Node*> vec;
	int cur , last;//设定该层的起始点和结束点
	vec.push_back(root);
	cur = 0;last = vec.size();
	while(cur<vec.size()){//循环截止条件:没有更多层了
		last = vec.size();
		vec.push_back(NULL);//作为层间分割标识
		while(cur<last){//遍历一层
			if(vec[cur]->pRChild)//有右孩子
				vec.push_back(vec[cur]->pRChild);
			if(vec[cur]->pLChild)//有左孩子
				vec.push_back(vec[cur]->pLChild);
			cur++;
		}
		cur++;//跳过NULL标志位
	}
	vec.pop_back();//去掉最后一个NULL
	//---逆序遍历数组
	for(int i=vec.size()-1 ; i>=0 ; i--){
		if(vec[i] == NULL){//碰到层间分割
			cout<<endl;
			continue;
		}
		cout<<vec[i]->data<<" ";
	}
	cout<<endl;
}

          2.如果把问题一中的左右次序颠倒,改为从右到左,应该怎么改进?           答:只需要稍加修改,将“先压入右孩子再压入左孩子”的次序改为“先压入左孩子再压入右孩子”,其他均不变。

          3.[百度面试题] 要求输出二叉树第m层第k个节点值(m,k均从0开始计数)           答:思路与本例非常相似,也是考虑第m层节点相当于考虑从根孩子节点出发的第m-1层节点,然后进行引用计数或使用全局变量。代码如下:

//寻找从根节点开始第m层的第k个节点也就是寻找从根节点的孩子节点开始的第m-1层第k个节点
int findKNode(Node *root , int m , int k , int &count)//找到root为根的子树第m层第k个节点值
{
	if(m<0 || !root)
		return 0;
	if(m==0){//找到第m层
		if(k == count){//找到第k个数
			cout<<root->data<<endl;
			return 1;
		}
		count++;//如果count是指针传入,注意*count++是count指针++后再解引用,正确做法是(*count)++
		return 0;//未找到第k个数
	}
	if(findKNode(root->pLChild,m-1,k,count) || findKNode(root->pRChild,m-1,k,count))
		return 1;
	return 0;
}
int nCount = 0;
int findKNode(Node *root , int m , int k )//找到root为根的子树第m层第k个节点值
{
	if(m<0 || !root)
		return 0;
	if(m==0){//找到第m层
		if(k == nCount){//找到第k个数
			cout<<root->data<<endl;
			return 1;
		}
		nCount++;
		return 0;//未找到第k个数
	}
	if(findKNode(root->pLChild,m-1,k) || findKNode(root->pRChild,m-1,k))//左子树和右子树的第m层是否找到第k个数
		return 1;
	return 0;
}

 

    原文作者:天剑客
    原文地址: https://blog.csdn.net/spaceyqy/article/details/38733951
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞