码迷,mamicode.com
首页 > 其他好文 > 详细

P1559 运动员最佳匹配问题

时间:2017-09-23 00:08:14      阅读:132      评论:0      收藏:0      [点我收藏+]

标签:max   个数   技术   男女   运动   cstring   etc   sizeof   输出   

 洛谷——P1559 运动员最佳匹配问题

题目描述

羽毛球队有男女运动员各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]。设计一个算法,计算男女运动员最佳配对法,使各组男女双方竞赛优势的总和达到最大。

输入输出格式

输入格式:

第一行有1 个正整数n (1≤n≤20)。接下来的2n行,每行n个数。前n行是p,后n行是q。

输出格式:

将计算出的男女双方竞赛优势的总和的最大值输出。

输入输出样例

输入样例#1:
3
10 2 3
2 3 4
3 4 5
2 2 2
3 5 3
4 5 1
输出样例#1:
52

思路


代码:
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#define N 40
#define maxn 9999999
using namespace std;
int n,a[N][N],b[N][N],love[N][N];
bool vis_boy[N],vis_girl[N];
int pre[N],remain[N],ex_girl[N],ex_boy[N];
int read()
{
    int x=0,f=1; char ch=getchar();
    while(ch<0||ch>9){if(ch==-) f=-1; ch=getchar();}
    while(ch>=0&&ch<=9){x=x*10+ch-0; ch=getchar();}
    return x*f;
}
int dfs(int girl)
{
    vis_girl[girl]=true;
    for(int i=1;i<=n;i++)
    {
        if(vis_boy[i]) continue;
        int ex=ex_boy[i]+ex_girl[girl]-love[girl][i];
        if(ex==0) 
        {
            vis_boy[i]=true;
            if(pre[i]==-1||dfs(pre[i]))
            {pre[i]=girl; return 1;}
        }
        else remain[i]=min(remain[i],ex);
    }
    return false;
}
int km()
{
    memset(ex_boy,0,sizeof(ex_boy));
    memset(pre,0,sizeof(pre));
    for(int i=1;i<=n;i++)
     for(int j=1;j<=n;j++)
      ex_girl[i]=max(ex_girl[i],love[i][j]);
    for(int i=1;i<=n;i++)
    {
        memset(remain,0x7fff,sizeof(remain));
        while(1)
        {
            memset(vis_boy,0,sizeof(vis_boy));
            memset(vis_girl,0,sizeof(vis_girl));
            if(dfs(i)) break;
            int d=maxn;
            for(int i=1;i<=n;i++)
             if(!vis_boy[i]) d=min(d,remain[i]);
            for(int i=1;i<=n;i++)
            {
                if(vis_boy[i]) ex_girl[i]-=d;
                if(vis_girl[i]) ex_boy[i]+=d;
                else remain[i]-=d;
            }
        }
    }
    int ans=0;
    for(int i=1;i<=n;i++)
     ans+=love[pre[i]][i];
    return ans;
}
int main()
{
    n=read();
    for(int i=1;i<=n;i++)
     for(int j=1;j<=n;j++)
      a[i][j]=read();
    for(int i=1;i<=n;i++)
     for(int j=1;j<=n;j++)
      b[i][j]=read();
    for(int i=1;i<=n;i++)
     for(int j=1;j<=n;j++)
      love[i][j]=a[i][j]*b[j][i];
   printf("%d\n",km());
   return 0;
 } 

 

#include<cstdio>
#include<iostream>
#define MAXN 110
#define INF 1000000000
using namespace std;
int lx[MAXN],ly[MAXN];
int f[MAXN][MAXN],n,m,pre[MAXN],pre2[MAXN];
bool vx[MAXN],vy[MAXN];
inline void read(int&x) {
    int f=1;x=0;char c=getchar();
    while(c>9||c<0) {if(c==-) f=-1;c=getchar();}
    while(c>=0&&c<=9) x=10*x+c-48,c=getchar();
    x=x*f;
}
inline void pra() {
    for(int i=1;i<=n;i++) 
      for(int j=1;j<=m;j++)
        lx[i]=max(lx[i],f[i][j]);
}
inline bool find(int u) {
    vx[u]=true;
    for(int i=1;i<=m;i++) {
        if(lx[u]+ly[i]==f[u][i]&&!vy[i]) {
            vy[i]=true;
            if(!pre[i]||find(pre[i])) {
                pre[i]=u;
                pre2[u]=i;
                return true;
            }
        }
    }
    return false;
}
inline void renew() {
    int t=INF;
    for(int i=1;i<=n;i++)
      if(vx[i])
        for(int j=1;j<=m;j++)
          if(!vy[j])
            t=min(t,lx[i]+ly[j]-f[i][j]);
    for(int i=1;i<=n;i++)
      if(vx[i]) lx[i]-=t;
    for(int j=1;j<=m;j++)
      if(vy[j]) ly[j]+=t;
}
inline void km() {
    for(int i=1;i<=n;i++) 
    while(true){
        for(int j=1;j<=n;j++) vx[j]=false;
        for(int j=1;j<=m;j++) vy[j]=false;
        if(find(i)) break;
        else renew();
    }
}
int main() {
    read(n);read(m);
    for(int i=1;i<=n;i++)
      for(int j=1;j<=m;j++)
        read(f[i][j]);
    pra();
    km();
    int ans=0;
    for(int i=1;i<=n;i++)
      ans+=f[i][pre2[i]];
    printf("%d\n",ans);
    return 0;
}

 

 

P1559 运动员最佳匹配问题

标签:max   个数   技术   男女   运动   cstring   etc   sizeof   输出   

原文地址:http://www.cnblogs.com/z360/p/6964880.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!