参考链接:翻转二叉树(C++)
对于树这种结构,我们经常采用的策略是使用递归的方式,在这里,我们也使用递归来解决这个问题。
递归算法步骤:
1、对二叉树的左子树进行翻转
2、对二叉树的右子树进行翻转
3、将左右子树的根节点进行交换(不用考虑原二叉树的根节点,因为原二叉树的根节点在翻转前后没有改变)
核心代码如下:
BinaryTreeNode * reverBT(BinaryTreeNode *myBT)//反转二叉树
{
if(myBT == nullptr)
return nullptr;
reverBT(myBT->m_pLeft);
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;
reverBT(myBT->m_pLeft);
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;
}
运行结果如下: