LA4043 KM算法

直接套用KM即可解决 要注意距离采用的是double 要在match函数中修改判断条件

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
//#include<map>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define pi acos(-1.0)
#define ls (rt<<1)
#define rs ((rt<<1)|1)
#define mid ((l+r)>>1)
using namespace std;
const int maxn = 110;
const double eps = 1e-10;
double w[maxn][maxn];
int n;
double lx[maxn],ly[maxn];   //顶标
int lf[maxn];
bool s[maxn],t[maxn];     //左右第i点是否标记
bool match(int i)
{
    s[i]=true;
    for(int j=1;j<=n;++j)
    if((fabs(lx[i]+ly[j]-w[i][j])<eps)&&!t[j])
    {
        t[j]=true;
        if(!lf[j]||match(lf[j]))
        {
            lf[j]=i;
            return true;
        }
    }
    return false;
    
}
void update()
{
    double  a=1<<30;
    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++) lf[i]=lx[i]=ly[i]=0;
    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();
        }
    }
}
double dist(double x1,double y1,double x2,double y2)
{
    return sqrt((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2));
}
int main()
{
    double  ax[maxn],ay[maxn],bx[maxn],by[maxn];
    while(~scanf("%d",&n))
    {
  //      memset(w,0,sizeof w );
        for(int i=1;i<=n;i++)
        {scanf("%lf%lf",&ax[i],&ay[i]);}   //ant
        for(int i=1;i<=n;++i)
        {scanf("%lf%lf",&bx[i],&by[i]);}    //tree
    
        for(int i=1;i<=n;i++)
          {
            for(int j=1;j<=n;j++)
            {
                w[i][j]=-dist(bx[i],by[i],ax[j],ay[j]);
             }
          }
        km();
        for(int i=1;i<=n;++i)
        {printf("%d\n",lf[i]);}
    }
}