【LeetCode刷题日志】:二叉树的层序遍历 II、二叉树的右视图

70 阅读3分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第4天,点击查看活动详情

1、写在前面

大家好,这里是【LeetCode刷题日志】。今天的两道题分别是:

  • 二叉树的层序遍历 II
  • 二叉树的右视图

2、内容

2.1、题目一

链接:107. 二叉树的层序遍历 II - 力扣(LeetCode)

(1) 描述

image.png

(2) 举例

image.png

image.png

image.png

(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) 描述

image.png

(2) 举例

image.png

image.png

image.png

(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、写在最后

好的,今天就先刷到这里。