leetcode 199 从右看二叉树

面试的时候不会写，只写完了层次遍历，挂了，Orz。

 

思路：

1.层次遍历，bfs。

2.遇到新层的第一个节点才会添加新层。

3.每层从右往左遍历，则最右的始终是每层的第一个。

4.结合2，3。当需要添加新层时，将节点值添加进结果集。

 

用到的数据结构：

1.队列，这里用双向Deque。

2.线性表，存结果。

3.Map，记录节点对应的level。

 

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<Integer> rightSideView(TreeNode root) {
        Deque<TreeNode> dq = new ArrayDeque<TreeNode>();
        List<Integer> ans = new ArrayList<Integer>();
        Map<TreeNode, Integer> level = new HashMap<TreeNode, Integer>();
        if (root != null){
            dq.addLast(root);
            level.put(root, 1);
            ans.add(root.val);
        }
        while (!dq.isEmpty()) {
            TreeNode top = dq.getFirst();
            dq.removeFirst();
            if (top.right != null) {
                dq.addLast(top.right);
                level.put(top.right, level.get(top) + 1);
                if (ans.size() < level.get(top) + 1) {
                    ans.add(top.right.val);
                }
            }
            if (top.left != null) {
                dq.addLast(top.left);
                level.put(top.left, level.get(top) + 1);
                if (ans.size() < level.get(top) + 1) {
                    ans.add(top.left.val);
                }
            }
        }
        return ans;
    }
}

---

 

4月5日更新。

一种效率更高，更简洁的版本。

这种写法确保每次while结束后队列里的元素永远是下一层的所有结点（或者没有结点）。

学习到了，原来用一个内置的for循环就能完成。

 

class Solution {
    public List<Integer> rightSideView(TreeNode root) {
        List<Integer> ans = new ArrayList<>();
        if (root == null)
            return ans;
        Deque<TreeNode> dq = new ArrayDeque<>();
        dq.addLast(root);
        while (!dq.isEmpty()) {
            //从根所在的level逐层更新
            int num = dq.size();
            for (int i = 0; i < num; i++) {
                TreeNode tmp = dq.getFirst();
                if (tmp.left != null)
                    dq.addLast(tmp.left);
                if (tmp.right != null)
                    dq.addLast(tmp.right);
                if (i == num - 1)
                    ans.add(tmp.val);
                dq.removeFirst();
            }
        }
        return ans;
    }
}