NOI / 1.11编程基础之二分查找 03:矩形分割 二分

描述

平面上有一个大矩形，其左下角坐标（0，0），右上角坐标（R,R)。大矩形内部包含一些小矩形，小矩形都平行于坐标轴且互不重叠。所有矩形的顶点都是整点。要求画一根平行于y轴的直线x=k（k是整数) ，使得这些小矩形落在直线左边的面积必须大于等于落在右边的面积，且两边面积之差最小。并且，要使得大矩形在直线左边的的面积尽可能大。注意：若直线穿过一个小矩形，将会把它切成两个部分，分属左右两侧。

输入

第一行是整数R，表示大矩形的右上角坐标是(R,R) (1 <= R <= 1,000,000)。
接下来的一行是整数N,表示一共有N个小矩形(0 < N <= 10000)。
再接下来有N 行。每行有4个整数，L,T, W 和 H, 表示有一个小矩形的左上角坐标是(L,T),宽度是W，高度是H (0<=L,T <= R, 0 < W,H <= R). 小矩形不会有位于大矩形之外的部分。

输出

输出整数n，表示答案应该是直线 x=n。 如果必要的话，x=R也可以是答案。

样例输入

1000
2
1 1 2 1
5 1 2 1

样例输出

5

#include <bits/stdc++.h>
using namespace std;
#define N 10005
struct Rect//矩形
{
    int l, t, w, h;//(l,t):矩形左上角坐标 w:宽 h:高
};
Rect rec[N];
int rx, n;//rx:题目中的R
long long totArea;//总面积
long long getLeftArea(int k)//x=k左侧所有小矩形面积加和
{
    long long leftArea = 0, lw;//lw:竖线x=k左侧矩形的宽
    for(int i = 1; i <= n; ++i)
    {
        if(rec[i].l+rec[i].w <= k)//矩形在x=k左边
            lw = rec[i].w;
        else if(rec[i].l <= k)//x=k穿过矩形
            lw = k-rec[i].l;
        else//矩形在x=k右侧
            lw = 0;
        leftArea += lw*rec[i].h;
    }
    return leftArea;
}
bool check(int k)//选择x=k这条竖线是否满足左侧面积大于等于右侧面积
{
    long long leftArea = getLeftArea(k);
    long long rightArea = totArea - leftArea;
    return leftArea >= rightArea;
}
int main()
{
    cin >> rx >> n;
    for(int i = 1; i <= n; i++)
    {
        cin >> rec[i].l >> rec[i].t >> rec[i].w >> rec[i].h;
        totArea += (long long)rec[i].w*rec[i].h;//求这个小矩形的面积，加和（注意，w，h都是int，而乘积会超过int的范围，要转为long long）
    }
    int l = 0, r = rx, mid, k;
    while(l < r)//二分答案求满足条件的最小值
    {
        mid = (l+r)/2;
        if(check(mid))//条件为：选择x=k这条竖线左侧面积大于等于右侧面积
            r = mid;
        else
            l = mid + 1;
    }
    k = l;//取到：满足x=k这条竖线左侧面积大于等于右侧面积的k的最小值
    int leftArea = getLeftArea(k);//此时x=k左侧小矩形面积
    while(getLeftArea(k+1) == leftArea && k+1 <= rx)//保持左侧面积不变（也就是保持左右侧面积差值不变），k从小向大遍历，取到可能取到的最大值
        k++;
    cout << k;
    return 0;
}