Given the roots of two binary trees p and q, write a function to check if they are the same or not.
Two binary trees are considered the same if they are structurally identical, and the nodes have the same value.
This problem tests tree recursion. The function isSameTree(p, q)
should recursively call isSameTree(p->left, q->left)
and
isSameTree(p->right, q->right)
and make sure both are true
. At
each level we check whether p
and q
are equal. The base case is
when one or both of them become nullptr
.
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) {
if (!p && !q) {
return true;
}
if (!p || !q) {
return false; // if only of them is nullptr
}
if (p->val != q->val) {
return false;
}
return isSameTree(p->left, q->left) && isSameTree(p->right, q->right);
}
};