层序遍历
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
queue<TreeNode*> que;
if (root != NULL) que.push(root);
vector<vector<int>> result;
while (!que.empty()) {
int size = que.size();
vector<int> vec;
// 这里一定要使用固定大小size,不要使用que.size(),因为que.size是不断变化的
for (int i = 0; i < size; i++) {
TreeNode* node = que.front();
que.pop();
vec.push_back(node->val);
if (node->left) que.push(node->left);
if (node->right) que.push(node->right);
}
result.push_back(vec);
}
return result;
}
};
使用队列完成的层序遍历代码如上,思路是遍历每一个元素,遍历到一个元素时将其入队,再将其左右子节点入队,每次遍历开始前记录队列当前大小,也就是该层元素的数量,创建一个这层元素的数组进行遍历,直至遍历完该层元素下一层的元素也已经全部入队,继续进行遍历。直至队列为空时也就表明全部元素已被遍历。
class Solution {
public:
void order(TreeNode* cur, vector<vector<int>>& result, int depth)
{
if (cur == nullptr) return;
if (result.size() == depth) result.push_back(vector<int>());
result[depth].push_back(cur->val);
order(cur->left, result, depth + 1);
order(cur->right, result, depth + 1);
}
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> result;
int depth = 0;
order(root, result, depth);
return result;
}
};
递归法的层序遍历代码如上。
这十道题大同小异,就不在此赘述。
226.翻转二叉树
题目链接/文章讲解/视频讲解: 代码随想录
TreeNode* invertTree(TreeNode* root) {
if (root == NULL) return root;
swap(root->left, root->right);
invertTree(root->left);
invertTree(root->right);
return root;
}
这道题只需要交换每个节点的子节点即可实现二叉树的翻转。
要注意前后序遍历可以,中序遍历不可以。
TreeNode* invertTree(TreeNode* root) {
if (root == NULL) return root;
invertTree(root->left);
swap(root->left, root->right);
invertTree(root->left);
return root;
}
如果是中序遍历,由于遍历后不会遍历到左子树,只需要将第二次地递归的右子树改为左子树即可
101. 对称二叉树
题目链接/文章讲解/视频讲解:代码随想录
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 outside = compare(left->left, right->right); // 左子树:左、 右子树:右
bool inside = compare(left->right, right->left); // 左子树:右、 右子树:左
bool isSame = outside && inside; // 左子树:中、 右子树:中 (逻辑处理)
return isSame;
}
bool isSymmetric(TreeNode* root) {
if (root == NULL) return true;
return compare(root->left, root->right);
}
};
这道题使用了一个后序遍历的思路,先分别遍历左右节点的外侧节点和内侧节点,比较它们是否相同,在递归下去,若全都相同就说明这个二叉树是对称的。