HDU1435 Stable Match 【稳定婚姻问题】

 

 

HDU1435
简单的稳定婚姻问题，寒假写过了，又拿出来看辽一遍嘻嘻。
算法的理解注释里写的非常清楚辽。
稳定婚姻问题又被称作不想单身的渣男的脱单之路。——鲁迅
参考：【稳定婚姻问题】【HDU1435】【Stable Match】

#include<iostream>
#include<cstring>
#include<string.h>
#include<cmath>
#include<algorithm>
#define maxn 205
#define eps 1e-8
using namespace std;
int a,b,n,T;
double c,d,e;
int mrk[maxn][maxn],frk[maxn][maxn],lady[maxn],man[maxn],num[maxn];
struct peo
{
	int v;
	double x,y,z;
}M[maxn],F[maxn];
struct node
{
	int l,id,v;
	double dis;
}tmp[maxn];
bool cmp(node A,node B)
{
	if (abs(A.dis-B.dis)<eps) return B.v<A.v;
		else return A.dis<B.dis;
}
bool cmpp(node A,node B)
{
	return A.id<B.id;
}
double getdis(peo A,peo B)
{
    return (A.x-B.x)*(A.x-B.x)+(A.y-B.y)*(A.y-B.y)+(A.z-B.z)*(A.z-B.z);
}
void init()
{
	for (int i=0;i<maxn;i++)
	{
		M[i].v=M[i].x=M[i].y=M[i].z=0; //男 
		F[i].v=F[i].x=F[i].y=F[i].z=0; //女 
		lady[i]=0; //女性的当前伴侣 
		man[i]=0; //男性的当前伴侣 
		num[i]=1;//男人要追求的女人是他心目中的第几位 1位开始 
	}
	memset(mrk,0,sizeof(mrk));//男人的心仪女生排序 
	memset(frk,0,sizeof(frk));//对frk[a][b]=t来说 男b是女a的第t位心仪对象 
}
int main()
{
	scanf("%d",&T);
	while (T--)
	{
		init();
		scanf("%d",&n);
		for (int i=1;i<=n;i++)
		{
			scanf("%d%d%lf%lf%lf",&a,&b,&c,&d,&e);
			M[a].v=b;
			M[a].x=c;M[a].y=d;M[a].z=e;
		} 
		for (int i=1;i<=n;i++)
		{
			scanf("%d%d%lf%lf%lf",&a,&b,&c,&d,&e);
			F[a].v=b;
			F[a].x=c;F[a].y=d;F[a].z=e;
		} 
		for (int i=1;i<=n;i++)
		{
			memset(tmp,0,sizeof(tmp));
			for (int j=1;j<=n;j++)
			{
				tmp[j].id=j;
				tmp[j].v=F[j].v;
				tmp[j].dis=getdis(M[i],F[j]);
			}
			sort(tmp+1,tmp+1+n,cmp);
			for (int j=1;j<=n;j++)
			{
				mrk[i][j]=tmp[j].id;//男人眼中的心仪女生排序 
			}
		} 
		for (int i=1;i<=n;i++)
		{
			memset(tmp,0,sizeof(tmp));
			for (int j=1;j<=n;j++)
			{
				tmp[j].id=j;
				tmp[j].v=M[j].v;
				tmp[j].dis=getdis(F[i],M[j]);
			}
			sort(tmp+1,tmp+1+n,cmp);//先按照心仪程度对男嘉宾排序 
			for (int j=1;j<=n;j++) tmp[j].l=j; //记录此时男嘉宾的排名 
			sort(tmp+1,tmp+1+n,cmpp);//按照原编号大小排序 方便下一步统计 
			for (int j=1;j<=n;j++)
			{
				frk[i][j]=tmp[j].l;//j男是i女心仪的第l位男嘉宾 
			}
		} 
		int ok=1;//初始状态表示可以进行婚姻配对 即有单身汉 
		while (ok)//当依然有单身男性时重复 
		{
			ok=0; 
			for (int i=1;i<=n;i++)
			{
				if (man[i]==0)//单身男子出现 
				{
					ok=1;
					int female=mrk[i][num[i]];//找到男i的最心仪女嘉宾 num[i]表示男i可追求的心仪女生在自己心目中的排名 递增 
					if (lady[female]==0)//该女嘉宾单身 可以下手了 
					{
						lady[female]=i;//锁了 
						man[i]=female;//锁了 
						num[i]++;//心仪排名+1 因为自己可能被女生踢掉 要继续追下一个女生 先告诉自己下一个要追谁【好渣】 
					}
					else if (frk[female][lady[female]]>frk[female][i])//不单身了 但男i比原配好 所以把原配踢掉 
					{
						man[lady[female]]=0;//原配被踢 恢复单身 
						man[i]=female;//锁了 
						lady[female]=i;//锁了 
						num[i]++;//渣男法则 
					}
					else num[i]++;//非单身 而且原配你也比不上 直接追下一个【渣男！】 
				}
			}
		}
		for (int i=1;i<=n;i++)
		{
			printf("%d %d\n",lady[i],i);
		}
	} 
	return 0; 
}