C++ 反转二叉树(镜像)

对于树这种结构，我们经常采用的策略是使用递归的方式，在这里，我们也使用递归来解决这个问题。

递归算法步骤：

1、对二叉树的左子树进行翻转

2、对二叉树的右子树进行翻转

3、将左右子树的根节点进行交换（不用考虑原二叉树的根节点，因为原二叉树的根节点在翻转前后没有改变）

核心代码如下:

BinaryTreeNode * reverBT(BinaryTreeNode *myBT)//反转二叉树 
{
	if(myBT == nullptr)
		return nullptr;
		
	myBT->m_pLeft = reverBT(myBT->m_pLeft);
	myBT->m_pRight = reverBT(myBT->m_pRight);
	
	BinaryTreeNode *temp = myBT->m_pLeft;
	myBT->m_pLeft = myBT->m_pRight;
	myBT->m_pRight = temp;
	
	return myBT;
}

程序运行完整代码：

#include<iostream>
#include<queue>
using namespace std;
 
struct BinaryTreeNode{
	int value;
	BinaryTreeNode *m_pLeft;
	BinaryTreeNode *m_pRight;
};
 
void addBTNode(BinaryTreeNode **myBT,int val);//添加节点，满足每个父亲节点大于左边的，小于右边的 
void levels_showBT(BinaryTreeNode *myBT);//层次遍历,利用队列实现 

BinaryTreeNode * reverBT(BinaryTreeNode *myBT);//反转二叉树，通过递归实现 
 
int main(){
	
	BinaryTreeNode *myBT = nullptr;
	
	addBTNode(&myBT,10);
	addBTNode(&myBT,2);
	addBTNode(&myBT,3);
	addBTNode(&myBT,15);
	addBTNode(&myBT,18);
	addBTNode(&myBT,1);
	addBTNode(&myBT,16);
	
	cout<<"反转前："<<endl;
	levels_showBT(myBT);//按层次打印 
	
	reverBT(myBT);//反转 
	cout<<"反转过后："<<endl;
	levels_showBT(myBT);
	
	return 0;
}

BinaryTreeNode * reverBT(BinaryTreeNode *myBT)//反转二叉树 
{
	if(myBT == nullptr)
		return nullptr;
		
	myBT->m_pLeft = reverBT(myBT->m_pLeft);
	myBT->m_pRight = reverBT(myBT->m_pRight);
	
	BinaryTreeNode *temp = myBT->m_pLeft;
	myBT->m_pLeft = myBT->m_pRight;
	myBT->m_pRight = temp;
	
	return myBT;
}
 
void addBTNode(BinaryTreeNode **myBT,int val){
	if(*myBT == nullptr){
		*myBT = new BinaryTreeNode();
		(*myBT)->value = val;
		(*myBT)->m_pLeft = nullptr;
		(*myBT)->m_pRight = nullptr;
		return;	
	}
		
	if(val == (*myBT)->value){
		return;
	}
	else if(val < (*myBT)->value){
		addBTNode(&(*myBT)->m_pLeft,val);
	}
	else{
		addBTNode(&(*myBT)->m_pRight,val);
	}			
}
 
void levels_showBT(BinaryTreeNode *myBT){//层次遍历,利用队列实现 
	if(myBT == nullptr )
		return;
	
	queue<BinaryTreeNode *> que;//构造一个树结点指针的队列 
	que.push(myBT);

	while(!que.empty()){
		
		if( (que.front())->m_pLeft != nullptr)
			que.push( (que.front())->m_pLeft );
			
		if( (que.front())->m_pRight != nullptr)
			que.push( (que.front())->m_pRight );
		
		cout<<(que.front())->value<<" ";
		que.pop();
	}
	cout<<endl;
		
}

运行结果如下:

C++ 反转二叉树(镜像)

猜你喜欢