开始cpp刷题之旅。
依旧是追求耗时0s的一天。
给你两棵二叉树的根节点 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
今天又是做一道树的题目,我感觉跟树相关的题目,用递归都比较好理解一点。
我看LeetCode官方的解法也是使用递归。比较好理解一点。
比较简单的思路。
如果两个二叉树都为空,则两个二叉树相同。如果两个二叉树中有且只有一个为空,则两个二叉树一定不相同。
如果两个二叉树都不为空,那么首先判断它们的根节点的值是否相同,若不相同则两个二叉树一定不同,若相同,再分别判断两个二叉树的左子树是否相同以及右子树是否相同。这是一个递归的过程,因此可以使用深度优先搜索,递归地判断两个二叉树是否相同。
class Solution {
public:
bool isSameTree(TreeNode* p, TreeNode* q) {
if (p == nullptr && q == nullptr) {
return true;
} else if (p == nullptr || q == nullptr) {
return false;
} else if (p->val != q->val) {
return false;
} else {
return isSameTree(p->left, q->left) && isSameTree(p->right, q->right);
}
}
};
看一下提交记录:
官方出品,必属精品,perfect
依旧是基于递归, 不过是一种更优雅的书写方式,借鉴官方解题思路,先判断是否为空,再判断根节点,接着再判断左右节点。
任意一环不相等,就返回false,全部通过后返回true,优雅的写法,我们可以直接这样去写。
return(a==b)
通过上述写法,我们就可以将解法一的代码这样压缩。
class Solution {
public:
bool isSameTree(TreeNode* p, TreeNode* q) {
if(p==nullptr || q== nullptr) return p==q;
return p->val==q->val && isSameTree(p->left,q->left) && isSameTree(p->right,q->right);
}
};
两行代码搞定,比较简洁,也挺好理解。
看一下提交记录:
OK,perfect!