Leetcode 226. 翻转二叉树
Leetcode 226. 翻转二叉树
- 1、问题分析
- 2、问题解决
- 3、总结
1、问题分析
题目链接:https://leetcode-cn.com/problems/invert-binary-tree/
本质上就是一个二叉树的遍历问题。代码我已经进行了详细的注释,理解应该没有问题,读者可以作为参考,如果看不懂(可以多看几遍),欢迎留言哦!我看到会解答一下。
2、问题解决
笔者以C++
方式解决。
#include "iostream"
using namespace std;
#include "algorithm"
#include "vector"
#include "queue"
#include "set"
#include "map"
#include "string"
#include "stack"
/** * Definition for a binary tree node. * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) { }
};
class Solution {
public:
TreeNode *invertTree(TreeNode *root) {
// 如果树为空,则返回空
if (root == NULL) {
return NULL;
}
// 如果树只有一个节点,则直接返回该节点
if (root->left == NULL && root->right == NULL) {
return root;
}
// 翻转二叉树
dealChen(root);
// 返回二叉树的根节点
return root;
}
void dealChen(TreeNode *root) {
// 节点不为空才处理
if (root != nullptr) {
// 保存树的左右子节点
TreeNode *leftNode = root->left;
TreeNode *rightNode = root->right;
// 将树的左右子节点位置互换
root->right = leftNode;
root->left = rightNode;
// 递归求解左右子树
dealChen(leftNode);
dealChen(rightNode);
}
}
};
int main() {
TreeNode *pNode1 = new TreeNode(1);
TreeNode *pNode2 = new TreeNode(2);
TreeNode *pNode3 = new TreeNode(3);
TreeNode *pNode4 = new TreeNode(4);
TreeNode *pNode6 = new TreeNode(6);
TreeNode *pNode7 = new TreeNode(7);
TreeNode *pNode9 = new TreeNode(9);
pNode4->left = pNode2;
pNode4->right = pNode7;
pNode2->left = pNode1;
pNode2->right = pNode3;
pNode7->left = pNode6;
pNode7->right = pNode9;
Solution *pSolution = new Solution;
TreeNode *pNode = pSolution->invertTree(pNode4);
queue<TreeNode *> Q;
Q.push(pNode);
while (!Q.empty()) {
TreeNode *pTreeNode = Q.front();
cout << pTreeNode->val << " ";
Q.pop();
if (pTreeNode->left != NULL) {
Q.push(pTreeNode->left);
}
if (pTreeNode->right != NULL) {
Q.push(pTreeNode->right);
}
}
cout << endl;
system("pause");
return 0;
}
运行结果
有点菜,有时间再优化一下。
3、总结
难得有时间刷一波LeetCode
, 这次做一个系统的记录,等以后复习的时候可以有章可循,同时也期待各位读者给出的建议。算法真的是一个照妖镜,原来感觉自己也还行吧,但是算法分分钟教你做人。前人栽树,后人乘凉。在学习算法的过程中,看了前辈的成果,受益匪浅。
感谢各位前辈的辛勤付出,让我们少走了很多的弯路!
哪怕只有一个人从我的博客受益,我也知足了。
点个赞再走呗!欢迎留言哦!
还没有评论,来说两句吧...