文章内容是自己刷leetcode题目的一些总结。
文章内容参考公众号: 代码随想录。
一.题目描述
二.分析问题
对于这道题呢,我们可以这么来分析。我们要比较二叉树是否是对称,那就是要比较左右对称节点处元素值是否相同,不外乎有这么几种情况:
1.左右两个节点都为空,那是对称的。
2.左右节点一个为空,一个不为空,不是对称的。
3.左右节点都不为空但 val 不同,不是对称的。
4.剩下最后一种情况,左右节点不空且 val 相同,分析他们的子树,子树对称那么就是对称的。
这道题我给出递归和迭代两种写法。
三.对称二叉树递归与迭代代码
1.递归
我们在分析问题中其实是给出了递归的终止条件,那么单次的递归逻辑是什么呢?就是对于目前处理的左右节点,我们要将二叉树分为里层和外层,分层的逻辑就是左节点的右孩子和右节点的左孩子是里层,左节点的左孩子和右节点的右孩子是外层。里层和外层分别对应了一对对称节点,如果里层和外层都是对称的,那么本层才是对称的。
class Solution {
public:
bool cmp(TreeNode* left, TreeNode* right) {
//递归终止条件
if(left == nullptr && right == nullptr) return true;
else if(left && right == nullptr) return false;
else if(right && left == nullptr) return false;
else if(left->val != right->val) return false;
//剩下一种情况就是都不空且相等
//单次逻辑,我们进行下一层
bool inside = cmp(left->right, right->left); //里层
bool outside = cmp(left->left, right->right); //外层
bool ans = inside && outside;
return ans;
}
bool isSymmetric(TreeNode* root) {
if(root == nullptr) return true;
return cmp(root->left, root->right);
}
};
2.迭代
迭代法我们使用队列来实现,与递归的思路相似,每次比较两个节点,为终止条件则返回或进行下一次迭代,否则判断孩子的对称性。
注意迭代过程中,每次进入队列的两个相邻元素在树中的位置是对称的。
class Solution {
public:
bool isSymmetric(TreeNode* root) {
//迭代写法:队列实现
if(root == nullptr) return true;
queue<TreeNode*> que;
que.push(root->left);
que.push(root->right);
while(!que.empty()) {
TreeNode* nodeLeft = que.front(); que.pop();
TreeNode* nodeRight = que.front(); que.pop();
if(!nodeLeft && !nodeRight) continue; //都为空说明对称
if(!nodeLeft || !nodeRight || nodeLeft->val != nodeRight->val) {
return false; //一个空,或者都不空但不相等为false
}
que.push(nodeLeft->left); //左左
que.push(nodeRight->right); //右右
que.push(nodeLeft->right); //左右
que.push(nodeRight->left); //右左
}
return true;
}
};
四.总结
这道题的终止条件相对而言比较多,需要仔细分析一下。
不论递归还是迭代,核心思想都是比较对称位置的节点值是否相同。
所以就有了递归中的分层,迭代中每次取出两个节点的思路。