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

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
typedef unsigned long long ull;

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

bool find(int x)
{
	for(int i=0;i<m;i++)
	{
		if(!st[i]&&g[x][i])
		{
			st[i]=true;
			if(match[i]==-1||find(match[i]))
			{
				match[i]=x;
				return true;
			}
		}
	}
	return false;
}

int main()
{
	//freopen("test.in","r",stdin);//设置 cin scanf 这些输入流都从 test.in中读取
    //freopen("test.out","w",stdout);//设置 cout printf 这些输出流都输出到 test.out里面去
	ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	while(cin>>n,n)
	{
		cin>>m>>k;
		memset(g,0,sizeof g);
		memset(match,-1,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=0;i<n;i++)
		{
			memset(st,0,sizeof st);
			if(find(i)) res++; 
		}
		cout<<res<<endl;
	}
	return 0;
}