蓝书里的一道经典问题,稳定婚姻模型。
具体内容参考蓝书,这里只提供代码。
#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 ",man[i][manpt[i]-1]); if(T) puts(""); } return 0; }