AcWing 376. 机器任务

题目描述
有两台机器 A，B 以及 K 个任务。

机器 A 有 N 种不同的模式（模式 0∼N−1），机器 B 有 M 种不同的模式（模式 0∼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≤a[i]<N
0≤b[i]<M
输入样例：

5 5 10
0 1 1
1 1 2
2 1 3
3 1 4
4 2 1
5 2 2
6 2 3
7 2 4
8 3 3
9 4 3
0

输出样例：

3

思路：
将同一个任务两个机器的状态连一条边。形成一张二分图。每条边被认为是一个任务。每条边中，至少要有一个点被选中表示这一个任务至少要有一个机器来做。在这种情况下，选出最少的点。
总结来说，就是在一张二分图中，选出最少的点使每条边的两点中最少有一个点被选中。
即最小点覆盖。
而最小点覆盖=最大匹配数，故可用匈牙利算法。
有一个细节，就是机器一开始设置的状态就是0，所以一条边中如果有任意一端的端点为0，那么就可以忽略这条边。
代码：

#include<iostream>
#include<cstring>
using namespace std;
int n,m,k;
const int N=1e6+10;
bool st[N];
int e[N],ne[N],h[N],idx=0;
int match[N];
void add(int a,int b)
{
    e[idx]=b;
    ne[idx]=h[a];
    h[a]=idx++;
}
bool find(int u)
{
    for(int i=h[u];i!=-1;i=ne[i])
    {
        int j=e[i];
        if(st[j])
        {
            continue;
        }
        st[j]=true;
        if(match[j]==-1 || find(match[j]))
        {
            match[j]=u;
            return true;
        }
    }
    return false;
}
int main()
{
    while(cin>>n)
    {
        memset(h,-1,sizeof h);
        memset(match,-1,sizeof match);
        idx=0;
        if(n==0)
        {
            break;
        }
        cin>>m>>k;
        for(int i=0;i<k;i++)
        {
            int a,b,c;
            cin>>a>>b>>c;
            if(b==0 || c==0)
            {
                continue;
            }
            add(b,c);
        }
        int res=0;
        for(int i=1;i<n;i++)
        {
            memset(st,false,sizeof st);
            if(find(i))
            {
                res++;
            }
        }
        cout<<res<<endl;
    }
}