代码随想录算法训练营|199. 二叉树的右视图 429.N叉树的层序遍历

199. 二叉树的右视图【中等】

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

本题的主体思想主要应用上一篇的二叉树层序遍历,只不过上两题要所以节点,而本题只需要每层的最后一个节点。所以只需要将特定节点push_back into result数组即可。

/**

* 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) {

        queue<TreeNode*> que;

        if(root != NULL) que.push(root);

        vector<int> result;

        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)que.push(node->left);

                if(node->right)que.push(node->right);

            }

        }

        return result;

    }

};

上述代码中我把题目自带的结构体也粘上了,目的是与下一题的“N叉树”的结构进行对比。

附一条讲解视频,是我之前发布在哈工大力扣讲解群当中的。(志愿时长++)(喵)

右视二叉树

429.N叉树的层序遍历【中等】

给定一个 N 叉树,返回其节点值的层序遍历。(即从左到右,逐层遍历)。树的序列化输入是用层序遍历,每组子节点都由 null 值分隔(参见示例)。

注意本题所给的树的结构,子节点children是以vector<int>形式给出的,加入队列时不再使用if(node->left)if(node->right),而采用for循环遍历vector children。其他与层序遍历思路一样喵。

/*

// Definition for a Node.

class Node {

public:

    int val;

    vector<Node*> children;

    Node() {}

    Node(int _val) {

        val = _val;

    }

    Node(int _val, vector<Node*> _children) {

        val = _val;

        children = _children;

    }

};

*/

class Solution {

public:

    vector<vector<int>> levelOrder(Node* root) {

        queue<Node*> que;

        vector<vector<int>> result;

        if(root!= NULL) que.push(root);

        while(!que.empty()){

            int size = que.size();

            vector<int> vec;

            for(int i=0;i<size;i++){

                Node* node = que.front();

                que.pop();

                for(int j=0; j<node->children.size();j++){

                    if(node->children[j]) que.push(node->children[j]);

                }

                vec.push_back(node->val);

            }  

            result.push_back(vec);

        }

        return result;

    }

};

注意result.push_back(vec);要写在while循环内部...

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值