二叉树的镜像、二叉树剪枝——C++

剑指 Offer 27. 二叉树的镜像

请完成一个函数，输入一个二叉树，该函数输出它的镜像。

例如输入：

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

     4
   /   \
  7     2
  / \   / \
9   6 3   1

其实这题的思路还是比较清晰的，只要不断地交换左子树和右子树即可，故考虑采用递归的方法。官解的思路与之相类似，就不贴出来了。

class Solution {
public:
    TreeNode* mirrorTree(TreeNode* root) {
       if(root == nullptr)
       return nullptr;
       
       root->left = mirrorTree(root->left);
       root->right= mirrorTree(root->right);
       return swap(root);
       
    }

    TreeNode* swap(TreeNode* root)
    {
       TreeNode* Temp = root->left;
       root->left = root->right;
       root->right = Temp;
       return root;
    }
};

有时候写递归还是有点晕，理不清思路。今天模模糊糊有点感觉，记录一下。之所以可以递归就是因为大的部分是由思路相同的小的部分组成，那么对于二叉树而言，递归相当于深度优先，递归到最后就是到了最小的部分，得到最小的部分的参数后就可以看作常规非递归的方式去解（比如本题中的swap函数，由递归获得了最小部分的左子树和右子树，就可以直接投放到swap中）。

——————————————————————————————————————————

​​​​​​剑指 Offer II 047. 二叉树剪枝

给定一个二叉树 根节点 root ，树的每个节点的值要么是 0，要么是 1。请剪除该二叉树中所有节点的值为 0 的子树。

节点 node 的子树为 node 本身，以及所有 node 的后代。

示例1：

 示例2：

 这题的思路和上面的差不多，也是递归，递归到尽头之后，若是发现某个节点值为0，且没有左子树，也没有右子树，则做剪枝的处理。

官解如下（Java）：

class Solution {
    public TreeNode pruneTree(TreeNode root) {
        if( root == null) {
            return root;
        }
        root.left = pruneTree(root.left);
        root.right = pruneTree(root.right);
        if (root.val == 0 && root.left == null && root.right == null){
            root = null;
        }
        return root;
    }
}