UVALive 4043 Ants（二分图最佳完美匹配、KM）

转自：

http://m.blog.csdn.net/blog/u010794465/21482821

题目：https://icpcarchive.ecs.baylor.edu/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=2044
题目大意：给你n个白点和n个黑点的平面坐标，要求用n条不相交的线连起来，每条线段连一个白点和黑点，每个点连一条线，也就是匹配。让你输出第i个白点所对应的黑点。
解题思路：二分图完美匹配问题。但是题目中有个线段不相交，怎么办？其实这个最佳完美匹配就是答案了。最佳完美匹配是权值和最大，那么我们就把两两点线段的权值搞成他们距离的负数即可。这样就不可能有相交的了。为什么？因为假设有相交，a1-b2，a2-b1，而dist(a1,b1)+dist(a2,b2) 肯定比前面交叉的小，画个四边形就很清楚了，那么负数就是大了，也就是说交叉的在我们设计的负权那里是小的，所以就是最佳，也就是不可能有交叉的。
这样分析清楚了之后，就只要直接套用KM就OK了！
        KM这里还有一个地方需要注意，我又WA了n次，就是点的下标不能从 0 开始，因为我们 0 表示的不存在，比如 !left[j] ,会冲突。一个个照着改，检查了好久的说。。

代码如下（O（n^4））：

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;

const double EPS = 1e-10;
const int INF = 0x0fffffff;
const int MAXN = 111;

struct Node
{
    int x,y;
    void read()
    {
        scanf("%d%d",&x,&y);
    }
} x[MAXN],y[MAXN];//由于0表示不存在，集合下标都是 1~n

double cal_dis(int i,int j)
{
    return sqrt((x[i].x-y[j].x)*(x[i].x-y[j].x)+(x[i].y-y[j].y)*(x[i].y-y[j].y));
}

int n;
double w[MAXN][MAXN];
double lx[MAXN],ly[MAXN];
int s[MAXN],t[MAXN],left[MAXN];

int match(int i)
{
    s[i] = 1;
    for(int j = 1;j <= n;j++)
        if(!t[j] && fabs(lx[i]+ly[j]-w[i][j]) < EPS)
        {
            t[j] = 1;
            if(!left[j] || match(left[j]))
            {
                left[j] = i;
                return 1;
            }
        }
    return 0;
}

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] = ly[i] = 0;
        lx[i] = -INF;
        for(int j = 1;j <= n;j++)
            lx[i] = max(lx[i],w[i][j]);
    }
    for(int i = 1;i <= n;i++)
        while(1)
        {
            for(int j = 1;j <= n;j++) s[j] = t[j] = 0;
            if(match(i)) break;
            else update();
        }
}

int main()
{
    int cas = 0;
    while(~scanf("%d",&n))
    {
        if(cas++ > 0) puts("");
        for(int i = 1;i <= n;i++)//必须从1开始！
            y[i].read();
        for(int i = 1;i <= n;i++)
            x[i].read();
        for(int i = 1;i <= n;i++)
            for(int j = 1;j <= n;j++)
                w[i][j] = -cal_dis(i,j);
        km();
        for(int i = 1;i <= n;i++)
            printf("%d\n",left[i]);
    }
    return 0;
}

用 slack 进行 update 部分的优化。

代码如下（O（n^3））：

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;

const double EPS = 1e-10;
const int INF = 0x0fffffff;
const int MAXN = 111;

struct Node
{
    int x,y;
    void read()
    {
        scanf("%d%d",&x,&y);
    }
} x[MAXN],y[MAXN];

double cas_dis(int i,int j)
{
    return sqrt((x[i].x-y[j].x)*(x[i].x-y[j].x)+(x[i].y-y[j].y)*(x[i].y-y[j].y));
}

int n;
double w[MAXN][MAXN];
double lx[MAXN],ly[MAXN],slack[MAXN];
int s[MAXN],t[MAXN],left[MAXN];

int match(int i)
{
    s[i] = 1;
    for(int j = 1;j <= n;j++)
        if(!t[j])
        {

            double tmp = lx[i]+ly[j]-w[i][j];
            if(fabs(tmp) < EPS)
            {
                t[j] = 1;
                if(!left[j] || match(left[j]))
                {
                    left[j] = i;
                    return 1;
                }
            }
            else slack[j] = min(tmp,slack[j]);
        }
    return 0;
}

void update()
{
    double a = INF;
    for(int i = 1;i <= n;i++)
        if(!t[i]) a = min(a,slack[i]);
    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] = ly[i] = 0;
        lx[i] = -INF;
        for(int j = 1;j <= n;j++)
            lx[i] = max(lx[i],w[i][j]);
    }
    for(int i = 1;i <= n;i++)
    {
        for(int j = 1;j <= n;j++)
            slack[j] = INF;
        while(1)
        {
            for(int j = 1;j <= n;j++) s[j] = t[j] = 0;
            if(match(i))
                break;
            else update();
        }
    }
}

int main()
{
    int cas = 0;
    while(~scanf("%d",&n))
    {
        if(cas++ > 0) puts("");
        for(int i = 1;i <= n;i++)
            y[i].read();
        for(int i = 1;i <= n;i++)
            x[i].read();
        for(int i = 1;i <= n;i++)
            for(int j = 1;j <= n;j++)
                w[i][j] = -cas_dis(i,j);
        km();
        for(int i = 1;i <= n;i++)
            printf("%d\n",left[i]);
    }
    return 0;
}