问题描述 :
我们给出了一个(轴对齐的)矩形列表 rectangles 。 对于 rectangle[i] = [x1, y1, x2, y2],其中(x1,y1)是矩形 i 左下角的坐标,(x2,y2)是该矩形右上角的坐标。
找出平面中所有矩形叠加覆盖后的总面积。 由于答案可能太大,请返回它对 10 ^ 9 + 7 取模的结果。
示例 1:
输入:[[0,0,2,2],[1,0,2,3],[1,0,3,1]]
输出:6
解释:如图所示。
示例 2:
输入:[[0,0,1000000000,1000000000]]
输出:49
解释:答案是 10^18 对 (10^9 + 7) 取模的结果, 即 49 。
输入说明 :
首先输入矩形列表 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
#include<iostream>
#include<vector>
#include<set>
#include<unordered_map>
#include<algorithm>
using namespace std;
const uint64_t MOD=1e9+7;
int X[1005]={0};
struct Node{
int cnt,len;
Node():cnt(0),len(0){};
};
namespace SegTree{
Node a[1005];
void build(){
for(int i=0;i<1005;i++)
a[i].cnt=0,a[i].len=0;
}
int update(int R,int l,int r,int l1,int r1,int val){ //更新与查询全部区间值写在了一起
int le=0,ri=0;
if(l1>=r1)
return 0;
if(l==l1&&r==r1){ //如果当前区间更新后并不需要更新子区间,因为每次查询都是全局的
a[R].cnt+=val;
}else{
int mid=l+r>>1;
update(R<<1,l,mid,min(l1,mid),min(r1,mid),val);
update(R<<1|1,mid,r,max(l1,mid),max(r1,mid),val);
}
if(a[R].cnt){ //如果存在直接返回该区间实际长度
a[R].len=X[r]-X[l];
}else
a[R].len=a[R<<1].len+a[R<<1|1].len;
return a[R].len; //不存在则返回左半区间和右半区间的长度
}
}
class Solution {
public:
int n;
int rectangleArea(vector<vector<int>>& rectangles) {
vector<vector<int>> events;
set<int> xs;
for(auto &rec:rectangles){
events.push_back({rec[1], 1, rec[0], rec[2]});
events.push_back({rec[3], -1, rec[0], rec[2]});
xs.insert(rec[0]);
xs.insert(rec[2]);
}
int cnt=0;
uint64_t ans=0;
for(auto x:xs)
X[cnt++]=x;
unordered_map<int,int> note;
for(int i=0;i<cnt;i++)
note[X[i]]=i;
sort(events.begin(),events.end(),[](vector<int> &a,vector<int> &b){return a[0]<b[0];});
SegTree::build();
int lasty=events[0][0],len=0;
for(auto &event:events){
ans+=(uint64_t)len*(uint64_t)(event[0]-lasty);
ans%=MOD;
len=SegTree::update(1,0,cnt-1,note[event[2]],note[event[3]],event[1]);
lasty=event[0];
}
return ans%MOD;
}
};
int main(){
int n;
vector<vector<int>> rectangles;
vector<int> tmp;
cin>>n;
int x;
for(int i=0;i<n;i++){
for(int j=0;j<4;j++){
cin>>x;
tmp.push_back(x);
}
rectangles.push_back(tmp);
tmp.clear();
}
int res=Solution().rectangleArea(rectangles);
cout<<res;
}