LeetCode199
问题
给定一个二叉树的 根节点 root,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
思路
相当于二叉树的按层级遍历的进阶,我们按照之前文章写过的二叉树层级遍历:https://blog.csdn.net/C_lea/article/details/152413818?spm=1001.2014.3001.5501
只要我们在每一层循环的时候,判断一下当前循环的是不是最后一个节点即可,只要是最后一个节点,那么就把结果加入进去
for (int i = 0;i < n;i ++) {TreeNode t = queue.poll();//...if (i == n - 1) {result.add(t.val); //如果是当前层的最后一个节点,就相当于是右视图,那么加入结果}
}
代码
public List<Integer> rightSideView(TreeNode root) {List<Integer> result = new ArrayList<>();Queue<TreeNode> queue = new LinkedList<>();if (root == null) {return result;}queue.offer(root);while (!queue.isEmpty()) {int n = queue.size();for (int i = 0;i < n;i ++) {TreeNode t = queue.poll();if (t.left != null) {queue.offer(t.left);}if (t.right != null) {queue.offer(t.right);}if (i == n - 1) {result.add(t.val);}}}return result;
}