创见博客
相同的树
七崽爱吃小饼干2026/01/08阅读 1专栏 算法合集

相同的树

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

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

示例 1:

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

示例 2:

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

示例 3:

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

提示:

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

解法:

按照相同的顺序对两颗树进行遍历即可,需要注意的是null也需要遍历

解法一:广度优先搜索(层次遍历)

ts
/**
 * Definition for a binary tree node.
 * class TreeNode {
 *     val: number
 *     left: TreeNode | null
 *     right: TreeNode | null
 *     constructor(val?: number, left?: TreeNode | null, right?: TreeNode | null) {
 *         this.val = (val===undefined ? 0 : val)
 *         this.left = (left===undefined ? null : left)
 *         this.right = (right===undefined ? null : right)
 *     }
 * }
 */

function isSameTree(p: TreeNode | null, q: TreeNode | null): boolean {
    // 是按照层序遍历给的输入,所以按照层序遍历进行检查即可
    // 要注意点是null节点也需要检查
    let q1 = []
    let q2 = []
    if(p !== null) q1.push(p)
    if(q !== null) q2.push(q)
    while(q1.length !== 0 && q2.length !== 0){
        let node1 = q1.shift()
        let node2 = q2.shift()
        if(node1 === null && node2 === null){ // 都为null则继续
            continue
        }else if(node1 === null || node2 === null){ // 只有一个为null
            return false
        }
        if(node1.val !== node2.val) return false
        q1.push(node1.left)
        q2.push(node2.left)
        q1.push(node1.right)
        q2.push(node2.right)
    }
    if(q1.length !== 0 || q2.length !== 0) return false
    return true
};

解法二:深度搜索

ts
/**
 * Definition for a binary tree node.
 * class TreeNode {
 *     val: number
 *     left: TreeNode | null
 *     right: TreeNode | null
 *     constructor(val?: number, left?: TreeNode | null, right?: TreeNode | null) {
 *         this.val = (val===undefined ? 0 : val)
 *         this.left = (left===undefined ? null : left)
 *         this.right = (right===undefined ? null : right)
 *     }
 * }
 */

function isSameTree(p: TreeNode | null, q: TreeNode | null): boolean {
    // 深度搜索(先序遍历)调换访问顺序,还有中/后序遍历(依据当前节点是先访问还是中/后访问)
    if(p === null && q === null){ // 都是null
        return true
    }else if(p === null || q === null){ // 只有一个是null
        return false
    }else if(p.val !== q.val){ // 节点值不等
        return false
    }else{
        return isSameTree(p.left, q.left) && isSameTree(p.right, q.right)
    }
};
评论
0/100