Leetcode 226. 翻转二叉树

Leetcode 226. 翻转二叉树

  • 1、问题分析
  • 2、问题解决
  • 3、总结

1、问题分析

题目链接:https://leetcode-cn.com/problems/invert-binary-tree/
  本质上就是一个二叉树的遍历问题。代码我已经进行了详细的注释,理解应该没有问题,读者可以作为参考,如果看不懂(可以多看几遍),欢迎留言哦!我看到会解答一下。

2、问题解决

  笔者以C++方式解决。

  1. #include "iostream"
  2. using namespace std;
  3. #include "algorithm"
  4. #include "vector"
  5. #include "queue"
  6. #include "set"
  7. #include "map"
  8. #include "string"
  9. #include "stack"
  10. /** * Definition for a binary tree node. * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */
  11. struct TreeNode {
  12. int val;
  13. TreeNode *left;
  14. TreeNode *right;
  15. TreeNode(int x) : val(x), left(NULL), right(NULL) { }
  16. };
  17. class Solution {
  18. public:
  19. TreeNode *invertTree(TreeNode *root) {
  20. // 如果树为空,则返回空
  21. if (root == NULL) {
  22. return NULL;
  23. }
  24. // 如果树只有一个节点,则直接返回该节点
  25. if (root->left == NULL && root->right == NULL) {
  26. return root;
  27. }
  28. // 翻转二叉树
  29. dealChen(root);
  30. // 返回二叉树的根节点
  31. return root;
  32. }
  33. void dealChen(TreeNode *root) {
  34. // 节点不为空才处理
  35. if (root != nullptr) {
  36. // 保存树的左右子节点
  37. TreeNode *leftNode = root->left;
  38. TreeNode *rightNode = root->right;
  39. // 将树的左右子节点位置互换
  40. root->right = leftNode;
  41. root->left = rightNode;
  42. // 递归求解左右子树
  43. dealChen(leftNode);
  44. dealChen(rightNode);
  45. }
  46. }
  47. };
  48. int main() {
  49. TreeNode *pNode1 = new TreeNode(1);
  50. TreeNode *pNode2 = new TreeNode(2);
  51. TreeNode *pNode3 = new TreeNode(3);
  52. TreeNode *pNode4 = new TreeNode(4);
  53. TreeNode *pNode6 = new TreeNode(6);
  54. TreeNode *pNode7 = new TreeNode(7);
  55. TreeNode *pNode9 = new TreeNode(9);
  56. pNode4->left = pNode2;
  57. pNode4->right = pNode7;
  58. pNode2->left = pNode1;
  59. pNode2->right = pNode3;
  60. pNode7->left = pNode6;
  61. pNode7->right = pNode9;
  62. Solution *pSolution = new Solution;
  63. TreeNode *pNode = pSolution->invertTree(pNode4);
  64. queue<TreeNode *> Q;
  65. Q.push(pNode);
  66. while (!Q.empty()) {
  67. TreeNode *pTreeNode = Q.front();
  68. cout << pTreeNode->val << " ";
  69. Q.pop();
  70. if (pTreeNode->left != NULL) {
  71. Q.push(pTreeNode->left);
  72. }
  73. if (pTreeNode->right != NULL) {
  74. Q.push(pTreeNode->right);
  75. }
  76. }
  77. cout << endl;
  78. system("pause");
  79. return 0;
  80. }

运行结果

在这里插入图片描述

有点菜,有时间再优化一下。

3、总结

  难得有时间刷一波LeetCode, 这次做一个系统的记录,等以后复习的时候可以有章可循,同时也期待各位读者给出的建议。算法真的是一个照妖镜,原来感觉自己也还行吧,但是算法分分钟教你做人。前人栽树,后人乘凉。在学习算法的过程中,看了前辈的成果,受益匪浅。
感谢各位前辈的辛勤付出,让我们少走了很多的弯路!
哪怕只有一个人从我的博客受益,我也知足了。
点个赞再走呗!欢迎留言哦!

发表评论

表情:
评论列表 (有 0 条评论,64人围观)

还没有评论,来说两句吧...

相关阅读

    相关 LeetCode226. 翻转

    今日份打卡力扣,这次用的是Java语言,之前都是用C/C++写的算法题,因为目前是从事Java开发方向,一会用C/C++,一会用Java,容易混淆,而且自己C++学的也不好,所