POJ 1151 Atlantis【扫描线】

题目链接

题目描述:

扫描线模板题：给一些矩形的左下和右上坐标，求这些矩形面积的并集。

---

题解:

扫描线朴素算法 $O(n^2)$：

设想有这样一条竖直(或水平)无限长的线，从左往右扫，扫的时候逐渐累加面积。
我们将每个矩形用一个四元组$<x,y_1,y_2,k>$表示，即表示矩形的两条对边，设矩形的左边那条边$k=1$，右边那条边$k=-1$。设某一时刻扫描线与矩形并集最外面的两个交点之间的距离为len，则存在一个区间$[x_1,x_2]$上两个交点的距离一直是len，扫描线从$x_1$->$x_2$这段累加的面积为$(x_2-x_1)\ast len$，$x_1$$和x_2$可以由排序得知先后顺序，在此基础上需要求len，对于不同区间$[x1,x2]$的$len$，都可能有不同的值，所以我们需要不断对$len$进行更新，需要对$len$进行分段，设$cnt$数组表示每一段存在的数量，如果有一段被$k$更新了大于0，则将这一段长度加入$len$，分段需要利用所有$y$坐标，最多可能有$2\ast n$个不同的$y$值，分段的方法是离散化。所以求解过程如下：对于每一个四元组，我们首先将$y_1$到$y_2$区间这个区间上的段都用$k$更新$cnt$数组，若某一段$cnt$大于0则将这一段的长度加入$len$，否则则从$len$中减去这一段的长度，然后利用下一个四元组的$x_2$减去当前四元组的$x_1$得出$x$区间的长度，将区间长度乘上$len$累加到答案中。

线段树优化$O(nlo{g}_{2}n)$：

在朴素算法中，我们用到了分段的思想，自然就会想到线段树，所以线段树维护两个值：$len$为当前节点区间上有效区间长度，$cnt$为该区间被完全覆盖次数。此时根节点 $t[1].len$ 就代表存在的区间总长度，每次用y1, y2离散化后的区间$[l,r]$带k更新线段树，l和r都是点，而我们更新的是区间，区间数比点数少1，所以我们让每个点更新它右边的区间，最后一个点谁也不更新，所以是update(l, r - 1)。在push_up时，若当前结点cnt > 0，则直接用val[r + 1] - val[l]更新len，否则若l != r，则它的len为两个儿子结点的len之和，否则则说明l == r并且cnt不大于0，此时表示该叶子结点所表示的右边区间已经不存在了，需要将len重置为0，以此来向上更新。

---

$O(nlo{g}_{2}n)$代码:

#include <iostream>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
//#include <unordered_set>
//#include <unordered_map>
#include <deque>
#include <list>
#include <iomanip>
#include <algorithm>
#include <fstream>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <cstdlib>
//#pragma GCC optimize(2)
using namespace std;
typedef long long ll;
//cout << fixed << setprecision(2);
//cout << setw(2);
const int N = 2e2 + 5, M = 1e9 + 7;

struct Node {
    int cnt;
    double len;
} t[N << 2];

struct Line {
    int k;
    double x, y1, y2;
    bool operator < (const Line& other) const {
        return x < other.x;
    }
} line[N];
double a[N << 2];

void push_up(int x, int l, int r) {
    if (t[x].cnt > 0) {
        t[x].len = a[r + 1] - a[l];
    } else if (l != r) {
        t[x].len = t[x << 1].len + t[x << 1 | 1].len;
    } else { //此处要注意！
        t[x].len = 0;
    }
}

void update(int x, int l, int r, int p, int q, int k) {
    if (p <= l && q >= r) {
        t[x].cnt += k;
        push_up(x, l, r);
        return;
    }
    int mid = (l + r) >> 1;
    if (p <= mid) update(x << 1, l, mid, p, q, k);
    if (q > mid) update(x << 1 | 1, mid + 1, r, p, q, k);
    push_up(x, l, r);
}

int main() {
    //freopen("/Users/xumingfei/Desktop/ACM/test.txt", "r", stdin);
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n, T = 1;
    while (cin >> n && n) {
        int cnt = 0;
        double x, y1, y2;
        int k;
        memset(t, 0, sizeof(t));
        for (int i = 0, m = 2 * n; i < m; i += 2) {
            cin >> line[i].x >> line[i].y1 >> line[i + 1].x >> line[i].y2;
            line[i + 1].y1 = line[i].y1, line[i + 1].y2 = line[i].y2;
            line[i].k = 1, line[i + 1].k = -1;
            a[cnt++] = line[i].y1, a[cnt++] = line[i].y2;
        }
        sort(a, a + cnt);
        sort(line, line + 2 * n);
        cnt = unique(a, a + cnt) - a;
        double ans = 0;
        for (int i = 0, m = 2 * n; i < m; i++) {
            int l = lower_bound(a, a + cnt, line[i].y1) - a;
            int r = lower_bound(a, a + cnt, line[i].y2) - a;
            update(1, 0, cnt - 1, l, r - 1, line[i].k);
            ans += t[1].len * (line[i + 1].x - line[i].x);
        }
        cout << "Test case #" << T++ << '\n';
        cout << "Total explored area: " << fixed << setprecision(2) << ans << "\n\n";
    }
    return 0;
}