P4382 [八省联考2018]劈配

题目链接

题意分析

受到了\(olinr\ \ julao\)的影响 写了匈牙利算法

首先 我们对于每一个人 从高到低枚举志愿

如果当前志愿的老师有剩余的话 那么我们就选

否则的话 我们看看谁的那个志愿选了这个老师

我们跑匈牙利算法 看看是否更优

如果当前可以在接受范围内匹配上的话

我们就选择

否则的话 我们就看看 匈牙利增广中遇到的编号最大的人是谁

如果我们增广过程中碰到了争夺的人 那么一定是排名靠前的人

我是这么思考的

我们是在增广的过程中找到了ta

那么取代ta之后 其他争夺的人一定会把我们最开始要的导师挪出来 并且不会影响到自己选志愿

所以我们就可以拥有这个导师 一定是符合要求的

所以我们直接挤掉\(TA\)就可以了

CODE:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdlib>
#include<string>
#include<queue>
#include<map>
#include<stack>
#include<list>
#include<set>
#include<deque>
#include<vector>
#include<ctime>
#define ll long long
#define inf 0x7fffffff
#define N 208
#define IL inline
#define M 1008611
#define D double
#define ull unsigned long long
#define R register
using namespace std;
template<typename T>IL void read(T &_)
{
    T __=0,___=1;char ____=getchar();
    while(!isdigit(____)) {if(____=='-') ___=0;____=getchar();}
    while(isdigit(____)) {__=(__<<1)+(__<<3)+____-'0';____=getchar();}
    _=___ ? __:-__;
}
/*-------------OI使我快乐-------------*/
int T,c;
int n,m,now,maxn;
int lim[N],f[N][N][N],g[N][N];
int ans1[N],ans2[N],have[N],fro[N];
bool vis[N];
IL bool dfs(int x,int at)
{
	if(x<now) maxn=max(maxn,x);
	for(R int i=1;i<=g[x][at];++i)
	{
		int v=f[x][at][i];
		if(vis[v]) continue;
		vis[v]=1;
		if(lim[v])
		{//我们当前选的老师还有空的话
			lim[v]--;
			ans1[x]=at;
			fro[x]=v;return 1;
		}
		else
		{
			for(R int j=1;j<=n;++j)
			{//我们就看看这个人在自己的志愿范围内还可以选择谁 从而让出这个老师
				if(j==x||fro[j]!=v) continue;
				if(dfs(j,ans1[j]))
				{
					ans1[x]=at;
					fro[x]=v;
					return 1;
				}
			}
		}
	}
	return 0;
}
int main()
{
//	freopen(".in","r",stdin);
//	freopen(".out","w",stdout);
	read(T);read(c);
	while(T--)
	{
		read(n);read(m);
		memset(g,0,sizeof g);
		memset(fro,0,sizeof fro);
		for(R int i=1;i<=m;++i) read(lim[i]);
		for(R int i=1;i<=n;++i)
		 for(R int j=1,x;j<=m;++j)
		 {
		 	read(x);
			if(x) f[i][x][++g[i][x]]=j;//我们使用一个数组存储
		 }
		for(R int i=1;i<=n;++i)
		{
			read(have[i]);
			ans2[i]=i;ans1[i]=m+1;
		}
		for(R int i=1;i<=n;++i)
		{
			memset(vis,0,sizeof vis);
			now=i;
			for(R int j=1;j<=m;++j)
			{
				if(!g[i][j]) continue;
				maxn=0;
//接受范围内有老师 那么就不沮丧了
				if(dfs(i,j)) {if(j<=have[i]) ans2[i]=0;break;}
//否则的话我们就看看接受范围内的争夺老师的排名最靠后的人是谁 挤掉ta就可以了
				if(j<=have[i]) ans2[i]=min(ans2[i],i-maxn);
			}
		}
		for(R int i=1;i<=n;++i) printf("%d%c",ans1[i],(i==n ? '\n':' '));
		for(R int i=1;i<=n;++i) printf("%d%c",ans2[i],(i==n ? '\n':' '));
	}
//	fclose(stdin);
//	fclose(stdout);
    return 0;
}

HEOI 2019 RP++