leetcode987. 二叉树的垂序遍历
给定二叉树,按垂序遍历返回其结点值。
对位于 (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 之间。
我的思路
创建一个对象用来存储节点和对应的x,y坐标,先遍历一遍树,把所有的节点xy坐标都求出来,然后放到list里面排序,这样排好序的list再拆成最终结果就简单了。3ms。
这里对comparator的理解:
return -1,代表return false,我不想交换当前的两个值。
return 1,代表true,我想交换当前的两个值。
先对x升序排列,再对y降序排列,最后排列val。
class Solution {
public class node{
int x;
int y;
int val;
public node(int val,int x,int y){
this.val=val;
this.x=x;
this.y=y;
}
}
public List<List<Integer>> verticalTraversal(TreeNode root) {
//对list排序,先按照y降序排序,再按照x升序排序
traverse(root,0,0);
Collections.sort(list, new Comparator<node>(){
@Override
public int compare(node o1, node o2) {
// TODO Auto-generated method stub
int a=o1.x-o2.x;
if(a>0){
return 1;
}else if(a<0){
return -1;
}else{
a=o1.y-o2.y;
if(a>0)return -1;
else if(a<0)return 1;
else {
a=o2.val-o1.val;
if(a>0)return -1;
else if(a<0)return 1;
else return 0;
}
}
}
});
/*for(node n:list)
System.out.println(n.val+" "+"x= "+n.x+" y= "+n.y);
System.out.println();*/
LinkedList<Integer>res=null;
int prev=1;//prev必须必0大,这是标记当前的x坐标有没有出现过
//因为x最小也要从0开始取值,所以prev必须必0大
//输出最终的结果
for(node n :list){
int now=n.x;
//System.out.println("val is "+n.val);
if(prev!=now){
if(res!=null)
ans.add(res);
res=new LinkedList<>();
prev=now;
}
res.add(n.val);
}
ans.add(res);
return ans;
}
public LinkedList<List<Integer>>ans=new LinkedList<>();
public LinkedList<node>list=new LinkedList<>();
public void traverse(TreeNode root,int x,int y){
if(root==null)return;
list.add(new node(root.val,x,y));
traverse(root.left,x-1,y-1);
traverse(root.right,x+1,y-1);
}
}
leetcode 34/100