题目描述
题目链接
给你一个二叉树的根节点 root
, 检查它是否轴对称。
题目分析
【迭代】层序遍历
轴对称可以看作是每一层结点头尾对称(相等)。那么层序遍历就直接上了:
/**
* 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 isSymmetric(TreeNode* root) {
// 对称实际上可以演变成每一层头尾对应的结点value是一样的
// 可以用层序遍历
queue<TreeNode*> q;
//vector<vecotr<int>> all;
if(root!=NULL) q.push(root);
while(!q.empty()) {
int size = q.size();
vector<int> vec;
while(size--) { // 遍历每层,然后记录他们的value到vec数组
TreeNode* node = q.front(); q.pop();
if(node) vec.push_back(node->val);
else vec.push_back(404); // 如果结点不存在,就记录他的值为404
if(node) {
q.push(node->left);
q.push(node->right);
}
}
// 对vec数组进行判断
for(int i = 0, j = vec.size()-1; i < j; i++, j--) {
cout<< vec[i] << " " << vec[j] << endl;
if(vec[i]!=vec[j]) return false;
}
}
return true;
}
};
注意这里每层遍历时还要记录不存在的结点,不然会出错。
【递归】
递归通过比较左右子树翻转是否相同来判断,那么就得采取后序遍历的方式,因为我们首先要收集左右孩子的信息传给上层结点,所以要先访问左右孩子节点,也就是后序遍历。
构造递归的三个步骤:
- 确定递归函数参数:因为我们比较的左右子树,所以应该同时传入根节点的左右孩子,即
bool compare(TreeNode* left, TreeNode* right)
- 确定递归终止条件:这里我们要考虑到多种情况,包括
(1)左孩子为空,右孩子也为空(left == NULL && right == NULL
),对称(true
);
(2)左孩子为空,右孩子不为空(left == NULL && right != NULL
),不对称(false
);
(3)左孩子不为空,右孩子为空(left != NULL && right == NULL
),不对称(false
);
(4)左孩子右孩子都不为空,但是值不相等(left->val != right->val
),不对称(false
)。 - 确定递归逻辑:我们比较左右子树有两条路径,即
left->left
与right->right
,left->right
与right->left
,所以我们的递归有两条路线:compare(left->left, right->right)
与compare(left->right, right->left)
。他们比较出来都是true才是对称的子树。
递归法的整体代码如下:
class Solution {
public:
// 确定递归参数
bool compare(TreeNode* left, TreeNode* right) {
// 确定递归终止条件
// 首先排除空节点的情况
if (left == NULL && right != NULL) return false;
else if (left != NULL && right == NULL) return false;
else if (left == NULL && right == NULL) return true;
// 排除了空节点,再排除数值不相同的情况
else if (left->val != right->val) return false;
// 确定递归逻辑
bool comp_1 = compare(left->left, right->right);
bool comp_2 = compare(left->right, right->left);
bool comp = comp_1 && comp_2;
return comp;
}
bool isSymmetric(TreeNode* root) {
if(root==NULL) return true;
return compare(root->left, root->right);
}
};