为了加深今天的学习理解写了这篇文章,试图对扫描线算法的原理进行直观化,如有不对之处还请指正
举Leetcode 218题为例子
附上大佬的代码:
class Solution {
public:
vector<vector<int>> getSkyline(vector<vector<int>>& buildings) {
vector<vector<int>> res;
vector<pair<int, int>> height;
for (auto &b : buildings) {
// 正负用于判别是左端点还是右端点,同时保证排序后:
// 左端点相同时,最高的楼排在前面,insert的一定是相同左端点中最大的高度
// 右端点相同时,最低的楼排在前面,erase的时候不会改变最大高度
height.push_back({b[0], -b[2]}); // 左端点
height.push_back({b[1], b[2]}); // 右端点
}
sort(height.begin(), height.end());
// 维护当前最大高度
multiset<int> st;
st.insert(0); // 保证端点全部删除之后能得到当前最大高度为 0
// preMaxHeight 表示遇到一个端点之前的最大高度
// curMaxHeight 表示遇到一个端点之后的当前最大高度
int preMaxHeight = 0, curMaxHeight = 0;
for (auto &h : height) {
if (h.second < 0) { // 左端点
st.insert(-h.second);
} else { // 右端点
st.erase(st.find(h.second));
}
curMaxHeight = *st.rbegin();
// 最大高度发生改变,一定是一个 key point,即一个水平线段的左端点
if (curMaxHeight != preMaxHeight) {
res.push_back({h.first, curMaxHeight});
preMaxHeight = curMaxHeight;
}
}
return res;
}
};
作者:阿晨
链接:https://leetcode.cn/problems/the-skyline-problem/description/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
为什么选择这道题?
在我看来,扫描线算法关注的是入边和出边的差值 对这个题来说即楼的高度和地面的差
地面是固定的 没有出现出边变动的情况 相当于入边减一个常数 所以我们只用关注入边的值 即楼的高度就好 是一个相对简单的情况
代码在做什么?
我心中有一种形象化的说法,就是马里奥从大楼的左边一直走到右边,测量他每时每刻的高度与地面的插值,来判断这个点是不是关键点
先从地面跳上大楼的最左端
if (h.second < 0) { // 左端点
st.insert(-h.second);
}
更新高度,并获得结果之一
if (curMaxHeight != preMaxHeight) {
res.push_back({h.first, curMaxHeight});
preMaxHeight = curMaxHeight;
}
获取此时的高度
curMaxHeight = *st.rbegin();
前进直到被挡住,马里奥发现不能前进了,于是跳跃到更高的平面上,并更新自己的高度
curMaxHeight = *st.rbegin();
// 最大高度发生改变,一定是一个 key point,即一个水平线段的左端点
if (curMaxHeight != preMaxHeight) {
res.push_back({h.first, curMaxHeight});
preMaxHeight = curMaxHeight;
}
而与之相对应的另一种情况,马里奥前进到这个位置的时候,前面没有阻碍,他选择继续前进(此时curMaxHeight==preMaxHeight,因此只发生绿色方块的左边界被加入优先队列)
马里奥前进到红色方块最右端的时候,此时前面没有路了,他选择跳下来,此时最大高度发生更新
到达红色方块的右端点,红色方块对应的最大高度被提出优先队列,导致最大高度发生更新,我们获得一个答案
else { // 右端点
st.erase(st.find(h.second));
}
curMaxHeight = *st.rbegin();
当马里奥到达绿色方块边缘的时候,前面没有路了,同理,他跳下来落到地上,此时高度变为0,我们获得最后一个答案
我们关注每次跳跃的目的地,即得到结果
简单的结语
扫描线算法关注的是入边与出边的差值是否变化,因此他需要扫描每一个可能会发生变化的关键节点,如果插值发生更新,我们对差值或节点进行处理即可