Codeforces Round #548 (Div. 2) E 二分图匹配(新坑) or 网络流 + 反向处理

https://codeforces.com/contest/1139/problem/E

题意

有n个学生,m个社团,每个学生有一个\(p_i\)值,然后每个学生属于\(c_i\)社团,

有d天,每天首先随机去除一个人,然后再从每个社团挑选一个学生(假如社团没有学生就跳过这个社团),使得这些学生组成的集合的mex值最大,输入每天的mex值

题解

• 假如这道题发现是二分图匹配就很好做
• 左边用\(p_i\)建点,右边用\(c_i\)建点,\(p_i\)从小到达和源点连边,然后跑个二分图匹配or最大流,直到没有新的匹配,就在中间加边
• 倒着处理,每次加边进去(技巧)

代码

#include<bits/stdc++.h>
#define MAXN 5005

using namespace std;
int n,m,i,j,link[MAXN],p[MAXN],c[MAXN],d,k[MAXN],vi[MAXN],ans[MAXN];
int del[MAXN];
vector<int>mp[MAXN];
int match(int u){
	for(auto i:mp[u]){
		if(!vi[i]){
			vi[i]=1;
			if(link[i]==-1||match(link[i])){
				link[i]=u;
				vi[i]=0;
				return 1;
			}
			vi[i]=0;
		}
	}
	return 0;
}
int main(){
	cin>>n>>m;
	memset(link,-1,sizeof(link));
	for(i=1;i<=n;i++)cin>>p[i];
	for(i=1;i<=n;i++)cin>>c[i];
	cin>>d;
	for(i=1;i<=d;i++){cin>>k[i];del[k[i]]=1;}
	for(i=1;i<=n;i++)if(!del[i]){mp[p[i]].push_back(c[i]);}
	j=0;
	for(i=d;i>=1;i--){
		int id=k[i];
		for(;j<5000;j++){
			//memset(vi,0,m*sizeof(int));
			if(!match(j))break;
		}
		ans[i]=j;
		mp[p[id]].push_back(c[id]);
	}
	for(i=1;i<=d;i++)cout<<ans[i]<<endl;
}