LeetCode 226.翻转二叉树 C++

题目描述

翻转一棵二叉树。

示例：

输入：

     4
   /   \
  2     7
 / \   / \
1   3 6   9
输出：

     4
   /   \
  7     2
 / \   / \
9   6 3   1
备注:
这个问题是受到 Max Howell 的 原问题 启发的 ：

谷歌：我们90％的工程师使用您编写的软件(Homebrew)，但是您却无法在面试时在白板上写出翻转二叉树这道题，这太糟糕了。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/invert-binary-tree

解题思路1

bfs解法：
1.首先特判空树时，直接return NULL。
2.非空时bfs扫一遍，return root。

bfs函数：
1.在搜索时用一个临时变量p记录当前结点
2.用一个temp变量记录当前结点的左孩子，即p->left
3.将左右孩子进行交换，这里的交换其实也就是左右子树的交换，因为是TreeNode*类型的，p->left=p->right，p->right=temp。
4.对其左右子孩子继续搜索判断。
5.搜索结束，完成反转

代码1

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    void bfs(TreeNode* root){
       queue<TreeNode* >q;  //队列用来暂存结点
       q.push(root);    //树根入队
       while(!q.empty()){   //扫队列
           TreeNode* p=q.front();   //p用来暂存队头结点
           q.pop(); //出队
           TreeNode* temp=p->left;  //完成左右子树的交换，用temp来过渡一下
            //这里注意，交换的话并非单一结点交换，这个结点时带着子树的。也就是带着它下面的子树交换了。
           p->left=p->right;    //左换成右
           p->right=temp;   //右换成左，因为前面左已经变换过了，所以不能用p->left，用我们之前记录的temp变量
           if(p->left!=NULL){   //对左孩子判断是否要入队
               q.push(p->left);
           }
           if(p->right!=NULL){  //对右孩子判断是否要入队
               q.push(p->right);
           }
       }
    }
    TreeNode* invertTree(TreeNode* root) {
        if(root==NULL){ //特判，空树时，直接return NULL
            return root;
        }
        bfs(root);  //非空时bfs扫一遍，并完成对root的翻转
        return root;    
    }
};

解题思路2

dfs的解法
首先：dfs函数的作用是将其左右子树进行互换。
退出条件：如果当前结点为空结点，那么就返回。
1.临时变量temp来记录当前结点的左孩子，然后通过temp完成左右子树的互换。
2.对其左子树和右子树进行递归操作。

代码2

class Solution {
public:
    void dfs(TreeNode* root){
        if(root==NULL){ //退出条件：如果当前结点为空结点
            return ;
        }
        TreeNode* temp=root->left;  //临时变量记录当前结点的左孩子
        root->left=root->right; //左孩子的位置变为右孩子
        root->right=temp;       //有孩子的位置变为左孩子
        dfs(root->left);    //对其左孩子进行递归操作
        dfs(root->right);   //对其右孩子进行递归操作
    }
    TreeNode* invertTree(TreeNode* root) {
        dfs(root);  //对根节点dfs操作一遍
        return root;
    }
};