算法笔记 -- 100. 相同的树

95 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第25天,点击查看活动详情

一、题目描述:

100. 相同的树 - 力扣(LeetCode)

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

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

 

示例 1:

image.png

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

示例 2:

image.png

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

示例 3:

image.png

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

提示:

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

二、思路分析:

我们首先尝试深度优先搜索来同时遍历两棵二叉树,对每个经过的结点做以下判断:

  • 如果两个结点都是NULL,那么必然是相同的返回true
  • 如果两个结点中只有一个为NULL,那么不相同,返回false
  • 如果都不是NULL且值不同,那么返回false
  • 如果都不是以上情况,也就是都有值且值相同,那么返回对左子树和右子树分别递归调用的判断

三、AC 代码:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    bool isSameTree(TreeNode* p, TreeNode* q) 
    {
        //尝试同时遍历DFS,判断根是否相同,若有一个是空或者不同值则直接返回0
        //若相同递归调用函数判断左子和左子,右子和右子
        if(p == NULL && q == 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)));
        }
    }
};

范文参考

如果两个树相同,条件就是左子树右子树都相同,且节点本身相同(值都为null或者val相等) - 相同的树 - 力扣(LeetCode)