UVa:1411 Ants(KM算法)

由黑白两色的结点很容易想到二分图匹配，但是线段相交这里很难处理。这里有一个性质，即不相交的线段和一定小于相交的线段和。由于一定存在解，那么这样线段和最小的情况一定是不相交的。这样由KM算法求最小权即可。

一开始以为用距离不开方用longlong可以存下，结果一直WA，改成double才过掉。注意两个样例之间有空行。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <vector>
#include <queue>
#include <map>
#include <algorithm>
#define ll long long
#define INF 1e30
#define inf -2139062144
#define MOD 20071027
#define MAXN 105
using namespace std;
bool visx[MAXN],visy[MAXN];
double lx[MAXN],ly[MAXN],slack[MAXN];
double w[MAXN][MAXN];
int link[MAXN];
int nx,ny,n;
bool find(int x)
{
    visx[x]=true;
    for(int y=1; y<=ny; ++y)
        if(!visy[y])
        {
            if(fabs(lx[x]+ly[y]-w[x][y])<1e-6)
            {
                visy[y]=true;
                if(link[y]==-1||find(link[y]))
                {
                    link[y]=x;
                    return true;
                }
            }
            else slack[y]=min(slack[y],lx[x]+ly[y]-w[x][y]);
        }
    return false;
}
void KM()
{
    memset(link,-1,sizeof(link));
    memset(lx,0x80,sizeof(lx));
    memset(ly,0,sizeof(ly));
    for(int i=1; i<=nx; ++i)
        for(int j=1; j<=ny; ++j)
            lx[i]=max(lx[i],w[i][j]);
    for(int x=1; x<=nx; ++x)
    {
        memset(slack,0x7f,sizeof(slack));
        while(1)
        {
            memset(visx,0,sizeof(visx));
            memset(visy,0,sizeof(visy));
            if(find(x)) break;
            double d=INF;
            for(int y=1; y<=ny; ++y)
                if(!visy[y]&&d>slack[y])
                    d=slack[y];
            for(int i=1; i<=nx; ++i)
                if(visx[i]) lx[i]-=d;
            for(int i=1; i<=ny; ++i)
                if(visy[i]) ly[i]+=d;
                else slack[i]-=d;
        }
    }
}
double dist(int x1,int y1,int x2,int y2)
{
    return sqrt((double)(x1-x2)*(x1-x2)+(double)(y1-y2)*(y1-y2));
}
int xx1[MAXN],yy1[MAXN];
int xx2[MAXN],yy2[MAXN];
int main()
{
    int kase=0;
    while(scanf("%d",&n)!=EOF)
    {
        nx=n;
        ny=n;
        if(kase) printf("\n");
        for(int i=1; i<=n; ++i)
            scanf("%d%d",&xx1[i],&yy1[i]);
        for(int i=1; i<=n; ++i)
            scanf("%d%d",&xx2[i],&yy2[i]);
        for(int i=1; i<=n; ++i)
            for(int j=1; j<=n; ++j)
                w[j][i]=-dist(xx1[i],yy1[i],xx2[j],yy2[j]);
        KM();
        for(int i=1; i<=n; ++i)
            printf("%d\n",link[i]);
        kase=1;
    }
    return 0;
}