思路:参考 剑指 Offer 32 – II. 从上到下打印二叉树,获取到的list中,每个list的最后一个元素即为右视图
复杂度:时间复杂度O(n),空间复杂度O(n)
class Solution {
public List<Integer> rightSideView(TreeNode root) {
List<Integer> list = new ArrayList<>();
if (root == null) {
return list;
}
List<List<Integer>> levelList = levelOrder(root);
for (List<Integer> subList : levelList) {
list.add(subList.get(subList.size() - 1));
}
return list;
}
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> list = new ArrayList<>();
if (root == null) {
return list;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
while (!queue.isEmpty()) {
int count = queue.size();// 本层节点数
List<Integer> tmpList = new ArrayList<>();
while (count-- > 0) {
TreeNode node = queue.poll();
tmpList.add(node.val);
if (node.left != null) {
queue.add(node.left);
}
if (node.right != null) {
queue.add(node.right);
}
}
list.add(tmpList);
}
return list;
}
}
看别人的题解,直接层序遍历,每层最后一个判断下加入到list即可
class Solution {
public List<Integer> rightSideView(TreeNode root) {
List<Integer> list = new ArrayList<>();
if (root == null) {
return list;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while (!queue.isEmpty()) {
int size = queue.size();
for (int i = 0; i < size; i++) {
TreeNode node = queue.poll();
if (node.left != null) {
queue.offer(node.left);
}
if (node.right != null) {
queue.offer(node.right);
}
if (i == size - 1) {
list.add(node.val);
}
}
}
}
}
class Solution {
public List<Integer> rightSideView(TreeNode root) {
List<Integer> res = new ArrayList<Integer>();
if(root==null){
return res;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
TreeNode node;
while (!queue.isEmpty()){
int n = queue.size();
for (int i = 0; i < n; i++) {
node = queue.poll();
if (node.left!=null){
queue.add(node.left);
}
if (node.right!=null){
queue.add(node.right);
}
if(i==n-1){
res.add(node.val);
}
}
}
return res;
}
}
共以下 10 个回答
层次遍历,每次只取队列最后一个
看别人的题解,直接层序遍历,每层最后一个判断下加入到list即可
层序保存每层最后一个
1、先序遍历变体,根->右->左
2、层序遍历,把每层最后一个存进去
按层遍历,保存每层最后一个值
O(n)