【九月打卡】第36天 数据结构和算法

翻转二叉树（leetcode - 226）

给你一棵二叉树的根节点 root ，翻转这棵二叉树，并返回其根节点。

示例

输入：root = [4,2,7,1,3,6,9]
输出：[4,7,2,9,6,3,1]

输入：root = [2,1,3]
输出：[2,3,1]

输入：root = []
输出：[]

思路

• 分：获取左右子树
• 解：递归地反转左右子树
• 合：将反转后的左右子树换个位置放到根节点上

采用递归来实现（实现的方式可以很多种）

var invertTree = function(root) {
    if(!root) return null
    return {
        val: root.val,
        left: invertTree(root.right),
        right: invertTree(root.left)
    }
};

var invertTree = function(root) {
    if(!root) return null;
    const left = invertTree(root.left);
    const right = invertTree(root.right);
    root.right = left;
    root.left = right;
    return root;
};

var invertTree = function(root) {
    if(!root) return null;
    const temp = invertTree(root.right);
    root.right = invertTree(root.left);
    root.left = temp;
    return root;
};

相同的树（leetcode - 100）

给你两棵二叉树的根节点 p 和 q ，编写一个函数来检验这两棵树是否相同。
如果两个树在结构上相同，并且节点具有相同的值，则认为它们是相同的。

示例

输入：p = [1,2,3], q = [1,2,3]
输出：true

输入：p = [1,2], q = [1,null,2]
输出：false

输入：p = [1,2,1], q = [1,1,2]
输出：false

思路

• 分： 获取两棵树的左子树和右子树
• 解：递归地判断两棵树的左子树是否相同，以及右子树是否相同
• 合：将结果合并，如果根节点的值也相同，树就相同。

var isSameTree = function(p, q) {
    if(!p && !q) return true;
    if(p && q && p.val === q.val &&
        isSameTree(p.left, q.left) && isSameTree(p.right, q.right)
    ) {
        return true;
    }
    return false;
};

var isSameTree = function(p, q) {
    if(!p && !q) return true;
    if(!p || !q) return false;
    if(p.val !== q.val) return false;
    return isSameTree(p.left, q.left) && isSameTree(p.right, q.right);
};

时间复杂度：O(N)
空间复杂度：O(H) 【H是树的深度】