NEUOJ1212 Coach's plan (二分答案+二分图匹配)

题目链接

2017ICPCECIC北方邀请赛I题

题目大意

有n个学生，m台电脑，每个学生使用每台电脑都对应一个适合值，现在让你给每个学生分配一台电脑（每台电脑仅可以分配给一个学生），问这些学生中最小的适合值最大是多少。

分析

最小值最大化问题，很容易想到二分答案，于是最优化问题就转换为判定性问题：即能否给每个学生分配一台电脑（每台电脑仅可以分配给一个学生）且每对人机的适合值都大于等于mid。
这种分配问题，往往可以联系到二分图。把n个人看成一类结点，把m台电脑看成一类结点，则人机之间舒适度的关系就是二分图。因此每次判定时将适合度权值大于等于mid的边连上构成一个二分图，若这个二分图的最大匹配数等于n，则这样的分配方案能实现。
PS：如果题目改成：“有一个n*m的矩阵，每个格子有一个数，现在要在矩阵中每一行取一个数，且同一列最多只能取1个数求所取的n个数的最小值的最大值”这样应该会更有思维含量，因为把二维矩阵转化为二分图是一种化归，二分图比较多的题目以这样的形式出现，故作此总结。

代码

#include <iostream>
#include <cstdio>
#include <cstring>
const int INF=1001;
const int MAXN=1005;
const int MAXM=2*500*500+10;
using namespace std;
struct Edge
{
    int to,next;
}edge[MAXM];
int w[MAXN][MAXN],head[MAXN],match[MAXN],vis[MAXN],edgenum,n,m;

void Add_edge(int u,int v)
{
    edge[++edgenum].to=v;
    edge[edgenum].next=head[u];
    head[u]=edgenum;
}

bool dfs(int u)
{
    for (int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].to;
        if (!vis[v])
        {
            vis[v]=1;
            if (match[v]==-1||dfs(match[v]))
            {
                match[v]=u;
                match[u]=v;
                return true;
            }
        }
    }
    return false;
}

int hungarian()//匈牙利算法
{
    int cnt=0;
    memset(match,-1,sizeof(match));
    for (int i=1;i<=n+m;i++)
    {
        if (match[i]==-1)
        {
            memset(vis,0,sizeof(vis));
            if (dfs(i))
                cnt++;
        }
    }
    return cnt;
}
bool ok(int x)//转化为判定性问题
{
    edgenum=0;
    memset(head,-1,sizeof(head));
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++)
            if (w[i][j]>=x)
            {
                Add_edge(i,j+n);//注意二分图是无向图
                Add_edge(j+n,i);
            }
    return (hungarian()==n);
}

int main()
{
    while (scanf("%d%d",&n,&m)!=EOF)
    {
        for (int i=1;i<=n;i++)
            for (int j=1;j<=m;j++)
                scanf("%d",&w[i][j]);
        /*bi_search_ans*/
        int low=0,high=INF;
        while (high-low>1)
        {
            int mid=(low+high)>>1;
            if (ok(mid))
                low=mid;
            else
                high=mid;
        }
        printf("%d\n",low);
    }
    return 0;
}