「前端刷题」100. 相同的树

122 阅读2分钟

「这是我参与2022首次更文挑战的第8天,活动详情查看:2022首次更文挑战」。

题目

链接:leetcode-cn.com/problems/sa…

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

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

示例 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、当两棵树的当前节点都为 null 时返回 true

2、当其中一个为 null 另一个不为 null 时返回 false

3、当两个都不为空但是值不相等时,返回 false

执行过程:当满足终止条件时进行返回,不满足时分别判断左子树和右子树是否相同,其中要注意代码中的短路效应

时间复杂度:O(n),n 为树的节点个数

代码

/**
 * 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 == null && q == null){
        return true;
    }
    if(p == null || q == null){
        return false;
    } 
        
    if(p.val != q.val){
        return false;
    }
        
    return isSameTree(p.left, q.left) && isSameTree(p.right, q.right);
};

思路2

使用递归,进行深度优先搜索。递归返回条件:

  • 当两个节点同时遍历完叶子节点时,返回true
  • 当两个节点只有一个遍历完叶子节点时,另一个没遍历完说明不相同,返回false
  • 当两个节点的值不相同时,返回false
  • 当两个节点的值相同时,继续向下进行遍历,返回左子树的遍历结果与右子树的遍历结果的与结果

代码

/**
 * @param {TreeNode} p
 * @param {TreeNode} q
 * @return {boolean}
 */
var isSameTree = function(p, q) {
    if(p === null && q === null){
        return true;
    }
    if(p === null || q === null){
        return false
    }
    if(p.val !== q.val){
        return false;
    }
    else{
        return isSameTree(p.left, q.left) && isSameTree(p.right, q.right);
    }
};

思路2

  • 运用递归的思路,对于当前节点只需要考虑
  • 如果当前俩节点都不为空,那它俩就是相同的
  • 在上面判断的基础上,如果当前俩节点当前某一个为空,则它俩肯定不相同
  • 最后就是判断这俩节点的值是否相同,如果相同的话就判断它们的左子树及右子树是否相同
/**
 * @param {TreeNode} p
 * @param {TreeNode} q
 * @return {boolean}
 */
var isSameTree = function (p, q) {
  // 如果两个节点都是空节点 则肯定相同  如果两个节点只有一个节点为空
  if (p == null || q == null) return p == q;
  // 两个节点的值是否相同,只有相同才去递归比较左右子树
  return p.val == q.val && isSameTree(p.left, q.left) && isSameTree(p.right, q.right)
};