稳定婚姻小模版(来自HDU)

稳定婚姻是二分图匹配的一种

原模版几个变量有问题 进行过修正 贴代码如下：

 

 

#include <iostream>
#include <queue>
using namespace std;

#define MAX 1010

int n;
int mlove[MAX][MAX];
int wlove[MAX][MAX]; 
int couple[MAX];
int pare[MAX];

queue < int > SQ;
void gale_shapley()
{
    int i,man,woman;
    
    while(!SQ.empty()) 
        SQ.pop();
    memset(couple, -1, sizeof(couple));
    
    for(i=1;i<=n;i++) 
        SQ.push(i);
    
    
    while(!SQ.empty())
    {
        man = SQ.front();    
        SQ.pop();
        for(i=1;i<=n;i++)
        {
            if(mlove[man][i] != -1)//如男生未选择过该女生
            {
                //选择未被拒绝且最喜欢的女生
                woman = mlove[man][i];//挑选最喜欢的女生
                mlove[man][i] = -1;//选择并标记之
                long pre = couple[woman];
                if(pre == -1) //若没有女生未配对
                {
                    couple[woman] = man;//撮合之
                    pare[man] = woman;//撮合之
                    break;
                }
                else //若已配对
                {
                    if(wlove[woman][man] > wlove[woman][pre])//如果女生喜欢这个男生超过现在的
                    {
                        SQ.push(pre);//把现在的放到队列里
                        couple[woman] = man;//撮合之
                        pare[man] = woman;//撮合之                
                        break;
                    }
                }
            }
        }
    }/while
}

int main()
{
    int i, j, t, hj;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d", &n);
        for(i = 1; i <= n; i++)
        {
            for(j = 1; j <= n; j++)
            {
                scanf("%d", &mlove[i][j]);//记录喜欢的人的号码
            }
        }
        for(i = 1; i <= n; i++)
        {
            for(j = 1; j <= n; j++)
            {
                scanf("%d", &hj);
                wlove[i][hj] = n - j;//记录喜欢程度从大到小
            }
        }
        
        gale_shapley();

        for(i = 1; i <= n; i++)
            printf("%d\n", pare[i]);//对于男方的配偶
    }
    return 0;
}