二叉树的镜像
这道题很简单就是交换结点,利用递归解解决问题思路清晰,代码简洁;
MirrorTree.cpp:
#include <iostream>
#include <cstdio>
#include <stack>
#include "BinaryTree.h"
using namespace std;
void MirrorRecursively(BinaryTreeNode *pNode)
{
if((pNode == NULL) || (pNode->m_pLeft == NULL && pNode->m_pRight))
{
return ;
}
BinaryTreeNode* pTemp = pNode->m_pLeft;
pNode->m_pLeft = pNode->m_pRight;
pNode->m_pRight = pTemp;
if(pNode->m_pLeft)
{
MirrorRecursively(pNode->m_pLeft);
}
if(pNode->m_pRight)
{
MirrorRecursively(pNode->m_pRight);
}
}
void MirrorIteratively(BinaryTreeNode* pRoot)
{
if(pRoot == NULL)
{
return ;
}
stack<BinaryTreeNode *> stackTreeNode;
stackTreeNode.push(pRoot);
while(stackTreeNode.size() > 0)
{
BinaryTreeNode* pNode = stackTreeNode.top();
stackTreeNode.pop();
BinaryTreeNode* pTemp = pNode->m_pLeft;
pNode->m_pLeft = pNode->m_pRight;
pNode->m_pRight = pTemp;
if(pNode->m_pLeft)
{
stackTreeNode.push(pNode->m_pLeft);
}
if(pNode->m_pRight)
{
stackTreeNode.push(pNode->m_pRight);
}
}
}
// ====================测试代码====================
// 测试完全二叉树:除了叶子节点,其他节点都有两个子节点
// 8
// 6 10
// 5 7 9 11
void Test1()
{
printf("=====Test1 starts:=====\n");
BinaryTreeNode* pNode8 = CreateBinaryTreeNode(8);
BinaryTreeNode* pNode6 = CreateBinaryTreeNode(6);
BinaryTreeNode* pNode10 = CreateBinaryTreeNode(10);
BinaryTreeNode* pNode5 = CreateBinaryTreeNode(5);
BinaryTreeNode* pNode7 = CreateBinaryTreeNode(7);
BinaryTreeNode* pNode9 = CreateBinaryTreeNode(9);
BinaryTreeNode* pNode11 = CreateBinaryTreeNode(11);
ConnectTreeNodes(pNode8, pNode6, pNode10);
ConnectTreeNodes(pNode6, pNode5, pNode7);
ConnectTreeNodes(pNode10, pNode9, pNode11);
PrintTree(pNode8);
printf("=====Test1: MirrorRecursively=====\n");
MirrorRecursively(pNode8);
PrintTree(pNode8);
printf("=====Test1: MirrorIteratively=====\n");
MirrorIteratively(pNode8);
PrintTree(pNode8);
DestroyTree(pNode8);
}
// 测试二叉树:出叶子结点之外,左右的结点都有且只有一个左子结点
// 8
// 7
// 6
// 5
// 4
void Test2()
{
printf("=====Test2 starts:=====\n");
BinaryTreeNode* pNode8 = CreateBinaryTreeNode(8);
BinaryTreeNode* pNode7 = CreateBinaryTreeNode(7);
BinaryTreeNode* pNode6 = CreateBinaryTreeNode(6);
BinaryTreeNode* pNode5 = CreateBinaryTreeNode(5);
BinaryTreeNode* pNode4 = CreateBinaryTreeNode(4);
ConnectTreeNodes(pNode8, pNode7, NULL);
ConnectTreeNodes(pNode7, pNode6, NULL);
ConnectTreeNodes(pNode6, pNode5, NULL);
ConnectTreeNodes(pNode5, pNode4, NULL);
PrintTree(pNode8);
printf("=====Test2: MirrorRecursively=====\n");
MirrorRecursively(pNode8);
PrintTree(pNode8);
printf("=====Test2: MirrorIteratively=====\n");
MirrorIteratively(pNode8);
PrintTree(pNode8);
DestroyTree(pNode8);
}
// 测试二叉树:出叶子结点之外,左右的结点都有且只有一个右子结点
// 8
// 7
// 6
// 5
// 4
void Test3()
{
printf("=====Test3 starts:=====\n");
BinaryTreeNode* pNode8 = CreateBinaryTreeNode(8);
BinaryTreeNode* pNode7 = CreateBinaryTreeNode(7);
BinaryTreeNode* pNode6 = CreateBinaryTreeNode(6);
BinaryTreeNode* pNode5 = CreateBinaryTreeNode(5);
BinaryTreeNode* pNode4 = CreateBinaryTreeNode(4);
ConnectTreeNodes(pNode8, NULL, pNode7);
ConnectTreeNodes(pNode7, NULL, pNode6);
ConnectTreeNodes(pNode6, NULL, pNode5);
ConnectTreeNodes(pNode5, NULL, pNode4);
PrintTree(pNode8);
printf("=====Test3: MirrorRecursively=====\n");
MirrorRecursively(pNode8);
PrintTree(pNode8);
printf("=====Test3: MirrorIteratively=====\n");
MirrorIteratively(pNode8);
PrintTree(pNode8);
DestroyTree(pNode8);
}
// 测试空二叉树:根结点为空指针
void Test4()
{
printf("=====Test4 starts:=====\n");
BinaryTreeNode* pNode = NULL;
PrintTree(pNode);
printf("=====Test4: MirrorRecursively=====\n");
MirrorRecursively(pNode);
PrintTree(pNode);
printf("=====Test4: MirrorIteratively=====\n");
MirrorIteratively(pNode);
PrintTree(pNode);
}
// 测试只有一个结点的二叉树
void Test5()
{
printf("=====Test5 starts:=====\n");
BinaryTreeNode* pNode8 = CreateBinaryTreeNode(8);
PrintTree(pNode8);
printf("=====Test4: MirrorRecursively=====\n");
MirrorRecursively(pNode8);
PrintTree(pNode8);
printf("=====Test4: MirrorIteratively=====\n");
MirrorIteratively(pNode8);
PrintTree(pNode8);
}
int main()
{
Test1();
Test2();
Test3();
Test4();
Test5();
return 0;
}
BinaryTree.h:
#ifndef _BINARY_TREE_H_
#define _BINARY_TREE_H_
struct BinaryTreeNode
{
int m_nValue;
BinaryTreeNode* m_pLeft;
BinaryTreeNode* m_pRight;
};
BinaryTreeNode* CreateBinaryTreeNode(int value);
void ConnectTreeNodes(BinaryTreeNode* pParent, BinaryTreeNode* pLeft, BinaryTreeNode* pRight);
void PrintTreeNode(BinaryTreeNode* pNode);
void PrintTree(BinaryTreeNode* pRoot);
void DestroyTree(BinaryTreeNode* pRoot);
#endif /*_BINARY_TREE_H_*/
BinaryTree.cpp:
#include <iostream>
#include "BinaryTree.h"
#include <cstdio>
BinaryTreeNode* CreateBinaryTreeNode(int value)
{
BinaryTreeNode* pNode = new BinaryTreeNode();
pNode->m_nValue = value;
pNode->m_pLeft = NULL;
pNode->m_pRight = NULL;
return pNode;
}
void ConnectTreeNodes(BinaryTreeNode* pParent, BinaryTreeNode* pLeft, BinaryTreeNode* pRight)
{
if(pParent != NULL)
{
pParent->m_pLeft = pLeft;
pParent->m_pRight = pRight;
}
}
void PrintTreeNode(BinaryTreeNode* pNode)
{
if(pNode != NULL)
{
printf("value of this node is: %d\n", pNode->m_nValue);
if(pNode->m_pLeft != NULL)
printf("value of its left child is: %d.\n", pNode->m_pLeft->m_nValue);
else
printf("left child is null.\n");
if(pNode->m_pRight != NULL)
printf("value of its right child is: %d.\n", pNode->m_pRight->m_nValue);
else
printf("right child is null.\n");
}
else
{
printf("this node is null.\n");
}
printf("\n");
}
void PrintTree(BinaryTreeNode* pRoot)
{
PrintTreeNode(pRoot);
if(pRoot != NULL)
{
if(pRoot->m_pLeft != NULL)
PrintTree(pRoot->m_pLeft);
if(pRoot->m_pRight != NULL)
PrintTree(pRoot->m_pRight);
}
}
void DestroyTree(BinaryTreeNode* pRoot)
{
if(pRoot != NULL)
{
BinaryTreeNode* pLeft = pRoot->m_pLeft;
BinaryTreeNode* pRight = pRoot->m_pRight;
delete pRoot;
pRoot = NULL;
DestroyTree(pLeft);
DestroyTree(pRight);
}
}
Makefile:
.PHONY:clean
CPP=g++
CFLAGS=-Wall -g
BIN=test
OBJS=MirrorTree.o BinaryTree.o
LIBS=
$(BIN):$(OBJS)
$(CPP) $(CFLAGS) $^ -o $@ $(LIBS)
%.o:%.cpp
$(CPP) $(CFLAGS) -c $< -o $@
clean:
rm -f *.o $(BIN)
运行结果:
=====Test1 starts:=====
value of this node is: 8
value of its left child is: 6.
value of its right child is: 10.
value of this node is: 6
value of its left child is: 5.
value of its right child is: 7.
value of this node is: 5
left child is null.
right child is null.
value of this node is: 7
left child is null.
right child is null.
value of this node is: 10
value of its left child is: 9.
value of its right child is: 11.
value of this node is: 9
left child is null.
right child is null.
value of this node is: 11
left child is null.
right child is null.
=====Test1: MirrorRecursively=====
value of this node is: 8
value of its left child is: 10.
value of its right child is: 6.
value of this node is: 10
value of its left child is: 11.
value of its right child is: 9.
value of this node is: 11
left child is null.
right child is null.
value of this node is: 9
left child is null.
right child is null.
value of this node is: 6
value of its left child is: 7.
value of its right child is: 5.
value of this node is: 7
left child is null.
right child is null.
value of this node is: 5
left child is null.
right child is null.
=====Test1: MirrorIteratively=====
value of this node is: 8
value of its left child is: 6.
value of its right child is: 10.
value of this node is: 6
value of its left child is: 5.
value of its right child is: 7.
value of this node is: 5
left child is null.
right child is null.
value of this node is: 7
left child is null.
right child is null.
value of this node is: 10
value of its left child is: 9.
value of its right child is: 11.
value of this node is: 9
left child is null.
right child is null.
value of this node is: 11
left child is null.
right child is null.
=====Test2 starts:=====
value of this node is: 8
value of its left child is: 7.
right child is null.
value of this node is: 7
value of its left child is: 6.
right child is null.
value of this node is: 6
value of its left child is: 5.
right child is null.
value of this node is: 5
value of its left child is: 4.
right child is null.
value of this node is: 4
left child is null.
right child is null.
=====Test2: MirrorRecursively=====
value of this node is: 8
left child is null.
value of its right child is: 7.
value of this node is: 7
left child is null.
value of its right child is: 6.
value of this node is: 6
left child is null.
value of its right child is: 5.
value of this node is: 5
left child is null.
value of its right child is: 4.
value of this node is: 4
left child is null.
right child is null.
=====Test2: MirrorIteratively=====
value of this node is: 8
value of its left child is: 7.
right child is null.
value of this node is: 7
value of its left child is: 6.
right child is null.
value of this node is: 6
value of its left child is: 5.
right child is null.
value of this node is: 5
value of its left child is: 4.
right child is null.
value of this node is: 4
left child is null.
right child is null.
=====Test3 starts:=====
value of this node is: 8
left child is null.
value of its right child is: 7.
value of this node is: 7
left child is null.
value of its right child is: 6.
value of this node is: 6
left child is null.
value of its right child is: 5.
value of this node is: 5
left child is null.
value of its right child is: 4.
value of this node is: 4
left child is null.
right child is null.
=====Test3: MirrorRecursively=====
value of this node is: 8
left child is null.
value of its right child is: 7.
value of this node is: 7
left child is null.
value of its right child is: 6.
value of this node is: 6
left child is null.
value of its right child is: 5.
value of this node is: 5
left child is null.
value of its right child is: 4.
value of this node is: 4
left child is null.
right child is null.
=====Test3: MirrorIteratively=====
value of this node is: 8
value of its left child is: 7.
right child is null.
value of this node is: 7
value of its left child is: 6.
right child is null.
value of this node is: 6
value of its left child is: 5.
right child is null.
value of this node is: 5
value of its left child is: 4.
right child is null.
value of this node is: 4
left child is null.
right child is null.
=====Test4 starts:=====
this node is null.
=====Test4: MirrorRecursively=====
this node is null.
=====Test4: MirrorIteratively=====
this node is null.
=====Test5 starts:=====
value of this node is: 8
left child is null.
right child is null.
=====Test4: MirrorRecursively=====
value of this node is: 8
left child is null.
right child is null.
=====Test4: MirrorIteratively=====
value of this node is: 8
left child is null.
right child is null.