给你两棵二叉树的根节点 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
解题思路:这道题与对称二叉树实现的思路相同。
- 确定参数和返回值:参数是当前节点的左孩子和右孩子,返回值为bool类型;
- 确定终止条件:
(1)如果左节点和右节点都为空,返回true;
(2)如果左节点为空,右节点不为空,返回false;
(3)如果左节点不为空,右节点为空,返回false;
(4)如果左右节点都不为空,并且左节点值不等于右节点值,返回false。 - 确定递归逻辑:
第二条四个条件都不满足,说明左节点和右节点相等,然后递归判断左节点的左孩子和右节点的左孩子是否相等,左节点的右孩子和右节点的右孩子是否相等,然后返回这两个值逻辑与的结果。
代码:
class Solution { //100. 相同的树
public:
bool isSameTree(TreeNode* p, TreeNode* q) {
if (p == nullptr && q == nullptr) return true;
if (p != nullptr&& q == nullptr) return false;
if (p == nullptr && q != nullptr) return false;
if (p != nullptr && q != nullptr && p->val != q->val) return false;
//前四个条件都不符合,说明p->val=q->val;
//然后判断p的左节点与q的左节点值是否相同 p的右节点与q的右节点值是否相同
//返回两者逻辑与的结果 表示两棵树是否相同
return isSameTree(p->left, q->left) && isSameTree(p->right, q->right);
}
};