102.二叉树层序遍历
题目
给你二叉树的根节点 root
,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。
解题思路
- 需要借用队列实现,队列先进先出,符合一层一层遍历的逻辑
代码
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> result;
queue<TreeNode*> que;
if (root != nullptr) que.push(root);
while(!que.empty()) {
int size = que.size();
vector<int> temp;
for(int i = 0; i < size; i++) {
TreeNode* node = que.front();
temp.push_back(node->val);
que.pop();
if (node->left != nullptr) que.push(node->left);
if (node->right != nullptr) que.push(node->right);
}
result.push_back(temp);
}
return result;
}
};
199.二叉树的右视图
题目
给定一个二叉树的 根节点 root,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
解题思路
- 题目意思和立体几何三视图中的左视图是类似的概念,只不过这个是右视图
- 和102题目思路差不多,只需要增加个条件,在遍历到每层最后一个的时候,加入到vector中,即遍历条件为
i==size-1
代码
class Solution {
public:
vector<int> rightSideView(TreeNode* root) {
vector<int> result;
queue<TreeNode*> que;
if (root != nullptr) que.push(root);
while(!que.empty()) {
int size = que.size();
for (int i = 0; i < size; i++) {
TreeNode* node = que.front();
que.pop();
if (i == (size -1)) result.push_back(node->val);
if (node->left != nullptr) que.push(node->left);
if (node->right != nullptr) que.push(node->right);
}
}
return result;
}
};
637.二叉树的层平均值
题目
给定一个非空二叉树的根节点 root , 以数组的形式返回每一层节点的平均值。与实际答案相差 10-5 以内的答案可以被接受。
解题思路
- 使用层序遍历之后,记录每层的所有节点值相加总和,再除以size即可
代码
class Solution {
public:
vector<double> averageOfLevels(TreeNode* root) {
vector<double> result;
queue<TreeNode*> que;
if (root != nullptr) que.push(root);
while(!que.empty()) {
int size = que.size();
double sum = 0;
for (int i = 0; i < size; i++) {
TreeNode* node = que.front();
que.pop();
sum += node->val;
if (node->left != nullptr) que.push(node->left);
if (node->right != nullptr) que.push(node->right);
}
result.push_back(sum/size);
}
return result;
}
};
429. N 叉树的层序遍历
题目
给定一个 N 叉树,返回其节点值的层序遍历。(即从左到右,逐层遍历)。
解题思路
- 层序遍历的主题思路不变
- 二叉树是left right 两个子节点,N叉树N个子节点,for循环遍历N个节点即可
代码
class Solution {
public:
vector<vector<int>> levelOrder(Node* root) {
vector<vector<int>> result;
queue<Node*> que;
if (root != nullptr) que.push(root);
while(!que.empty()) {
int size = que.size();
vector<int> temp;
for(int i = 0; i < size; i++) {
Node* node = que.front();
temp.push_back(node->val);
que.pop();
for(int j = 0; j < node->children.size(); j++) {
if (node->children[j]) que.push(node->children[j]);
}
}
result.push_back(temp);
}
return result;
}
};
515.在每个树行中找最大值
题目
给定一棵二叉树的根节点 root ,请找出该二叉树中每一层的最大值。
解题思路
- 其实这些题的核心的都是层序遍历,遍历每一个节点,都比大小就行
- 比如:先记录第一个节点为最大值,到第二个节点的时候,跟当前最大值比大小,然后更新最大值即可
代码
class Solution {
public:
vector<int> largestValues(TreeNode* root) {
vector<int> result;
queue<TreeNode*> que;
if (root != nullptr) que.push(root);
while(!que.empty()) {
int size = que.size();
int maxVal = INT_MIN;
for(int i = 0; i < size; i++) {
TreeNode* node = que.front();
que.pop();
maxVal = node->val > maxVal ? node->val : maxVal;
if (node->left != nullptr) que.push(node->left);
if (node->right != nullptr) que.push(node->right);
}
result.push_back(maxVal);
}
return result;
}
};
116.填充每个节点的下一个右侧节点指针
题目
给定一个完美二叉树,其所有叶子节点都在同一层,每个父节点都有两个子节点。二叉树定义如下:
struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
解题思路
- 仍然是层序遍历,让每一层的节点next指针都指向该层的下一个节点,如果遍历到最后一个节点,则将next赋值为空
- 这样需要记录下来,相邻的两个节点,使用两个指针即可
代码
class Solution {
public:
Node* connect(Node* root) {
queue<Node*> que;
if (root != NULL) que.push(root);
while (!que.empty()) {
int size = que.size();
// vector<int> vec;
Node* nodePre;
Node* node;
for (int i = 0; i < size; i++) {
if (i == 0) {
nodePre = que.front(); // 取出一层的头结点
que.pop();
node = nodePre;
} else {
node = que.front();
que.pop();
nodePre->next = node; // 本层前一个节点next指向本节点
nodePre = nodePre->next;
}
if (node->left) que.push(node->left);
if (node->right) que.push(node->right);
}
nodePre->next = NULL; // 本层最后一个节点指向NULL
}
return root;
}
};