多重匹配模板题……也是从薛昊那偷来的题……

貌似有更好的做法……看到了什么二进制压缩什么的……

这题数据很大,时间卡的很紧,使得我对模板使用了解更深了一点。

题意:
地球上有n个人要移民,有m的星球可以提供移民,这些人都很挑剔,只去自己感兴趣的星球,问是否存在一个方案使得每个人都能去自己感兴趣的星球。
每个星球容纳人数有限。
人数1e5,星球数 10。

思路:
二分图多重匹配裸题。这里如果你对模板使用比较了解肯定是一发水过的……

这里要对数组强调一下。 记 左元数组 l ,右元数组 r

  • mat[n][m] 二维可达矩阵。
  • link[m][n] 二维连接矩阵,记录每个 r 元素 连接的 l 元素
  • vis[m] 单次dfs的访问记录数组
  • vlink[m] 记录每个 r 元素连接的 l 元素数量
  • num[m] 记录每个 r 元素的限制数量

简单说,只有可达矩阵是n开头的!!!!!!

靠着输入挂800+ms……

AC Code

#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>

#define each(i, n) for (int(i) = 0; (i) < (n); (i)++)
#define reach(i, n) for (int(i) = n - 1; (i) >= 0; (i)--)
#define range(i, st, en) for (int(i) = (st); (i) <= (en); (i)++)
#define rrange(i, st, en) for (int(i) = (en); (i) >= (st); (i)--)
#define fill(ary, num) memset((ary), (num), sizeof(ary))

using namespace std;
const int maxn = 1e5 + 5;
const int maxm = 12;
const int inf = 0x3f3f3f3f;

int n, m;
int mat[maxn][maxm];
int link[maxm][maxn];
int vlink[maxm], vis[maxm], num[maxm];

inline bool scan_d(int& num)
{
    char in;
    bool IsN = false;
    in = getchar();
    if (in == EOF)
        return false;
    while (in != '-' && (in < '0' || in > '9'))
        in = getchar();
    if (in == '-') {
        IsN = true;
        num = 0;
    } else
        num = in - '0';
    while (in = getchar(), in >= '0' && in <= '9') {
        num *= 10, num += in - '0';
    }
    if (IsN)
        num = -num;
    return true;
}

bool dfs(int u)
{
    each(v, m) if (mat[u][v] && !vis[v])
    {
        vis[v] = true;
        if (vlink[v] < num[v]) {
            link[v][vlink[v]++] = u;
            return true;
        }
        each(j, vlink[v]) if (dfs(link[v][j]))
        {
            link[v][j] = u;
            return true;
        }
    }
    return false;
}

bool hungary()
{
    fill(vlink, 0);
    each(i, n)
    {
        fill(vis, false);
        if (!dfs(i))
            return false;
    }
    return true;
}

int main()
{
    while (scanf("%d %d", &n, &m) != EOF) {
        each(i, n) each(j, m) scan_d(mat[i][j]);
        each(i, m) scan_d(num[i]);
        if (hungary())
            puts("YES");
        else
            puts("NO");
    }
    return 0;
}