C/C++教程

LeetCode 199.二叉树的右视图(DFS和层序遍历 通俗易懂)

本文主要是介绍LeetCode 199.二叉树的右视图(DFS和层序遍历 通俗易懂),对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

题目描述:

给定一个二叉树的 根节点 root,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。

输入: [1,2,3,null,5,null,4]
输出: [1,3,4]

一、DFS实现

思路:

从根部开始向右搜索,定义一个step表示层数,从0开始,再定义一个vector容器res储存每一层的最右边的元素,直到最底层的最右边元素添加到res中。

代码:

class Solution {
public:
    vector<int> res;
    vector<int> rightSideView(TreeNode* root) {
        dfs(root,0); //从根部开始
        return res;
    }

    void dfs(TreeNode* root,int step){
        if(root==nullptr){
            return ; 
        }
        if(step==res.size()){ 
            res.push_back(root->val); //每一层的最后边的元素加入到容器中
        }
        dfs(root->right,step+1); //继续向右搜索
        dfs(root->left,step+1); //右边到达叶子节点时向左边搜索
    }
};

二、层序遍历实现

1.思路:

根据层序遍历的思想,每遍历一层的最右边的元素就是右视图中的元素。

2.代码:

class Solution {
public:
    vector<int> rightSideView(TreeNode* root) {
        vector<int> res; //储存符合条件的元素
        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)){ //把每一层层序遍历的最后一个元素加入到容器中
                    res.push_back(node->val);
                }
                if (node->left!=nullptr){
                    que.push(node->left);
                }
                if (node->right!=nullptr){
                    que.push(node->right);
                } 
            }
        }
        return res;
    }
};

 

这篇关于LeetCode 199.二叉树的右视图(DFS和层序遍历 通俗易懂)的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!