网络流24题——9.试题库问题

题目链接

https://www.luogu.org/problemnew/show/P2763

试题库问题

可以发现这道题有点类似二分图匹配，唯一不同的是每个类型的数量有限制，我们回顾一下用最大流时的做法，设两边集合为A,B,S向A连流量为1的边，A向B连流量为1的边，B向T连流量为1的边，这样保证了每对AB只会产生一的贡献，那么我们这里所有B（表示类型）向T连的边的流量改为该题目联系要求的数量，其他不变，那么每条S-A-B-T的增广路表示的就是A这道题选B类型，因为S-A的流量为1，就保证了每个A只能流到一个B，B-T则表示这个类型的题目我只需要这么多，那么最后最大流流完判一下maxflow是不是于要求的题目数量相同，如果相同说明是可以做到的，否则No Solution！这个非常显然，至于输出方案，每个A看他这个流量流向哪个B,他就属于哪个类型的题目

#include<cstdio>
#include<algorithm>
#include<cctype>
#include<cstring>
#include<iostream>
#include<vector>
#define LL long long
#define INF (2139062143)
#define N (1000001)
using namespace std;
int tot,k,n,x,S,T,y,all,sum,ans; 
int a[N],fl[N],nxt[N],head[N],h[N],cou[N],cur[N];
vector <int> q[N]; 
template <typename T> void read(T&t) {
    t=0;
    bool fl=true;
    char p=getchar();
    while (!isdigit(p)) {
        if (p=='-') fl=false;
        p=getchar();
    }
    do {
        (t*=10)+=p-48;p=getchar();
    }while (isdigit(p));
    if (!fl) t=-t;
}
inline void add(int x,int y,int f){
    a[++tot]=y,nxt[tot]=head[x],head[x]=tot,fl[tot]=f;
}
int dfs(int u,int maxflow){
    if (u==T) return maxflow;
    int used=0;
    for (int p=cur[u];~p;p=nxt[p]){
        cur[u]=p;
        if (fl[p]>0&&h[a[p]]+1==h[u]){
            int now=dfs(a[p],min(maxflow-used,fl[p]));
            used+=now;
            fl[p]-=now;
            fl[p^1]+=now;
            if (maxflow==used) return maxflow;
        }
    }
    cur[u]=head[u];
    if (--cou[h[u]]==0) h[S]=all;
    h[u]++;
    cou[h[u]]++;
    return used;
}
int main(){
    read(k),read(n);
    S=k+n+1,T=k+n+2;
    tot=1;
    memset(head,-1,sizeof(head));
    memset(cur,-1,sizeof(cur));
    for (int i=1;i<=k;i++){
        read(x);
        sum+=x;
        add(i+n,T,x);
        add(T,i+n,0);
    }
    for (int i=1;i<=n;i++){
        add(S,i,1);
        add(i,S,0);
        read(x);
        while (x--){
            read(y);
            add(i,y+n,1);
            add(y+n,i,0);
        }
    }
    memset(h,0,sizeof(h));
    all=n+k+2;
    cou[0]=all;
    while (h[S]<all) ans+=dfs(S,INF);
    if (ans<sum){
        puts("No Solution!");
        return 0;
    }
    for (int i=1;i<=n;i++){
        for (int p=head[i];p;p=nxt[p]){
            if (fl[p]==0){
                q[a[p]].push_back(i);
            } 
        }
    }
    for (int i=1;i<=k;i++){
        printf("%d: ",i);
        for (int j=0;j<q[i+n].size();j++){
            printf("%d ",q[i+n][j]);
        }
        puts("");
    }
    return 0;
}