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循环内部...