给你一个二叉树的根节点 root
, 检查它是否轴对称。
示例 1:
输入:root = [1,2,2,3,4,4,3] 输出:true
示例 2:
输入:root = [1,2,2,null,3,null,3] 输出:false
提示:
- 树中节点数目在范围
[1, 1000]
内 -100 <= Node.val <= 100
进阶:你可以运用递归和迭代两种方法解决这个问题吗?
解法一:递归
/**
* 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:
// 辅助函数 check,用于递归检查两个子树是否对称
bool check(TreeNode* p, TreeNode* q) {
// 如果两个子树都为空,则它们对称
if (!q && !p)
return true;
// 如果只有一个子树为空,则它们不对称
if (!q || !p)
return false;
// 递归检查两个子树的根节点的值是否相同,并且左子树的右子树和右子树的左子树对称,以及右子树的左子树和左子树的右子树对称
return p->val == q->val && check(p->right, q->left) && check(p->left, q->right);
}
// 主函数 isSymmetric,用于检查整个二叉树是否对称
bool isSymmetric(TreeNode* root) {
// 调用辅助函数 check,从根节点开始检查
return check(root, root);
}
};
这道题的解题思路是利用递归来检查二叉树是否对称。
我们可以定义一个辅助函数check
,该函数接受两个参数,分别是节点p和节点q。在每一次调用check
函数时,我们分别比较节点p和节点q的值是否相同。如果不相同,则说明二叉树不对称,直接返回false。
如果节点p和节点q的值相同,我们还需要递归检查左子树的右子树和右子树的左子树是否对称,以及右子树的左子树和左子树的右子树是否对称。如果这些条件都满足,则说明二叉树对称,返回true;否则,返回false。
在主函数isSymmetric
中,我们调用辅助函数check
,传入根节点root作为参数,从根节点开始检查整个二叉树是否对称。
最终,主函数返回的结果就是二叉树是否对称的结果。
具体的代码实现请参考之前的代码注释部分。
解法二:迭代
class Solution { // 解决方案类
public:
// 辅助函数 check,用于迭代检查两个子树是否对称
bool check(TreeNode* u, TreeNode* v) {
queue<TreeNode*> q; // 创建一个队列来存储待比较的节点
q.push(u); // 将第一个子树的根节点入队
q.push(v); // 将第二个子树的根节点入队
while (!q.empty()) { // 当队列不为空时,继续循环
u = q.front(); q.pop(); // 从队列中取出第一个节点
v = q.front(); q.pop(); // 从队列中取出第二个节点
if (!u && !v) continue; // 如果两个节点都为空,则继续下一对节点
// 如果只有一个节点为空,或者两个节点的值不相同,则返回false
if ((!u || !v) || (u->val != v->val)) return false;
// 将第一棵树的左子节点和第二棵树的右子节点成对入队
q.push(u->left);
q.push(v->right);
// 将第一棵树的右子节点和第二棵树的左子节点成对入队
q.push(u->right);
q.push(v->left);
}
// 如果所有节点都被正确匹配,则对称
return true;
}
// 主函数 isSymmetric,用于检查整个二叉树是否对称
bool isSymmetric(TreeNode* root) {
// 调用辅助函数 check,从根节点开始检查
return check(root, root);
}
};
这段代码是使用迭代的方式来检查二叉树是否对称。
我们定义了一个辅助函数check
,该函数接受两个参数,分别是节点u
和节点v
。在主函数isSymmetric
中,我们调用check
函数,传入根节点root
作为参数,从根节点开始检查整个二叉树是否对称。
在check
函数中,我们使用队列来存储待比较的节点。我们首先将第一个子树的根节点u
和第二个子树的根节点v
入队。
在每一次循环中,我们从队列中取出两个节点u
和v
,然后进行如下操作:
- 如果两个节点都为空,则说明这一对节点对称,继续下一对节点;
- 如果只有一个节点为空,或者两个节点的值不相同,则说明二叉树不对称,返回false;
- 如果两个节点的值相同,我们将第一棵树的左子节点和第二棵树的右子节点成对入队;
- 同样地,我们将第一棵树的右子节点和第二棵树的左子节点成对入队。
如果队列为空,说明所有节点都被正确匹配,二叉树对称,返回true。
具体的代码实现请参考之前的代码注释部分。