P3073 Tractor

P3073 Tractor

题目描述

FJ有块农田太崎岖了，他要买一辆新拖拉机才能在这里巡视。这块农田由N x N个格子的非负整数表示高度(1<=N<=500)。拖拉机从当前格子走到相邻格子（东、南、西、北四个方向）的代价为高度差D，则FJ驶过这两个格子的拖拉机最少也要值D块钱。

FJ愿意花足够的钱买一辆新的拖拉机使得他能以最小的高度差走遍所有格子的一半(如果格子总数是奇数，那么一半的值为四舍五入的值)。因为FJ很懒，所以他找到你帮他编程计算他最小需要花多少钱买到符合这些要求的拖拉机。

思路

我们可以二分这个答案，因为它符合单调性，然后接下来我们就需要去求出在这个答案下能否驶过一半的格子，我们可以用一个并查集维护，假如当前这个格子能够到达下一个格子，那么就可以合并起来，同时维护根节点的大小

贴代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#define maxn 600
using namespace std;
int dx[2]={1,0};
int dy[2]={0,1};
int m[maxn][maxn];
int n,ans;
int l,r,mid;
int d[maxn*maxn];
int f[maxn*maxn];
struct Edge{
	int x,y,v;
	bool operator < (const Edge &a)const{
		return v<a.v;
	}
}e[maxn*maxn*4];
int ecnt;
int abs(int x){
	return x<0?-x:x;
}
void adde(int x,int y,int v){
	e[++ecnt].x=x;
	e[ecnt].y=y;
	e[ecnt].v=v;
}
bool check(int x,int y){
	if(x<1||x>n||y<1||y>n){
		return false;
	}
	else{
		return true;
	}
}
int num(int x,int y){
	return n*(x-1)+y;
}
int find(int x){
	return f[x]==x?x:f[x]=find(f[x]);
}
void merge(int a,int b){
	int fa=find(a);
	int fb=find(b);
	if(fa==fb){
		return;
	}
	if(d[fa]>d[fb]){
		swap(fa,fb);
	}
	f[fa]=fb;
	d[fb]+=d[fa];
	return;
}
bool solve(int limit){
	for(int i=1;i<=n*n;i++){
		f[i]=i;
		d[i]=1;
	}
	for(int i=1;i<=ecnt;i++){
		if(e[i].v>limit){
			break;
		}
		merge(e[i].x,e[i].y);
	}
	for(int i=1;i<=n*n;i++){
		if(d[find(i)]>=n*n/2){
			return true;
		}
	}
	return false;
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			cin>>m[i][j];
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			for(int k=0;k<2;k++){
				int nx=i+dx[k];
				int ny=j+dy[k];
				if(check(nx,ny)){
					int de=abs(m[i][j]-m[nx][ny]);
					r=max(r,de);
					adde(num(i,j),num(nx,ny),de);
				}
			}
		}
	}
	sort(e+1,e+1+ecnt);
	while(l<=r){
		mid=(l+r)/2;
		if(solve(mid)){
			ans=mid;
			r=mid-1;
		}
		else{
			l=mid+1;
		}
	}
	printf("%d",ans);
}