题目: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] ,会冲突。一个个照着改,检查了好久的说。。
题目大意:给你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;
}