开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第4天,点击查看活动详情
1、写在前面
大家好,这里是【LeetCode刷题日志】。今天的两道题分别是:
- 二叉树的层序遍历 II
- 二叉树的右视图
2、内容
2.1、题目一
链接:107. 二叉树的层序遍历 II - 力扣(LeetCode)
(1) 描述
(2) 举例
(3) 解题
参考代码:
/**
* 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:
vector<vector<int>> levelOrderBottom(TreeNode* root) {
queue<TreeNode*> que; // 定义一个工作队列
if (root != NULL) que.push(root); // 如果根结点不为空,则入队
vector<vector<int>> result; // 定义二维数组来存储最终的结果
int size; // 队列大小
// 当队列为空时退出循环
while (!que.empty()) {
vector<int> v; // 定义一维数组来存储每层遍历的结果
size = que.size(); // size取到当前队列的大小
// 处理当前队列中的结点
for (int i = 0; i < size; i++) {
TreeNode* node = que.front(); // 取到队头结点
v.push_back(node->val); // 存入本层遍历的结果中
que.pop(); // 弹出队头元素
if (node->left) que.push(node->left); // 如果左结点不为空,则入队
if (node->right) que.push(node->right); // 如果右结点不为空,则入队
}
result.push_back(v); // 最后将本层循环的结果存入最终的结果result中
}
reverse(result.begin(), result.end());
return result; // 最后返回结果
}
};
2.2、题目二
链接:199. 二叉树的右视图 - 力扣(LeetCode)
(1) 描述
(2) 举例
(3) 解题
参考代码:
1)广度优先遍历
/**
* 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:
vector<int> rightSideView(TreeNode* root) {
vector<int> result; // 定义一个vector容器用于存放结果
queue<TreeNode*> que; // 定义一个工作队列
if(!root) return result; // 如果根结点为空,则返回空数组
else que.push(root); // 如果根结点不为空,则入队
// 当队列为空时退出循环
while(!que.empty()) {
// 保存当前队列的长度
int queSize = que.size();
// 遍历当前队列
for(int i = 0; i < queSize; ++i) {
TreeNode* node = que.front(); // 取到队头元素
que.pop(); // 弹出队头元素
if(node->left) que.push(node->left); // 入队左非空结点
if(node->right) que.push(node->right); // 入队右非空结点
// 如果当前元素属于本层的最后一个函数,则存入结果集中
if(i == queSize - 1) {
result.push_back(node->val);
}
}
}
return result;
}
};
2)深度优先遍历
class Solution {
public:
void dfs (TreeNode* cur, int depth, vector<int>& res) {
if(cur == NULL) return; // 如果当前结点为空,则结束本层递归
// 如果本层结点的深度不等于结果集的大小,则表示当前深度中的结点还未访问,此时第一个访问的结点就是最右边的结点
// 将该结点存入结果集中
if(depth == res.size() ) {
res.push_back(cur->val);
}
// 深度加一
depth++;
// 向右访问
dfs(cur->right, depth, res);
// 向左访问
dfs(cur->left, depth, res);
}
vector<int> rightSideView(TreeNode* root) {
vector<int> result; // 定义一个结果集
dfs(root, 0, result); // 从根节点开始访问,根节点深度是0
return result; // 最后返回结果
}
};
3、写在最后
好的,今天就先刷到这里。