[前端]_一起刷leetcode 100. 相同的树

大家好，我是挨打的阿木木，爱好算法的前端摸鱼老。最近会频繁给大家分享我刷算法题过程中的思路和心得。如果你也是想提高逼格的摸鱼老，欢迎关注我，一起学习。

题目

100. 相同的树

给你两棵二叉树的根节点 p 和 q ，编写一个函数来检验这两棵树是否相同。

如果两个树在结构上相同，并且节点具有相同的值，则认为它们是相同的。

示例 1：

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

示例 2：

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

示例 3：

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

提示：

• 两棵树上的节点数目都在范围 [0, 100] 内
• -104 <= Node.val <= 104

递归解法

思路

1. 首先对特殊情况做判断，如果p、q两个节点都为空值，说明相等；
2. 如果p、q中只有一个节点是空值，说明不相等；
3. 递归判断当前值是否相等，当前节点的左节点和右节点是否都相等。

实现

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} p
 * @param {TreeNode} q
 * @return {boolean}
 */
var isSameTree = function(p, q) {
    if (!p && !q) return true;
    if (!p || !q) return false;

    return p.val === q.val && isSameTree(p.left, q.left) && isSameTree(p.right, q.right);
};

迭代解法

思路

1. 迭代的话需要通过数组去存储当前剩余未比较的节点，一开始都是根节点；
2. 然后每一轮对比第一个节点的值是否相等，以及是否同时拥有左右节点；
3. 如果上述条件都符合，那么就进行下一轮，当然当前节点如果有子节点的话放到数组末尾；
4. 递归比较直至数组为空，如果所有条件都符合则说明两个树相同。

实现

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} p
 * @param {TreeNode} q
 * @return {boolean}
 */
var isSameTree = function(p, q) {
    if (!p && !q) return true;
    if (!p || !q) return false;

    let tree1 = [ p ], tree2 = [ q ];

    while (tree1.length) {
        let cur1 = tree1.shift(),
            cur2 = tree2.shift();

        if (cur1.val !== cur2.val || !!cur1.left !== !!cur2.left || !!cur1.right !== !!cur2.right) {
            return false;
        }

        if (cur1.left) {
            tree1.push(cur1.left);
            tree2.push(cur2.left);
        }

        if (cur1.right) {
            tree1.push(cur1.right);
            tree2.push(cur2.right);
        }
    }

    return true;
};

看懂了的小伙伴可以点个关注、咱们下道题目见。如无意外以后文章都会以这种形式，有好的建议欢迎评论区留言。