Leetcode - The Skyline Problem

My code:

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

public class Solution {
    private class Edge {
        int height;
        int x;
        boolean isStart;
        
        Edge(int x, int height, boolean isStart) {
            this.x = x;
            this.height = height;
            this.isStart = isStart;
        }
    }
    
    
    public List<int[]> getSkyline(int[][] buildings) {
        List<int[]> ret = new ArrayList<int[]>();
        if (buildings == null || buildings.length == 0 || buildings[0].length == 0) {
            return ret;
        }
        
        List<Edge> edges = new ArrayList<Edge>();
        for (int i = 0; i < buildings.length; i++) {
            edges.add(new Edge(buildings[i][0], buildings[i][2], true));
            edges.add(new Edge(buildings[i][1], buildings[i][2], false));
        }
        
        Collections.sort(edges, new Comparator<Edge>() {
            public int compare(Edge e1, Edge e2) {
                if (e1.x != e2.x) {
                    return e1.x - e2.x;
                }
                else if (e1.isStart && e2.isStart) {
                    return e2.height - e1.height;
                }
                else if (!e1.isStart && !e2.isStart) {
                    return e1.height - e2.height;
                }
                else {
                    return e1.isStart ? -1 : 1;
                }
            }    
        });
        
        PriorityQueue<Integer> pq = new PriorityQueue<Integer>(10, new Comparator<Integer>() {
            public int compare(Integer e1, Integer e2) {
                return e2.compareTo(e1);
            }    
        });
        
        for (Edge edge : edges) {
            if (edge.isStart) {
                if (pq.isEmpty() || edge.height > pq.peek()) {
                    ret.add(new int[]{edge.x, edge.height});
                }
                pq.offer(edge.height);
            }
            else {
                pq.remove(edge.height);
                if (pq.isEmpty()) {
                    ret.add(new int[]{edge.x, 0});
                }
                else if (edge.height > pq.peek()) {
                    ret.add(new int[]{edge.x, pq.peek()});
                }
            }
        }
        
        return ret;
    }
}

reference:
http://www.programcreek.com/2014/06/leetcode-the-skyline-problem-java/

这道题目看了将近一个半小时。终于基本理解了在做什么。
首先, skyline 的点分为两种:
升序时候的 skyline 点 只可能是左侧点
降序时候的 skyline 点 只可能是右侧点

所以每当我们遇到一个更高的左侧点时,他一定是一个 skyline 点
每当我们遇到一个右侧点,高度比当前大楼都高时,就意味着要开始下降了,他也一定是一个 skyline 点。
所以我们要有一个最大堆,用来存放当前大楼的高度。top就是当前大楼的最大高度。

我们有个类 Edge,表示点,以及这个点是左侧点还是右侧点
然后我们把所有的输入转换成这些点,一个大楼有两个点。
然后排序。遵守的原则是
1 . 如果 e1.x != e2.x
那么谁小谁放前面。
2 . 如果相等
那么就有讲究了。
如果同时左边点。应该先遍历哪个呢?
如果把height 矮一些的放前面,那么这个点就会变成一个 skyline 点
然后height 更高的又出现了,又会有一个skyline点,在刚刚那个点的垂直上方。这是不对的。
所以先把 height 高一些的放前面。这样,矮一些的就不会产生skyline 点

如果同时右边点。
如果把高的放前面,那么高的会在矮的高度处产生一个skyline点
然后矮的在地平线处产生一个 skyline 点。这是不对的。
所以矮的放前面,因为高的仍然在堆里,所以矮的无法产生skyline点,然后再轮到高的,在地平线处产生一个skyline 点。

如果一个是左边点,一个是右边点。
如果把右边点放在前面,会在地平线处产生一个skyline 点,然后再进入左边点,于是在左边点处产生一个 skyline 点。于是一个垂直方向上,同时出现了两个点,是不对的。
如果把左边点放在前面,如果右边点高度比他高,那么左边点不会产生skyline 点,然后右边点产生了skyline 点,下降。
如果右边点比他低,那么左边点产生一个skyline 点,上升。然后右边点发现当前top的高度比他高,无法产生skyline 点。

Anyway, Good luck, Richardo! — 09/29/2016

    原文作者:Richardo92
    原文地址: https://www.jianshu.com/p/ef44e79462e8
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞