bzoj5251 [2018多省省队联测]劈配（网络流+贪心+二分）

我们就贪心地，对于每个人，按第一志愿第二志愿。。。试图匹配，直到能满流。
然后对于第二问我们二分一下。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 210
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int b[N],ans[N],ans2[N],lst[N],n,m,h[N<<1],num=1,T=405,lev[N<<1],cur[N<<1];
struct node{
    int h[N],nxt[N];
    void clear(){memset(h,0,sizeof(h));}
}a[N];
struct edge{
    int to,next,val;
}data[5010];
struct Geo{
    edge data[5010];int h[N<<1],num;
}t[210];
inline void add(int x,int y,int val){
    data[++num].to=y;data[num].next=h[x];h[x]=num;data[num].val=val;
    data[++num].to=x;data[num].next=h[y];h[y]=num;data[num].val=0;
}
inline void solve0(){
    for(int i=1;i<=n;++i){
        a[i].clear();ans[i]=m+1;
        for(int j=1;j<=m;++j){
            int x=read();if(!x) continue;
            a[i].nxt[j]=a[i].h[x];a[i].h[x]=j;
        }for(int j=1;j<=m;++j){
            if(!a[i].h[j]) continue;int x=a[i].h[j];if(!b[x]) continue;
            ans[i]=j;if(--b[x]==0) lst[x]=i;break;
        }
    }for(int i=1;i<=n;++i){
        int s=read();if(ans[i]<=s){ans2[i]=0;continue;}int tmp=0;
        for(int j=1;j<=s;++j)
            if(a[i].h[j]) tmp=max(tmp,lst[a[i].h[j]]);
        ans2[i]=i-tmp;
    }for(int i=1;i<=n;++i) printf("%d ",ans[i]);puts("");
    for(int i=1;i<=n;++i) printf("%d ",ans2[i]);puts("");
}
inline bool bfs(){
    queue<int>q;memset(lev,0,sizeof(lev));
    q.push(0);lev[0]=1;
    while(!q.empty()){
        int x=q.front();q.pop();
        for(int i=h[x];i;i=data[i].next){
            int y=data[i].to;if(!data[i].val) continue;
            if(lev[y]) continue;lev[y]=lev[x]+1;if(y==T) return 1;q.push(y);
        }
    }return 0;
}
inline int dinic(int x,int low){
    if(x==T) return low;int tmp=low;
    for(int &i=cur[x];i;i=data[i].next){
        int y=data[i].to;if(!data[i].val||lev[y]!=lev[x]+1) continue;
        int res=dinic(y,min(tmp,data[i].val));
        if(!res) lev[y]=0;else tmp-=res,data[i].val-=res,data[i^1].val+=res;
        if(!tmp) return low;
    }return low-tmp;
}
inline bool jud(int mid,int x,int s){//Éýµ½midÃûÐв»ÐÐ
    memcpy(h,t[mid-1].h,sizeof(h));num=t[mid-1].num;
    memcpy(data,t[mid-1].data,sizeof(data));add(0,x,1);
    for(int i=1;i<=s;++i)
        for(int j=a[x].h[i];j;j=a[x].nxt[j]) add(x,n+j,1);
    if(!bfs()) return 0;return 1;
}
int main(){
    int tst=read(),C=read();
    while(tst--){
        n=read();m=read();
        for(int i=1;i<=m;++i) b[i]=read();
        if(C==1){solve0();continue;}
        memset(h,0,sizeof(h));num=1;
        for(int i=1;i<=m;++i) add(n+i,T,b[i]);
        memcpy(t[0].data,data,sizeof(data));
        memcpy(t[0].h,h,sizeof(h));t[0].num=num;
        for(int i=1;i<=n;++i){
            a[i].clear();ans[i]=m+1;
            for(int j=1;j<=m;++j){
                int x=read();if(!x) continue;
                a[i].nxt[j]=a[i].h[x];a[i].h[x]=j;
            }for(int j=1;j<=m;++j){
                if(!a[i].h[j]) continue;memcpy(h,t[i-1].h,sizeof(h));num=t[i-1].num;
                memcpy(data,t[i-1].data,sizeof(data));add(0,i,1);
                for(int x=a[i].h[j];x;x=a[i].nxt[x]) add(i,n+x,1);
                if(!bfs()) continue;memcpy(cur,h,sizeof(h));dinic(0,inf);
                ans[i]=j;memcpy(t[i].h,h,sizeof(h));
                memcpy(t[i].data,data,sizeof(data));t[i].num=num;break;
            }if(ans[i]==m+1) t[i]=t[i-1];
        }for(int i=1;i<=n;++i){
            int s=read();if(ans[i]<=s){ans2[i]=0;continue;}
            int l=1,r=i-1;
            while(l<=r){
                int mid=l+r>>1;
                if(jud(mid,i,s)) l=mid+1;
                else r=mid-1;
            }ans2[i]=i-(l-1);
        }for(int i=1;i<=n;++i) printf("%d ",ans[i]);puts("");
        for(int i=1;i<=n;++i) printf("%d ",ans2[i]);puts("");
    }return 0;
}