POJ 3545 Ants(最佳二分图匹配)

#include<iostream>
#include<cmath>
#include<cstdio>
using namespace std;
const int maxn=100+5;
const double INF=1e30;
int n;
double w[maxn][maxn];
double lx[maxn],ly[maxn];
int Left[maxn];
bool s[maxn],t[maxn];
struct point{int x,y;}point1[maxn],point2[maxn];
bool eq(double a,double b) {return fabs(a-b)<1e-9;}
bool match(int i)
{
    s[i]=true;
    for(int j=1;j<=n;j++){
        if(eq(lx[i]+ly[j],w[i][j])&&!t[j]){
            t[j]=true;
            if(!Left[j]||match(Left[j])){
                Left[j]=i;
                return true;
            }
        }
    }
    return false;
}
void update(){
    double a=INF;
    for(int i=1;i<=n;i++) if(s[i])
        for(int j=1;j<=n;j++) if(!t[j])
            a=min(a,lx[i]+ly[j]-w[i][j]);
    for(int i=1;i<=n;i++){
        if(s[i]) lx[i]-=a;
        if(t[i]) ly[i]+=a;
    }
}
void KM()
{
    for(int i=1;i<=n;i++){
        Left[i]=lx[i]=ly[i]=0;
        for(int j=1;j<=n;j++){
            lx[i]=max(lx[i],w[i][j]);
        }
    }
    for(int i=1;i<=n;i++){
        for(;;){
            for(int j=1;j<=n;j++) s[j]=t[j]=0;
            if(match(i)) break;else update();
        }
    }
}
double dis(point p1,point p2)
{
    double x=p1.x-p2.x;
    double y=p1.y-p2.y;
    return hypot(x,y);
}
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++) {scanf("%d%d",&point1[i].x,&point1[i].y);}
        for(int i=1;i<=n;i++) {scanf("%d%d",&point2[i].x,&point2[i].y);}
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++)
                w[j][i]=-dis(point1[i],point2[j]);//w[j][i]，则left中存的是白点对应的黑点，反一反就是黑对白
        }
        KM();
        for(int i=1;i<=n;i++) cout<<Left[i]<<endl;
    }
    return 0;
}