LeetCode987—二叉树的垂序遍历（java版）

题目描述：

标签：树  深度优先搜索  广度优先搜索  哈希表  二叉树

给你二叉树的根结点 root ，请你设计算法计算二叉树的 垂序遍历 序列。

对位于 (row, col) 的每个结点而言，其左右子结点分别位于 (row + 1, col - 1) 和 (row + 1, col + 1) 。树的根结点位于 (0, 0) 。

二叉树的 垂序遍历 从最左边的列开始直到最右边的列结束，按列索引每一列上的所有结点，形成一个按出现位置从上到下排序的有序列表。如果同行同列上有多个结点，则按结点的值从小到大进行排序。

返回二叉树的 垂序遍历 序列。

代码：

 思路分析：

总得来说，题目是想先知道各个节点的col,row,val值，然后先按照列col排序，从左到右；接着按照行row排序，从上到下；最后按照值val排列，从小到大。

1、定义一个哈希表map，key是TreeNode，value是new int[]{col, row, val}

2、使用深度优先搜索dfs将各个节点的这三个关键值col,row,val存放在map中

3、取出map的values存放到List<int []> list中，然后使用Collections.sort(list,(a,b) -> {z自定义排序规则})对list按照先列col后行row最后值val的顺序排列。

4、循环遍历list，判断同一列是否有多个节点，有的话先保存在一个List中，最后该列结束后再添加到最终结果中。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    Map<TreeNode, int[]> map = new HashMap<>();
    public List<List<Integer>> verticalTraversal(TreeNode root) {
        map.put(root, new int[]{0, 0, root.val});
        dfs(root);
        List<int []> list = new ArrayList<>(map.values());
        Collections.sort(list, (a,b) -> {
            if(a[0] != b[0]) return a[0] - b[0];
            if(a[1] != b[1]) return a[1] - b[1];
            return a[2] - b[2];
        });
        List<List<Integer>> ans = new ArrayList<>();
        int n = list.size();
        for(int i = 0;i < n; ){
            int j = i;
            List<Integer> tmp = new ArrayList<>();
            while(j < n && list.get(j)[0] == list.get(i)[0]){
                tmp.add(list.get(j++)[2]);
            }
            ans.add(tmp); 
            i = j;
        }
        return ans;
    }

    public void dfs(TreeNode node){
        if(node == null){
            return;
        }
        int[] info = map.get(node);
        int col = info[0];
        int row = info[1];
        int val = info[2];
        if(node.left != null){
            map.put(node.left, new int[]{col - 1, row + 1, node.left.val});
            dfs(node.left);
        }
        if(node.right != null){
            map.put(node.right, new int[]{col + 1, row + 1, node.right.val});
            dfs(node.right);
        }
    }
}