扫一眼题目,看到n个点和m个点匹配的问题,你就会有一种预感——是不是要跑二分图匹配?
是的,就是要跑二分图匹配。
第一问很显然,大力让一个导师能够被多个学生匹配,然后跑一个类似于匈牙利的操作。然后如果匹配满了,我们就考虑将这个导师匹配的其他学生尝试匹配同一个优先级的其他节点。
这个复杂度肯定是很松的。
那么关键在于第二问要怎么处理。
观察发现答案具有单调性。也就是说,如果一个选手在较高的优先度会感到沮丧,那么他在一个较低的优先度也会感到沮丧;
因此我们考虑二分答案来检验。
这样的复杂度大概是三方乘对数级的。时间比较紧张。
我们考虑一种优化策略:发现选手提高到某个优先度了以后,他就相当于顶替了本来在这个优先度的选手的位置。
故而,我们可以对于前缀的选手,储存到每个选手时的匹配状态。每次找到某个优先度,就把他的信息加入到对应的匹配状态中,然后跑上述操作。
这样可以优化到平方乘对数级,可能可以通过此题。
这一题的具体操作——也就是那个一点多匹配的操作并不是特别会,以后需要重新研究。
我好菜啊。
#include<iostream>
#include<cstdio>
int c,n,m;
int mx[205],a[205][205],val[205][205][15],exp[205];
int usd[205],usd1[205],usd2[205][205],vis[205];
int prusd[205][205],prusd1[205][205],prusd2[205][205][205];
inline bool dfs(int X,int V){
int nw,nxt;
for(int i=1;i<=val[X][V][0];++i){
nw=val[X][V][i];
if(vis[nw]){
continue;
}
vis[nw]=1;
if(usd[nw]<mx[nw]){
usd1[X]=nw;
usd2[nw][++usd[nw]]=X;
return 1;
}
for(int j=1;j<=mx[nw];++j){
nxt=usd2[nw][j];
if(dfs(nxt,a[nxt][nw])){
usd1[X]=nw;
usd2[nw][j]=X;
return 1;
}
}
}
return 0;
}
inline void calc1(){
for(int i=1;i<=n;++i){
for(int j=1;j<=m;++j){
vis[j]=0;
}
for(int j=1;j<=m;++j){
if(!val[i][j][0]){
continue;
}
if(dfs(i,j)){
break;
}
}
for(int j=1;j<=m;++j){
prusd[i][j]=usd[j];
}
for(int j=1;j<=i;++j){
prusd1[i][j]=usd1[j];
}
for(int j=1;j<=m;++j){
for(int k=1;k<=usd[j];++k){
prusd2[i][j][k]=usd2[j][k];
}
}
}
}
inline bool chck(int X,int Y){
for(int i=1;i<=m;++i){
usd[i]=prusd[Y-1][i];
}
for(int i=1;i<=Y-1;++i){
usd1[i]=prusd1[Y-1][i];
}
for(int i=1;i<=m;++i){
for(int j=1;j<=usd[i];++j){
usd2[i][j]=prusd2[Y-1][i][j];
}
}
for(int i=1;i<=m;++i){
vis[i]=0;
}
for(int i=1;i<=exp[X];++i){
if(!val[X][i][0]){
continue;
}
if(dfs(X,i)){
return 1;
}
}
return 0;
}
void init(){
// 注意清空数组!!!
scanf("%d%d",&n,&m);
for(int i=1;i<=m;++i){
scanf("%d",&mx[i]);
}
for(int i=1;i<=200;++i){
usd1[i]=0;
usd[i]=0;
for(int j=1;j<=200;++j){
for(int k=0;k<=10;++k){
val[i][j][k]=0;
}
}
}
for(int i=1;i<=n;++i){
for(int j=1;j<=m;++j){
scanf("%d",&a[i][j]);
if(a[i][j]){
val[i][a[i][j]][++val[i][a[i][j]][0]]=j;
}
}
}
for(int i=1;i<=n;++i){
a[i][0]=m+1;
}
for(int i=1;i<=n;++i){
scanf("%d",&exp[i]);
}
calc1();
for(int i=1;i<=n;++i){
printf("%d ",a[i][usd1[i]]);
}
puts("");
int l,r,mid,ans;
for(int i=1;i<=n;++i){
if(a[i][usd1[i]]<=exp[i]){
printf("0 ");
continue;
}
l=1,r=i-1,ans=0;
while(l<=r){
mid=(l+r)>>1;
if(chck(i,mid)){
ans=mid;
l=mid+1;
}else{
r=mid-1;
}
}
printf("%d ",i-ans);
}
puts("");
}
int main(){
int T;
scanf("%d%d",&T,&c);
while(T--){
init();
}
return 0;
}