題目

https://leetcode-cn.com/problems/the-skyline-problem/

題解

線段樹問題，根據左神的思路改編，外加我想到的壓縮的 tricks（數字范圍太大，實在想不出別的辦法了，一開始是直接把過大的數除以某個 scale，但是這樣是有損的，無法復原，于是想到了用 map 存壓縮前后的映射關系。只要保持壓縮前后的數字大小順序不變，這個壓縮就是無損的。）

class Solution {public static final int N = 1 << 15; // 2^15=32768，考慮到 buildings.length <= 10^4，壓縮后總共不超過2*10^4個橫坐標public List<List<Integer>> getSkyline(int[][] buildings) {SegmentTree segmentTree = new SegmentTree(N);// 壓縮（編碼）// 排序，然后映射，因為不超過10000，所以肯定能映射開Set<Integer> set = new TreeSet<>();for (int[] task : buildings) {set.add(task[0]);set.add(task[1]);}Map<Integer, Integer> decodeMap = new HashMap<>();Map<Integer, Integer> encodeMap = new HashMap<>();int cipher = 0;for (int plain : set) {cipher += 2; // 用+2而不是+1，目的是留出地面的間隔，避免兩個樓挨在一起時，丟失了地面decodeMap.put(cipher, plain);encodeMap.put(plain, cipher);}for (int[] task : buildings) {task[0] = encodeMap.get(task[0]);task[1] = encodeMap.get(task[1]);}// 更新線段樹（核心）for (int[] task : buildings) {segmentTree.update(task[0], task[1], task[2], 0, N, 1); // 把從task[0]到task[1]的位置更新為task[2]}segmentTree.flush();// 打補丁（之前為了方便下標索引，線段樹是從1位置開始構建的，現在要把0位置也考慮進去）segmentTree.max[N - 2] = 0;segmentTree.max[2 * N - 1] = 0;for (int[] task : buildings) {if (task[0] == 0) segmentTree.max[N - 1] = Math.max(segmentTree.max[N - 1], task[2]);}// 構造返回值（遍歷所有葉子）List<List<Integer>> result = new ArrayList<>();for (int i = N - 1; i < 2 * N; i++) {if (segmentTree.max[i] != segmentTree.max[i - 1]) {ArrayList<Integer> list = new ArrayList<>();list.add(segmentTree.max[i - 1] < segmentTree.max[i] ? i - N + 1 : i - N);list.add(segmentTree.max[i]);result.add(list);}}// 解碼for (List<Integer> list : result) {list.set(0, decodeMap.get(list.get(0)));}return result;}public static class SegmentTree {private int SIZE;private int[] max;private int[] change;private boolean[] update;public SegmentTree(int N) {SIZE = N + 1;max = new int[SIZE << 2]; // 用來支持腦補概念中，某一個范圍的累加和信息change = new int[SIZE << 2]; // 用來支持腦補概念中，某一個范圍有沒有更新操作的任務update = new boolean[SIZE << 2]; // 用來支持腦補概念中，某一個范圍是否需要更新（而不是更新成0）}// 之前的所有懶增加、懶更新，從父范圍發給左右兩個子范圍，分發策略是什么private void pushDown(int rt) {if (update[rt]) {update[rt << 1] = true;update[rt << 1 | 1] = true;change[rt << 1] = change[rt];change[rt << 1 | 1] = change[rt];max[rt << 1] = Math.max(max[rt << 1], change[rt]);max[rt << 1 | 1] = Math.max(max[rt << 1 | 1], change[rt]);update[rt] = false;}}// 想要把 L~R 所有的值變成 H// 當前影響 l~r，當前范圍信息存在數組的 rt 位置public void update(int L, int R, int H, int l, int r, int rt) {if (L <= l && r <= R) {if (H >= max[rt]) {update[rt] = true;change[rt] = H;max[rt] = H;}return;}// 當前任務躲不掉，無法懶更新，要往下發一層int mid = (l + r) >> 1;pushDown(rt);if (L <= mid) {update(L, R, H, l, mid, rt << 1);}if (R > mid) {update(L, R, H, mid + 1, r, rt << 1 | 1);}}// 把所有的懶更新都下放到底部，類似于自底向上的堆排序public void flush() {for (int i = max.length - 1; i >= 0; i--) {if (max[i] == 0) heapify(i);}}public int heapify(int i) {if (i == 0) return 0;max[i] = Math.max(max[i], heapify(i / 2));return max[i];}} }

總結

以上是生活随笔為你收集整理的leetcode 218. The Skyline Problem | 218. 天际线问题（线段树）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。