题目大意:给出n个蚁群与n个苹果树坐标(任意三点不共线),问能否使得每一个蚁群对应一个苹果树,且蚁群到苹果树的路线不相交。
分析:
容易想到完美匹配模型。
然后考虑路线相交的问题。对于四个定点,构成的两条不共点的线段。相交线段长度之和一定大于不相交的长度之和。
对于每条边的权值,另其等于两点的距离。那么问题转化为求最佳匹配(权值最小)。
由于KM算法一般是求权值最大的最佳匹配,因此套用KM算法时,可以让权值变为负值,这样问题转化为求权值最大的最佳匹配。
注意问题要输出的是蚁群所连接的苹果树编号。
那么匹配的时候,应该枚举每个苹果树,为其寻找匹配的蚁群。即L记录的是蚁群对应的苹果树编号。
#include<bits/stdc++.h> using namespace std; #define N 105 #define INF 1e9 #define eps 1e-10 typedef double db; struct P { db x,y; } a[N],b[N]; db W[N][N],Lx[N],Ly[N],slack[N]; int L[N],n; bool S[N],T[N]; bool match(int u) { S[u]=true; for(int j=1; j<=n; ++j) if(!T[j]) { if(Lx[u]+Ly[j]-W[u][j]<eps) { T[j]=true; if(!L[j]||match(L[j])) { L[j]=u; return true; } } else slack[j]=min(slack[j],Lx[u]+Ly[j]-W[u][j]); } return false; } void KM() { int i,j; for(i=1; i<=n; ++i) { L[i]=Ly[i]=0; Lx[i]=-INF; for(j=1; j<=n; ++j) Lx[i]=max(Lx[i],W[i][j]); } for(i=1; i<=n; ++i) { for(j=1; j<=n; ++j) slack[j]=INF; while(1) { for(j=1; j<=n; ++j) S[j]=T[j]=0; if(match(i)) break; db a=INF; for(j=1; j<=n; ++j) if(!T[j]) a=min(a,slack[j]); for(j=1; j<=n; ++j) { if(S[j]) Lx[j]-=a; if(T[j]) Ly[j]+=a; } } } } int main() { int i,j,cnt=0; while(~scanf("%d",&n)) { if(cnt++) puts(""); for(i=1; i<=n; ++i) scanf("%lf%lf",&a[i].x,&a[i].y); for(i=1; i<=n; ++i) scanf("%lf%lf",&b[i].x,&b[i].y); for(i=1; i<=n; ++i) for(j=1; j<=n;++j) W[i][j]=-sqrt((b[i].x-a[j].x)*(b[i].x-a[j].x)+(b[i].y-a[j].y)*(b[i].y-a[j].y)); KM(); for(i=1; i<=n; ++i) printf("%d\n",L[i]); } return 0; }