1. 程式人生 > 其它 >218. 天際線問題 set 掃描線

218. 天際線問題 set 掃描線

------------恢復內容開始------------
城市的天際線是從遠處觀看該城市中所有建築物形成的輪廓的外部輪廓。給你所有建築物的位置和高度,請返回由這些建築物形成的 天際線 。

每個建築物的幾何資訊由陣列 buildings 表示,其中三元組 buildings[i] = [lefti, righti, heighti] 表示:

lefti 是第 i 座建築物左邊緣的 x 座標。
righti 是第 i 座建築物右邊緣的 x 座標。
heighti 是第 i 座建築物的高度。
天際線 應該表示為由 “關鍵點” 組成的列表,格式 [[x1,y1],[x2,y2],...] ,並按 x 座標 進行 排序 。關鍵點是水平線段的左端點。列表中最後一個點是最右側建築物的終點,y 座標始終為 0 ,僅用於標記天際線的終點。此外,任何兩個相鄰建築物之間的地面都應被視為天際線輪廓的一部分。

注意:輸出天際線中不得有連續的相同高度的水平線。例如 [...[2 3], [4 5], [7 5], [11 5], [12 7]...] 是不正確的答案;三條高度為 5 的線應該在最終輸出中合併為一個:[...[2 3], [4 5], [12 7], ...]

來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/the-skyline-problem
著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。

class Solution {
public:
    vector<vector<int>> getSkyline(vector<vector<int>>& buildings) {
        multiset <pair<int, int>> all;
        vector<vector<int>> ans;

        // multiset 按照座標排序,高度<0表示左端點,高度>0表示右端點
        for (auto b : buildings) {
            all.insert(make_pair(b[0], -b[2]));
            all.insert(make_pair(b[1], b[2]));
        }

        // 儲存當前位置的所有建築物的高度,因為都是左端點,高度是負數,所以rbegin表示最高高度
        multiset<int> h({0});
        vector<int> last = {0, 0};

        for (auto p : all) {
            // 左端點,高度入堆
            if (p.second < 0) {
                h.insert(-p.second);
            }
            // 右端點,移除高度
            else {
                h.erase(h.find(p.second));
            }

            // 當前關鍵點的最大高度
            auto maxH = *h.rbegin();

            // 如果當前最大高度和上一個高度不同,說明當前點事轉折點
            if (last[1] != maxH) {
                last[0] = p.first;
                last[1] = maxH;
                ans.push_back(last);
            }
        }

        return ans;
    }
};