回溯法之运动员匹配问题

问题描述：

羽毛球队有男女运动员各n 人。给定2 个n×n 矩阵P 和Q。P[i][j]是男运动员i 和女运动员j配对组成混合双打的男运动员竞赛优势；Q[i][j]是女运动员i和男运动员j配合的女运动员竞赛优势。由于技术配合和心理状态等各种因素影响，P[i][j]不一定等于Q[j][i]。男运动员i和女运动员j配对组成混合双打的男女双方竞赛优势为P[i][j]*Q[j][i]。设计一个算法，对于给定的男女运动员竞赛优势，计算男女运动员最佳配对法，使各组男女双方竞赛优势的总和达到最大。

编程任务：

给定2 个n×n 矩阵P 和Q。P[i][j]是男运动员i 和女运动员j配对组成混合双打的男运动员竞赛优势；Q[i][j]是女运动员i和男运动员j配合的女运动员竞赛优势。设计一个算法，对于给定的男女运动员竞赛优势，计算男女运动员最佳配对法，使得p[i][j]*q[i][j]的和值最大化。并输出最大值。

输入样例：（第一行是男队员（或女队员）的个数，第二、三、四行是男运动员i 和女运动员j配对组成混合双打的男运动员竞赛优势，第五、六、七行是女运动员i和男运动员j配合的女运动员竞赛优势）

3

10 2 3

2 3 4

3 4 5

2 2 2

3 5 3

4 5 1

输出样例：（第一行是竞赛优势的最大和）

52

#include <iostream>
#include <algorithm>
int n;
int p[100][100],q[100][100];
int data[100][100];//计算p[i][j]*q[j][i]
int maxvalue=0;//所求的最大值
int maxsum=0;//临时最大值
int sum[100]={0};
int flag[100]={0};//标记女运动员是否已经配对过
using namespace std;
void dfs(int t)
{
    if(t>=n)
    {
        maxvalue=max(maxsum,maxvalue);
    }
    int left=0;
    for(int i=t;i<n;i++)
        left+=sum[i];
    if(maxsum+left<maxvalue)//maxsum表示已配对的，而left表示剩下不考虑女运动员时的最大值，如果和仍比maxvalue小，则进行剪枝
        return;
    for(int i=0;i<n;i++)
    {
        if(!flag[i])
        {
            flag[i]=1;
            maxsum+=data[t][i];
            dfs(t+1);
            maxsum-=data[t][i];
            flag[i]=0;
        }
    }
}
int main()
{
    cin>>n;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            cin>>p[i][j];
        }
    }
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            cin>>q[i][j];
        }
    }
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            data[i][j]=p[i][j]*q[j][i];
            sum[i]=max(sum[i],data[i][j]);//先不考虑女运动员，计算最大值，便于剪枝
        }
    }
    dfs(0);
    cout<<maxvalue<<endl;
    return 0;
}