1. 程式人生 > 實用技巧 >314. 二叉樹的垂直遍歷

314. 二叉樹的垂直遍歷

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()); } }