100. 相同的树
给你两棵二叉树的根节点 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
提示:
两棵树上的节点数目都在范围 [0, 100] 内
-104 <= Node.val <= 104
层序遍历
同时层序遍历两棵树就好,但是开始之前要进行判断,如果根节点都空,就对称,返回true
一个空一个不空,不对称,返回false
/**
* 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) {
if (p == NULL && q == NULL) return true;
//下面这一行没想出来
if (p == NULL || q == NULL) return false;
queue<TreeNode*> que;
que.push(p);
que.push(q);
while (!que.empty()) {
TreeNode* x = que.front();
que.pop();
TreeNode* y = que.front();
que.pop();
if (!x && !y)//对称不停止,继续下一轮循环
continue;
if (!x || !y || x->val != y->val)
return false;
que.push(x->left);
que.push(y->left);
que.push(x->right);
que.push(y->right);
}
return true;
}
};
递归
前中后序递归都可以的,只不过把一个参数改成相同位置的两棵树的两个结点
递归开始的时候先判断:
- 节点都空,对称直接return true(因为此时没有子节点了,不需要向下递归了,所以可以直接return)
- 一个空,一个不空,不对称
- 都不空但是对应的值不相等,也不对称
剩下的一种情况就是当前结点对称,要继续向下递归。
之后分别同时向两棵树的左右子树递归(注意,两棵树要同方向)
/**
* 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 cmp(TreeNode* left, TreeNode* right) {
if (left == NULL && right == NULL) return true;//都空,对称
else if (left == NULL) return false;//一个空,一个不空,不对称
else if (right == NULL) return false;
else if (left->val != right->val) return false;
//左右结点都不为空,且数值相等的情况,不能直接return回去,要继续向下递归
bool l = cmp(left->left, right->left);//向同一个方向递归
bool r = cmp(left->right, right->right);
bool res = l && r;
return res;
}
bool isSameTree(TreeNode* p, TreeNode* q) {
return cmp(p, q);
}
};