LeetCode 850. 矩形面积 II (扫描线)

题目描述

我们给出了一个（轴对齐的）矩形列表 rectangles 。 对于 rectangle[i] = [x1, y1, x2, y2]，其中（x1，y1）是矩形 i 左下角的坐标，（x2，y2）是该矩形右上角的坐标。

找出平面中所有矩形叠加覆盖后的总面积。 由于答案可能太大，请返回它对 10 ^ 9 + 7 取模的结果。

输入格式

首先输入arr数组的长度n，然后输入n个整数，表示arr中的元素
首先输入矩形列表 rectangles的长度n
然后输入n行，每行表示一个矩形的坐标[x1, y1, x2, y2]
1 $<=$ n $<=$ 50
0 $<=$ x1, y1, x2, y2 $<=10^9$
矩形叠加覆盖后的总面积不会超越 2^63 - 1 ，这意味着可以用一个 64 位有符号整数来保存面积结果。

输出格式

输出结果。

输入样例

3
0 0 2 2
1 0 2 3
1 0 3 1

输出样例

6

**解释：
**
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-sQVaMjvV-1662538428025)(https://s3-lc-upload.s3.amazonaws.com/uploads/2018/06/06/rectangle_area_ii_pic.png)]

---

分析

由于计算不同的正方形之间交集的方法非常复杂，所以我们考虑使用扫描线，把整体问题分块来看。
我们首先将每个正方形的2个横坐标x1,x2加入到一个数组中，相当于按横坐标进行分块：

之后将所有横坐标排序，每次拿出相邻的两个横坐标$[l,r]$，计算这个区间中纵轴上的总长度tot，面积就等于tot*(r-l)

C++ 代码

#include<bits/stdc++.h>
using namespace std;
const int N = 1010;
using namespace std;
int n;
class Solution {
public:
    vector<vector<int>> v;
    typedef long long LL;
    const int MOD = 1e9 + 7;
    struct node{
        int l,r;
        
        bool operator<(const node&p) const{
            if(l!=p.r) return l<p.l;
            return r<p.r;
        };
    };
    
    LL calc(int l, int r) {
        vector<node> s;
        
        //将所有横跨[l,r]的正方形的两个纵坐标y1,y2加入到数组中
        for(auto item:v)
        {
            if(item[0]<=l && r<=item[2])
                s.push_back({item[1],item[3]});
        }

        sort(s.begin(),s.end());
        
        //纵轴上总长度，当前线段左端点，右端点
        int tot=0,st=-1,ed=-1;
        
        //计算纵轴上的总长度，每次求出重叠部分的长度后加到tot中
        for (int i=0;i<s.size();i++) {
            node seg = s[i];
            if (seg.l>ed) tot+=ed-st,st=seg.l;
            ed=max(ed,seg.r);
        }
        //最后也要加上一个线段长度
        tot+=ed-st;
        
        //求出面积
        return (LL)tot*(r-l);
    }


    int rectangleArea(vector<vector<int>>& rectangles) {
        v = rectangles;

        vector<int> a;
        //分割图形，将每个正方形的两个横坐标加入到数组中
        for(auto item:v)
        {
            a.push_back(item[0]);
            a.push_back(item[2]);
        }
        sort(a.begin(),a.end());
        
        LL ans=0;
        for(int i=1;i<a.size();i++)
            if(a[i-1]!=a[i]) //如果区间长度大于0，就计算这个区间中纵轴上的总长度
                ans=(ans+calc(a[i-1],a[i]))%MOD;
        
        return ans%MOD;
    }
};


int main()
{

    int n,a,m,x_1,y_1,x_2,y_2;
    vector<vector<int>> arr;
    cin>>n;
    for(int i=0; i<n; i++)
    {
        vector<int> temp;
        cin>>x_1>>y_1>>x_2>>y_2;
        temp.push_back(x_1);
        temp.push_back(y_1);
        temp.push_back(x_2);
        temp.push_back(y_2);        
        arr.push_back(temp);
    }

    cout<<Solution().rectangleArea(arr);
    
    return 0;
}