JAVA程序设计：二叉树的垂序遍历（LeetCode：987）

给定二叉树，按垂序遍历返回其结点值。

对位于 (X, Y) 的每个结点而言，其左右子结点分别位于 (X-1, Y-1) 和 (X+1, Y-1)。

把一条垂线从 X = -infinity 移动到 X = +infinity ，每当该垂线与结点接触时，我们按从上到下的顺序报告结点的值（ Y 坐标递减）。

如果两个结点位置相同，则首先报告的结点值较小。

按 X 坐标顺序返回非空报告的列表。每个报告都有一个结点值列表。

 

示例 1：

输入：[3,9,20,null,null,15,7]
输出：[[9],[3,15],[20],[7]]
解释： 
在不丧失其普遍性的情况下，我们可以假设根结点位于 (0, 0)：
然后，值为 9 的结点出现在 (-1, -1)；
值为 3 和 15 的两个结点分别出现在 (0, 0) 和 (0, -2)；
值为 20 的结点出现在 (1, -1)；
值为 7 的结点出现在 (2, -2)。
示例 2：

输入：[1,2,3,4,5,6,7]
输出：[[4],[2],[1,5,6],[3],[7]]
解释：
根据给定的方案，值为 5 和 6 的两个结点出现在同一位置。
然而，在报告 "[1,5,6]" 中，结点值 5 排在前面，因为 5 小于 6。
 

提示：

树的结点数介于 1 和 1000 之间。
每个结点值介于 0 和 1000 之间。

思路：我们通过遍历整棵树，按照每个节点的坐标存在list里，之后对于横坐标相同的节点我们按照题意排序即可，为了使左边不出现负值，我们考虑将根节点定为(1000,1000)。

class Solution {
	
	private List<List<int[]>> list=new ArrayList<>();
	
    public List<List<Integer>> verticalTraversal(TreeNode root) {
    	
    	list=new ArrayList<>();
    	List<List<Integer>> ans=new ArrayList<>();
    	
    	for(int i=0;i<=2000;i++)
    		list.add(new ArrayList<>());
    	
    	dfs(root,1000,1000);
    	
    	for(int i=0;i<=2000;i++) {
    		if(list.get(i).size()==0)
    			continue;
    		Collections.sort(list.get(i), (a,b)->a[0]==b[0]?a[1]-b[1]:b[0]-a[0]);
    		int index=ans.size();
    		ans.add(new ArrayList<>());
    		for(int j=0;j<list.get(i).size();j++)
    			ans.get(index).add(list.get(i).get(j)[1]);
    	}
    	
    	return ans;
    	
    }
    
    private void dfs(TreeNode root,int x,int y) {
    	
    	if(root==null) return;
    	
    	list.get(x).add(new int[] {y,root.val});
    	
    	dfs(root.left,x-1,y-1);
    	dfs(root.right,x+1,y-1);
    	
    }
}