【网络流24题】试题库（二分图+最大流）

传送门

    试题库

I think

    点集x,y分别放置试题与类型。源点向x集点连容量为1的边，x集点向y中其所属类型连容量为1的边，y集点向T连容量为所需量的边，求解最大流若等于总题数则有解。

Code

#include<cstdio>#include<queue>#include<vector>using namespace std;const int sm = 1100;const int sn = 22000;const int Inf = 0x3f3f3f3f;int N,k,M,S,T,tot=1,Flw; bool fl;int to[sn],nxt[sn],hd[sm],c[sn];int lev[sm],cur[sm];vector<int>Que[sm];int Min(int x,int y) { return x<y?x:y; } void Add(int u,int v,int w) {    to[++tot]=v,nxt[tot]=hd[u],hd[u]=tot,c[tot]=w;    to[++tot]=u,nxt[tot]=hd[v],hd[v]=tot,c[tot]=0;}bool Bfs() {    for(int i=1;i<=T;++i) lev[i]=0;    int t; queue<int>q;    q.push(S);lev[S]=1;    while(!q.empty()) {        t=q.front(),q.pop();        for(int i=hd[t];i;i=nxt[i])            if(c[i]>0&&!lev[to[i]]) {                lev[to[i]]=lev[t]+1;                if(to[i]==T) return 1;                q.push(to[i]);            }    }    return 0;}int Dfs(int x,int mx) {    if(x==T||!mx) return mx;    int f;    for(int i=cur[x]?cur[x]:hd[x];i;i=nxt[i]) {        cur[x]=i;        if(c[i]>0&&lev[to[i]]==lev[x]+1)             if(f=Dfs(to[i],Min(mx,c[i])))                return c[i]-=f,c[i^1]+=f,f;    }    return 0;}void Dinic() {    int f;    while(Bfs()) {        for(int i=1;i<=T;++i) cur[i]=0;        while(f=Dfs(S,Inf)) Flw+=f;    }}int main() {    int u,v;    scanf("%d%d",&k,&N);    S=k+N+1,T=S+1;    for(int i=1;i<=k;++i) {        scanf("%d",&u);        Add(N+i,T,u),M+=u;    }    for(int i=1;i<=N;++i) {        scanf("%d",&v);        Add(S,i,1);        for(int j=1;j<=v;++j)            scanf("%d",&u),Add(i,u+N,1);    }    Dinic();    if(Flw<M) puts("No Solution!");    else {        for(int i=1;i<=N;++i)               for(int j=hd[i];j;j=nxt[j])                if(to[j]!=S&&!c[j]) {                    Que[to[j]-N].push_back(i);                    break;                }        for(int i=1;i<=k;++i) {            printf("%d:",i);            for(int j=0;j<Que[i].size();++j)                printf("%d ",Que[i][j]);            putchar(10);        }    }    return 0;}