Leetcode 226. 翻转二叉树

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