洛谷 P3138 [USACO16FEB]Load Balancing S（二维前缀和，离散化）

Posted 2021-08-07 尹昱钦

传送门

---

解题思路

先把x和y分别离散化，然后求一下前缀和，枚举栅栏，求一下即可。

AC代码

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=1e3+5;
int n,bx[maxn],by[maxn],cntx,cnty,ma[maxn][maxn],d[maxn][maxn],ans=1e9;
struct node{
	int x,y;
}a[maxn];
inline int work(int i,int j){
	return max(max(d[i][j],d[cntx][j]-d[i][j]),max(d[i][cnty]-d[i][j],d[cntx][cnty]-d[cntx][j]-d[i][cnty]+d[i][j]));
}
int main()
{
    ios::sync_with_stdio(false); 
    cin>>n;
    for(int i=1;i<=n;i++){
    	cin>>a[i].x>>a[i].y;
    	bx[++cntx]=a[i].x;
    	by[++cnty]=a[i].y;
	}
	sort(bx+1,bx+cntx+1);
	cntx=unique(bx+1,bx+cntx+1)-bx-1;
	sort(by+1,by+cnty+1);
	cnty=unique(by+1,by+cnty+1)-by-1;
	for(int i=1;i<=n;i++){
		a[i].x=lower_bound(bx+1,bx+cntx+1,a[i].x)-bx;
		a[i].y=lower_bound(by+1,by+cnty+1,a[i].y)-by;
		ma[a[i].x][a[i].y]=1;
	}
	for(int i=1;i<=cntx;i++){
		for(int j=1;j<=cnty;j++){
			d[i][j]=d[i-1][j]+d[i][j-1]-d[i-1][j-1]+ma[i][j];
		}
	}
	for(int i=1;i<=cntx;i++){
		for(int j=1;j<=cnty;j++){
			ans=min(ans,work(i,j));
		}
	}
	cout<<ans;
    return 0;
}