给你一个二叉树的根节点 root
, 检查它是否轴对称。
输入:root = [1,2,2,3,4,4,3]
输出:true
源代码如下:
//递归
class Solution {
public:
bool dfs(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 dfs(p->left,q->right)&&dfs(p->right,q->left);//左右子树必须同时满足,才能使对称的
}
}
bool isSymmetric(TreeNode* root) {
return dfs(root->left,root->right);
}
};
//迭代,借助队列,先进先出的特点
class Solution {
public:
bool isSymmetric(TreeNode* root) {
//左右子树同时为空,返回空
if(root==nullptr||(root->left==nullptr&&root->right==nullptr))
return true;
queue<TreeNode*> qq;
//先将左右子树的根节点入队
qq.push(root->left);
qq.push(root->right);
while(!qq.empty())
{
//一次出队两个元素
TreeNode* lnode=qq.front();
qq.pop();
TreeNode* rnode=qq.front();
qq.pop();
//如果左节点与右节点同时为空,继续
if(lnode==nullptr&&rnode==nullptr) continue;
//如果只有一个为空,一个不为空,说明不对称,返回false
if(lnode==nullptr||rnode==nullptr) return false;
//左右节点值不相同,返回false
if(lnode->val!=rnode->val) return false;
//将左子树的左节点和右子树的右节点入队
qq.push(lnode->left);
qq.push(rnode->right);
//将左子树的右节点和右子树的左节点入队
qq.push(lnode->right);
qq.push(rnode->left);
}
return true;
}
};