UVA 1175 Ladies' Choice

 

 

蓝书里的一道经典问题，稳定婚姻模型。

具体内容参考蓝书，这里只提供代码。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdlib>
#include<queue>
#define ll long long
#define maxn 1005
using namespace std;
queue<int> q;
int man[maxn][maxn];
int woman[maxn][maxn];
int n,T,manpt[maxn],hus[maxn];

inline void init(){
	fill(manpt+1,manpt+n+1,1);
	memset(hus,0,sizeof(hus));
}

inline bool marry(int x,int y){
	if(!hus[y]){
		hus[y]=x;
		return 1;
	}
	else if(woman[y][x]<woman[y][hus[y]]){
		q.push(hus[y]);
		hus[y]=x;
		return 1;
	}
	else return 0;
}

inline void solve(){
	for(int i=1;i<=n;i++) q.push(i);
	
	int x,girl;
	while(!q.empty()){
		x=q.front(),q.pop();
		girl=man[x][manpt[x]++];
		if(!marry(x,girl)) q.push(x);
	}
}

int main(){
	scanf("%d",&T);
	while(T--){
		scanf("%d",&n);
		init();
		for(int i=1;i<=n;i++)
		    for(int j=1;j<=n;j++) scanf("%d",&man[i][j]);
		int now;
		for(int i=1;i<=n;i++)
		    for(int j=1;j<=n;j++){
		    	scanf("%d",&now);
		    	woman[i][now]=j;
			}
		
		solve();
		
		for(int i=1;i<=n;i++) printf("%d\n",man[i][manpt[i]-1]);
		if(T) puts("");
	}
	
	return 0;
}