【ACWing】376. 机器任务

题目地址：

https://www.acwing.com/problem/content/378/

有两台机器$A$，$B$以及$K$个任务。机器$A$有$N$种不同的模式（模式$0\sim N-1$），机器$B$有$M$种不同的模式（模式$0\sim M-1$）。两台机器最开始都处于模式$0$。每个任务既可以在$A$上执行，也可以在$B$上执行。对于每个任务$i$，给定两个整数$a[i]$和$b[i]$，表示如果该任务在$A$上执行，需要设置模式为$a[i]$，如果在$B$上执行，需要模式为$b[i]$。任务可以以任意顺序被执行，但每台机器转换一次模式就要重启一次。求怎样分配任务并合理安排顺序，能使机器重启次数最少。

输入格式：
输入包含多组测试数据。每组数据第一行包含三个整数$N,M,K$。接下来$K$行，每行三个整数$i$，$a[i]$和$b[i]$，$i$为任务编号，从$0$开始。当输入一行为$0$时，表示输入终止。

输出格式：
每组数据输出一个整数，表示所需的机器最少重启次数，每个结果占一行。

数据范围：
$N,M<100,K<1000$
$0\le a[i]<N$
$0\le b[i]<M$

可以用二分图建模，左部有$N$个点，右部有$M$个点，如果某个任务在两台机器上的模式是$(a[i],b[i])$，则连一条边。那么一个方案，等价于选一些点，使得每条边都被覆盖（即每个任务都至少有一个机器去完成它）。而最小点覆盖就等于最大匹配数，可以用匈牙利算法来做。注意，由于一开始两个机器都处于模式$0$，所以任何可以用模式$0$完成的任务都是不需要消耗重启次数的，可以直接略过。代码如下：

#include <iostream>
#include <cstring>
using namespace std;

const int N = 110;
int n, m, k;
bool g[N][N], st[N];
int match[N];

bool dfs(int u) {
    for (int i = 1; i <= m; i++)
        if (!st[i] && g[u][i]) {
            st[i] = true;
            if (!match[i] || dfs(match[i])) {
                match[i] = u;
                return true;
            }
        }

    return false;
}

int main() {
    while (cin >> n, n) {
        cin >> m >> k;
        memset(g, 0, sizeof g);
        memset(match, 0, sizeof match);

        while (k--) {
            int t, a, b;
            cin >> t >> a >> b;
            if (!a || !b) continue;
            g[a][b] = true;
        }

        int res = 0;
        for (int i = 1; i <= n; i++) {
            memset(st, 0, sizeof st);
            if (dfs(i)) res++;
        }
        
        cout << res << endl;
    }
    
    return 0;
}

时间复杂度$O(nm)$，$n,m$分别是图的点数和边数，空间$O(n)$。