LightOJ 1237 KM求最大匹配最小匹配

http://acm.hust.edu.cn/vjudge/contest/view.action?cid=9954#problem/G
题意:有一些人来咖啡馆开始时间和离开咖啡馆的时间,但是对应的顺序被打乱了,也就是说不知道每个人来的时间和离开的时间。有一个函数用来计算某人来咖啡馆所需要的费用,求最大能获得多少费用,最少能获得多少费用,还要判断数据是不是合法。

思路:由于每个人对应一个开始时间和结束时间,因此我们可以以此建立二分图,边的权值就是所对应的费用。当二分图匹配为人的个数时,说明数据是合法的,然后用KM算法求最大匹配最小匹配就可以了。若二分图的匹配小于人的个数,说明给的数据是非法的,输出“impossible”即可。

KM算法求最大匹配就是模板,求最小匹配可以转化为求最大匹配。在建图时,若正常情况下边的权值是a,则求最小匹配时,把边的权值赋为-a,接下来还是用KM求最大匹配,最后求和时也加边的相反数即可。
真的随意套个板子就可以过~

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#include<climits>
using namespace std;
const int MAXN = 61;
const int  inf = 1e9+10;
const double eps = 1e-7;
int cost[MAXN][MAXN], slack[MAXN];
int lx[MAXN], ly[MAXN];
int mat[MAXN];
bool sx[MAXN], sy[MAXN];int tt1[MAXN],tt2[MAXN];
int N;

bool extend(int now)
{
    sx[now] = true;
    int temp;

    for(int i = 0; i < N; i++)
        if(!sy[i])
        {
            temp = -(cost[now][i]-lx[now]-ly[i]);

            if(temp<eps)//此处不能改成0 
            {
                sy[i] = true;
                if(mat[i]==-1 || extend(mat[i]))
                {
                    mat[i] = now;
                    return true;
                }
            }
            else if(temp < slack[i])
                slack[i] = temp;
        }

        return false;
}

int KM() //finding the maximum value of perfect matching
{
    int ret = 0;
    memset(lx, 0, sizeof(lx));
    memset(ly, 0, sizeof(ly));
    memset(mat, -1, sizeof(mat));
    //matching precalculation
    for(int i = 0; i < N; i++)
    {
        lx[i] = -inf;
        for(int j = 0; j < N; j++)
            lx[i] = max(lx[i], cost[i][j]);
    }
    //KM
    for(int i = 0; i < N; i++)
    {
        for(int j = 0; j < N; j++)
            slack[j] = inf;

        while(true)
        {
            memset(sx, false, sizeof(sx));
            memset(sy, false, sizeof(sy));

            if(extend(i)) break;
            int themin = inf+1;

            for(int j = 0; j < N; j++)
                if(!sy[j] && slack[j] < themin)
                    themin = slack[j];

            for(int j = 0; j < N; j++)
            {
                if(sx[j]) lx[j] -= themin;
                if(sy[j]) ly[j] += themin;
                else slack[j] -= themin;
            }
        }
    }

    for(int i = 0; i < N; i++)
        ret += cost[mat[i]][i];
    return ret;
}
int main(){
    int numcase;  
    scanf("%d",&numcase);  
    for(int k = 1; k <= numcase; ++k){  
      int K,g;  
      scanf("%d%d%d",&N,&K,&g);  
      memset(cost,0,sizeof(cost));  
     // memset(money2,0,sizeof(money2));  
      for(int i = 0;i < N; ++i)  
          scanf("%d",&tt1[i]);  
      for(int i = 0;i < N;++i)  
          scanf("%d",&tt2[i]);  
      bool flag = false;  
      for(int i = 0;i < N;++i){  
          flag = false;  
          for(int j = 0;j < N;++j){  
              if(tt1[i] < tt2[j]){  
                  flag = true;  
                  cost[i][j] = min(g,(tt2[j] - tt1[i] - K) * (tt2[j] - tt1[i] - K));  
              }  
              else  
                  cost[i][j] = -10000000;  
          }  
          if(flag == false)  
              break;  
      }  
      printf("Case %d: ",k);  
      if(!flag)  
          printf("impossible\n");  
      else{   
      KM();  
      int sum1 = 0,sum2 = 0;  
      bool isok = true;  
      for(int i = 0;i < N;++i){  
         // printf("matchx[%d]=%d\n",i,matchx[i]);  
          if(cost[mat[i]][i] == -10000000){  
            printf("impossible\n");  
            isok = false;  
            break;  
          }  
      }  
      if(isok){  
      for(int i = 0;i < N;++i){  
          sum1 += cost[mat[i]][i];  
      }  
      memset(cost,0,sizeof(cost));
      for(int i = 0;i < N;++i){  
        for(int j = 0;j < N;++j){  
            if(tt1[i] < tt2[j]){  
              cost[i][j] = 0 - (min(g,(tt2[j] - tt1[i] - K) * (tt2[j] - tt1[i] - K)));   
            }  
            else  
                cost[i][j] = -10000000;  
        }  
      }  
      KM();
      for(int i = 0;i < N;++i){  
          sum2 += (0-cost[mat[i]][i]);  
      }  
      printf("%d %d\n",sum2,sum1);  
      }  
      }  
    }  
    return 0;  

}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值