层序遍历代码模版:
视频讲解:讲透二叉树的层序遍历 | 广度优先搜索 | LeetCode:102.二叉树的层序遍历_哔哩哔哩_bilibili
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
queue<TreeNode*> qu;
vector<vector<int>> result;
if(root == nullptr)return result;
qu.push(root);
while(!qu.empty()){
int size = qu.size();
vector<int> cur;
while(size--){
TreeNode* node = qu.front();
qu.pop();
cur.push_back(node->val);
if(node->left)qu.push(node->left);
if(node->right)qu.push(node->right);
}
result.push_back(cur);
}
return result;
}
};
学会二叉树的层序遍历,可以一口气打完以下十题:
- 102.二叉树的层序遍历(opens new window)
- 107.二叉树的层次遍历II(opens new window)
- 199.二叉树的右视图(opens new window)
- 637.二叉树的层平均值(opens new window)
- 429.N叉树的层序遍历(opens new window)
- 515.在每个树行中找最大值(opens new window)
- 116.填充每个节点的下一个右侧节点指针(opens new window)
- 117.填充每个节点的下一个右侧节点指针II(opens new window)
- 104.二叉树的最大深度(opens new window)
- 111.二叉树的最小深度
上面这十道题都可以用层序遍历的模版来做,无非就是在遍历的时候操作和顺序不一样。
挺简单的,前序遍历翻转左右孩子。
class Solution {
public:
TreeNode* invertTree(TreeNode* root) {
InvertTree(root);
return root;
}
void InvertTree(TreeNode* node){
if(node == nullptr) return;
swap(node->left,node->right);
InvertTree(node->left);
InvertTree(node->right);
}
};
这题有点难理解的,一开始我想用层序遍历再用双指针遍历每一层的结果,后来发现,这个方法只能判断节点的值是否对称,并不能判断节点的位置是否对称。
此题可以用前序遍历比较左子树根节点的左孩子和右孩子是否和右子树的右孩子和左孩子相等,再利用返回的结果判断两个结果是否都为true。
边界条件:左孩子为空,右孩子不为空。
右孩子为空,左孩子不为空。
左孩子的值和右孩子的值不相等。
左孩子为空,右孩子也为空。
前面四种情况只有最后一种情况能返回true,其他都会返回false。
class Solution {
public:
bool isSymmetric(TreeNode* root) {
if(root == nullptr)return true;
return compare(root->left,root->right);
}
bool compare(TreeNode* left,TreeNode* right){
if(left == nullptr && right != nullptr)return false;
else if(left != nullptr && right == nullptr)return false;
else if(left == nullptr && right == nullptr)return true;
else if(left->val != right->val)return false;
bool outside = compare(left->left,right->right);
bool inside = compare(left->right,right->left);
return outside && inside;
}
};