POJ365Ants_二分图最佳匹配

https://blog.csdn.net/lianai911/article/details/44835659

题意：

在坐标系中有N只蚂蚁，N棵苹果树，给你蚂蚁和苹果树的坐标。让每只蚂蚁去一棵苹果树，

一棵苹果树对应一只蚂蚁。这样就有N条直线路线，问：怎样分配，才能使总路程和最小，且

N条线不相交

思路：

假设A、B为蚂蚁，C、D为苹果树。则存在两种匹配：第一种是AD、BC，第二种是AC、BD。

根据三角形不等式AD+BC < AC+BD，最后得到很重要的一个性质——满足总路程之和最小

的方案一定不相交。现在来构建二分图，一边为蚂蚁，另一边为苹果树，以距离为边权值，题

目就变为了求带权二分图最小权和的最佳匹配。反向来思考，将距离乘以-1取负值建图，那么

就变为了求带权二分图最大权和的最佳匹配
Tips：用的距离的平方做比较是错的。，我还傻傻调试了半天，aiyawo…

也就是说 a^2 + b ^ 2 > c ^ c + d ^ 2 不能推出 a + b > c + d
比如1^2 + 100^2 > 50 ^ 2 + 51 ^2 但是 1 + 100 < 51 + 52

#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#define fi first
#define se second
#define pii pair<int,int>
using namespace std;
typedef long long LL;
const LL INF = 1000000000000;
const int maxn = 100+5;
int n;
struct Point{
	LL x,y;
}p1[maxn], p2[maxn];

LL G[maxn][maxn];
LL lx[maxn], ly[maxn];
int cx[maxn], cy[maxn];
int visx[maxn], visy[maxn];
LL slack[maxn];

bool GetAugumentPath(int u){
	visx[u] = 1;
	for(int v = 0; v < n; ++v){
		if(visy[v]) continue;
		LL d = lx[u] + ly[v] - G[u][v];
		if(d == 0){
			visy[v] = 1;
			if(cy[v] == -1||GetAugumentPath(cy[v])){
				cx[u] = v;
				cy[v] = u;
				return true;
			}
		}
		else
			slack[v] = min(slack[v], d);
	}
	return false;
}
void km(){
	for(int x = 0; x < n; ++x){
		for(int i = 0; i < n; ++i) slack[i] = INF;
		while(1){
			memset(visx, 0, sizeof(visx));
			memset(visy, 0, sizeof(visy));
			if(GetAugumentPath(x)) break;
			LL d = INF;
			for(int i = 0; i < n; ++i) if(!visy[i]) d = min(d, slack[i]);
			for(int i = 0; i < n; ++i) if(visx[i]) lx[i] -= d;
			for(int i = 0; i < n; ++i){
				if(visy[i]) ly[i] += d;
				else slack[i] -= d;
			}
		}
	}
}
void solve(){
	// 初始化
	memset(cx, -1, sizeof(cx));
	memset(cy, -1, sizeof(cy));
	for(int i = 0; i < n; ++i) lx[i] = -INF;
	for(int i = 0; i < n; ++i){
		for(int j = 0; j < n; ++j){
			lx[i] = max(lx[i], G[i][j]);
			ly[j] = 0;
		}
	}
	km();
	for(int i = 0; i < n; ++i) printf("%d\n",cx[i]+1);
}

int main()
{
	//freopen("in.txt","r",stdin);
	while(scanf("%d",&n) == 1&&n){
		for(int i = 0; i < n; ++i) scanf("%lld%lld", &p1[i].x, &p1[i].y);
		for(int i = 0; i < n; ++i) scanf("%lld%lld", &p2[i].x, &p2[i].y);
		for(int i = 0; i < n; ++i){
			for(int j = 0; j < n; ++j){
				G[i][j] = -((p1[i].x-p2[j].x)*(p1[i].x-p2[j].x) + (p1[i].y-p2[j].y)*(p1[i].y-p2[j].y));
			}
		}
		solve();
	}
	return 0;
}