121. 赶牛入圈

121. 赶牛入圈

考察

二分+前缀和+离散化(二维数组)

思路

首先对每个坐标进行离散化：将所有横纵坐标加入alg数组里，排序去重alg，查找使用二分
同时需要记录原先的坐标关系到zb，通过alg离散化坐标和zb的坐标关系映射到二维数组g上，并计算g的前缀和
然后就是二分边长，这里的难点是check写法。
check的过程是，遍历x2和y2坐标，同时计算合适的x1和y1坐标使得边长尽可能接近len。
然后计算前缀和是否大于等于c即可。

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
typedef pair<int,int> PII;
const int N = 1010; //纵和横坐标的范围分别500，一共可能为1000
vector<int> alg;
PII zb[N];
int g[N][N],c,n;

int find(int x){
    int l = 0, r = alg.size()-1;
    while(l < r){
        int mid = l+r>>1;
        if(alg[mid] >= x) r = mid;
        else l = mid+1;
    }
    return l;
}

bool check(int len){
    for(int x1 = 0, x2 = 1; x2 < alg.size(); x2++){
        while(alg[x2] - alg[x1+1] + 1 > len) x1++; //计算适合的x1坐标
        for(int y1 = 0, y2 = 1; y2 < alg.size(); y2++){
            while(alg[y2] - alg[y1+1] + 1 > len) y1++; 计算适合的y1坐标
            if(g[x2][y2]-g[x2][y1]-g[x1][y2]+g[x1][y1] >= c)
                return true;
        }
    }
    return false;
}

int main(){
    cin>>c>>n;
    alg.push_back(0);
    for(int i = 0; i < n; i++){
        int x,y; cin>>x>>y;
        zb[i] = {x,y};
        alg.push_back(x);
        alg.push_back(y);
    }
    sort(alg.begin(),alg.end());
    alg.erase(unique(alg.begin(),alg.end()),alg.end());
    for(int i = 0; i < n; i++){
        int x = find(zb[i].first);
        int y = find(zb[i].second);
        g[x][y]++;
    }
    for(int i = 1; i < alg.size(); i++)
        for(int j = 1; j < alg.size(); j++)
            g[i][j] += g[i-1][j] + g[i][j-1] - g[i-1][j-1];
    
    int l = 1, r = 10000;
    while(l < r){
        int mid = l+r>>1;
        if(check(mid)) r = mid;
        else l = mid+1;
    }
    cout<<l;
    return 0;    
}