314. 二叉樹的垂直遍歷
阿新 • • 發佈:2020-07-14
class Solution { public List<List<Integer>> verticalOrder(TreeNode root) { if(root == null) return new ArrayList<>(); Map<Integer,List<Integer>> map = new TreeMap<>(); // 存放按位置排序的list Map<TreeNode,Integer> posMap = new HashMap<>(); //存放節點的位置 posMap.put(root,0); Queue<TreeNode> queue = new LinkedList<>(); queue.add(root); while(!queue.isEmpty()) { // 層序遍歷 TreeNode node = queue.poll(); int i = posMap.get(node); map.computeIfAbsent(i,k->new ArrayList<>()).add(node.val);if(node.left != null) { queue.add(node.left); posMap.put(node.left,i-1); } if(node.right != null) { queue.add(node.right); posMap.put(node.right,i+1); } } return new ArrayList<>(map.values()); } }