987. 二叉树的垂序遍历

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

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/vertical-order-traversal-of-a-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;

class Solution {

    List<Info> nodes = new ArrayList<>();

    public List<List<Integer>> verticalTraversal(TreeNode root) {
        dfs(root, 0, 0);
        Collections.sort(nodes, new Comparator<Info>() {
            @Override
            public int compare(Info o1, Info o2) {
                if (o1.col != o2.col) {
                    return Integer.compare(o1.col, o2.col);
                } else if (o1.row != o2.row) {
                    return Integer.compare(o1.row, o2.row);
                } else {
                    return Integer.compare(o1.val, o2.val);
                }
            }
        });
        List<List<Integer>> ans = new ArrayList<>();
        int lastCol = Integer.MIN_VALUE;
        for (Info node : nodes) {
            if (node.col != lastCol) {
                lastCol = node.col;
                ans.add(new ArrayList<>());
            }
            ans.get(ans.size() - 1).add(node.val);
        }
        return ans;
    }

    public void dfs(TreeNode node, int row, int col) {
        if (node == null) {
            return;
        }
        nodes.add(new Info(row, col, node.val));
        dfs(node.left, row + 1, col - 1);
        dfs(node.right, row + 1, col + 1);
    }
}

class Info {
    int row;
    int col;
    int val;

    public Info(int row, int col, int val) {
        this.row = row;
        this.col = col;
        this.val = val;
    }
}

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