bzoj5251: [2018多省省队联测]劈配
Description
一年一度的综艺节目《中国新代码》又开始了。
Zayid从小就梦想成为一名程序员,他觉得这是一个展示自己的舞台,于是他毫不犹豫地报名了。
题目描述
轻车熟路的Zayid顺利地通过了海选,接下来的环节是导师盲选,这一阶段的规则是这样的:
总共n名参赛选手(编号从1至n)每人写出一份代码并介绍自己的梦想。接着由所有导师对这些选手进行排名。
为了避免后续的麻烦,规定不存在排名并列的情况。
同时,每名选手都将独立地填写一份志愿表,来对总共m位导师(编号从1至m)作出评价。
志愿表上包含了共m档志愿。
对于每一档志愿,选手被允许填写最多C位导师,每位导师最多被每位选手填写一次(放弃某些导师也是被允许的)。
在双方的工作都完成后,进行录取工作。
每位导师都有自己战队的人数上限,这意味着可能有部分选手的较高志愿、甚至是全部志愿无法得到满足。节目组对”
前i名的录取结果最优“作出如下定义:
前1名的录取结果最优,当且仅当第1名被其最高非空志愿录取(特别地,如果第1名没有填写志愿表,那么该选手出局)。
前i名的录取结果最优,当且仅当在前i-1名的录取结果最优的情况下:第i名被其理论可能的最高志愿录取
(特别地,如果第i名没有填写志愿表、或其所有志愿中的导师战队均已满员,那么该选手出局)。
如果一种方案满足‘‘前n名的录取结果最优’’,那么我们可以简称这种方案是最优的。
举例而言,2位导师T老师、F老师的战队人数上限分别都是1人;2位选手Zayid、DuckD分列第1、2名。
那么下面3种志愿表及其对应的最优录取结果如表中所示:
可以证明,对于上面的志愿表,对应的方案都是唯一的最优录取结果。
每个人都有一个自己的理想值si,表示第i位同学希望自己被第si或更高的志愿录取,如果没有,那么他就会非常沮丧。
现在,所有选手的志愿表和排名都已公示。巧合的是,每位选手的排名都恰好与它们的编号相同。
对于每一位选手,Zayid都想知道下面两个问题的答案:
在最优的录取方案中,他会被第几志愿录取。
在其他选手相对排名不变的情况下,至少上升多少名才能使得他不沮丧。
作为《中国新代码》的实力派代码手,Zayid当然轻松地解决了这个问题。
不过他还是想请你再算一遍,来检验自己计算的正确性。
Input
每个测试点包含多组测试数据
第一行2个用空格隔开的非负整数T;C,分别表示数据组数、每档志愿最多允许填写的导师数目。
接下来依次描述每组数据,对于每组数据:
第1行两个用空格隔开的正整数n;m。
n;m分别表示选手的数量、导师的数量。
第2行m个用空格隔开的正整数:其中第i个整数为bi。
Bi表示编号为i的导师战队人数的上限。
第3行至第n+2行,每行m个用空格隔开的非负整数:其中第i+2行左起第j个数为ai,j
ai,j表示编号为i的选手将编号为j的导师编排在了第ai,j志愿。特别地,如果ai,j=0,则表示该选手没有将该导师填入志愿表。
在这一部分,保证每行中不存在某一个正数出现超过C次(0可能出现超过C次),同时保证所有ai,j<=m。
第n+3行n个用空格隔开的正整数,其中第i个整数为Si
Si表示编号为i的选手的理想值。
在这一部分,保证Si<=m。
T<=5,m<=n<=200,Bi<=N
Output
按顺序输出每组数据的答案。对于每组数据,输出2行:
第1行输出n个用空格隔开的正整数,其中第i个整数的意义为:
在最优的录取方案中,编号为i的选手会被该档志愿录取。
特别地,如果该选手出局,则这个数为m+1。
第2行输出n个用空格隔开的非负整数,其中第i个整数的意义为:
使编号为i的选手不沮丧,最少需要让他上升的排名数。
特别地,如果该选手一定会沮丧,则这个数为i。
Sample Input
3 5
2 2
1 1
2 2
1 2
1 1
2 2
1 1
1 2
1 2
2 1
2 2
1 1
0 1
0 1
2 2
Sample Output
2 1
1 0
1 2
0 1
1 3
0 1
三组数据分别与【题目描述】中的三个表格对应。
对于第1 组数据:由于选手1 没有填写第一志愿,所以他一定无法被第一志愿录取,也就一定会沮丧。
选手2 按原排名就不沮丧,因此他不需要提升排名。
对于第2 组和第3 组数据:1 号选手都不需要提升排名。
而希望被第一志愿录取 的2 号选手都必须升到第1 名才能如愿。
题解
显然是网络流。
对于第一问,每个导师向汇点连一条容量\(b_i\)的边。然后依次考虑每个选手。对于某个选手,先从源点向他连一条容量为1的边,然后从高到低考虑每个志愿,每次向一个志愿中所有老师连边,判断流量是否增加,如果增加,答案即是这个志愿,否则再将这些边删掉。
对于第二问,由于\(i\)要实现梦想,因此他必须被前\(s_i\)志愿录取。我们把选手\(i\)与这些志愿中包含的导师全部进行连边。接着,我们依次检查第\(1,…,i-1\)名选手的志愿是否会发生变化:检查选手\(k\)时,只需要连\(k\)在第一小问中求出的答案志愿的边即可。如果检查到\(k\)时增广失败,则表示\(i\)不能完成梦想,一旦出现这种情况,则立刻可知\(i\)选手第二问的答案即为\(i-k\)。
很显然,这个算法的复杂度是\(Ο(f(n,nC)n)\)的。其中\(f(n,m)\)表示的是\(Ο(n)\)个点\(Ο(m)\)条边的二分图跑匹配的复杂度。用\(dinic\)算法最终的复杂度为\(O(n^{2.5}C)\)。
需要注意的是,第一问中求出每个选手的答案之后,必须将非答案志愿中的边都删掉,否则会使边数达到\(O(mnC)\)的级别。
代码
#include
#define MAXN 410
#define INF 0x3f3f3f3f
namespace IO{
char buf[1<<15],*fs,*ft;
inline char gc(){return (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),fs==ft))?0:*fs++;}
inline int qr(){
int x=0,rev=0,ch=gc();
while(ch<'0'||ch>'9'){if(ch=='-')rev=1;ch=gc();}
while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+ch-'0';ch=gc();}
return rev?-x:x;}
}using namespace IO;
using namespace std;
struct Edge{int t,next,v;}e[100010];
struct Data{int cnt,head[MAXN];Edge e[100010];}E[201];
int head[MAXN],cnt=1,S,T,d[MAXN];
inline void Add_Edge(int from,int to,int cap){
e[++cnt].t=to;e[cnt].next=head[from];head[from]=cnt;e[cnt].v=cap;
e[++cnt].t=from;e[cnt].next=head[to];head[to]=cnt;e[cnt].v=0;
}
inline void Init(){
cnt=1;
memset(head,0,sizeof(head));
}
queueq;
inline bool BFS(){
memset(d,-1,sizeof(d));
q.push(S);d[S]=0;
while(!q.empty()){
int u=q.front();q.pop();
for(int i=head[u];i;i=e[i].next){
int v=e[i].t;
if(e[i].v&&d[v]==-1){
d[v]=d[u]+1;
q.push(v);
}
}
}
return d[T]!=-1;
}
int DFS(int u,int now){
if(u==T||!now)return now;
int f,flow=0;
for(int i=head[u];i;i=e[i].next){
int v=e[i].t;
if(d[v]==d[u]+1&&e[i].v&&(f=DFS(v,min(e[i].v,now-flow)))){
e[i].v-=f;e[i^1].v+=f;flow+=f;
if(flow==now)return flow;
}
}
if(!flow)d[u]=-1;
return flow;
}
int Q,C,N,M,b[MAXN],s[MAXN],ans1[210],ans2[210];
vector a[210][210];
inline void Build(){
Init();
for(int i=1;i<=M;i++)Add_Edge(i+N,T,b[i]);
}
inline void Solve1(){
int last,now=0;
for(int i=1;i<=N;i++)Add_Edge(S,i,1);
for(int i=1;i<=M;i++)Add_Edge(i+N,T,b[i]);
E[0].cnt=cnt;
for(int i=1;i<=T;i++)E[0].head[i]=head[i];
for(int i=1;i<=cnt;i++)E[0].e[i]=e[i];
memset(ans1,0,sizeof(ans1));
for(int i=1;i<=N;i++){
last=now;
for(int j=1;j<=M;j++){
int siz=a[i][j].size();
for(int k=0;klast){ans1[i]=j;break;}
cnt=E[i-1].cnt;
for(int k=1;k<=T;k++)head[k]=E[i-1].head[k];
}
if(!ans1[i])ans1[i]=M+1;
E[i].cnt=cnt;
for(int j=1;j<=T;j++)E[i].head[j]=head[j];
for(int j=1;j<=cnt;j++)E[i].e[j]=e[j];
}
}
inline void Solve2(){
int last,now;
for(int i=1;i<=N;i++){
if(ans1[i]<=s[i]){ans2[i]=0;continue;}
Build();now=0;ans2[i]=i;
Add_Edge(S,i,1);
for(int j=1;j<=s[i];j++){
int siz=a[i][j].size();
for(int k=0;k