问题描述:
羽毛球队有男女运动员各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;
}