蓝书里的一道经典问题,稳定婚姻模型。
具体内容参考蓝书,这里只提供代码。
#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;
}