要求:https://leetcode-cn.com/problems/the-skyline-problem/
思路:天际线,题目用的是左端点。
一句话抽象本题:对所有矩形的所有端点,找到那个端点处,矩形给它贡献的最大高度
我们把每一个矩阵拆成(左端点)和(右端点)这样2个元组,然后按大小排序。暴力法是对每个端点都遍历所有矩形找端点处的最大高度。注意矩形右端点高度不算,左才算,例如例1红色矩形右端点高度不算,即[left,right)
为了优化存这个最大高度,要用到大根堆,即优先队列,因为矩形本身按左端点排序,我们只需要一次遍历
class Solution {
public:
struct cmp{
bool operator()(pair<int,int> &a,pair<int,int> &b){
return a.second<b.second;
}
};
vector<vector<int>> getSkyline(vector<vector<int>>& buildings) {
priority_queue<pair<int, int>, vector<pair<int, int>>, cmp> que;
vector<int> boundaries;
for (auto& building : buildings) {
boundaries.emplace_back(building[0]);
boundaries.emplace_back(building[1]);
}
sort(boundaries.begin(), boundaries.end());
vector<vector<int>> ret;
int n = buildings.size(), idx = 0;
for (auto& boundary : boundaries) {
//左端点小于等于就入队
while (idx < n && buildings[idx][0] <= boundary) {
que.emplace(buildings[idx][1], buildings[idx][2]);
idx++;
}
//右端点小于等于的话都没用,右端点没有贡献
//不能和上一步合并,因为可能会遇到窄矩形提取退出
while (!que.empty() && que.top().first <= boundary)
que.pop();
//算最大高度
int maxn = que.empty() ? 0 : que.top().second;
//高度不能和之前一样
if (ret.size() == 0 || maxn != ret.back()[1])
ret.push_back({boundary, maxn});
}
return ret;
}
};