网络流24题7

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

基本套路啦

#include <iostream>
#include <stdio.h>
#include <cstring>
#include <queue>
using namespace std;
const int maxn = 2000;
int head[maxn];
int tot = 0;
struct edge{
    int v,nex,w;
}e[maxn*22];
void addedge(int u,int v,int w){
    e[tot] = (edge){v,head[u],w};
    head[u] = tot++;
    e[tot] = (edge){u,head[v],0};
    head[v] = tot++;
}
int deep[maxn];
bool bfs(int S,int T){
    memset(deep,0,sizeof(deep));
    deep[S] = 1;
    queue<int> q;
    q.push(S);
    while(!q.empty()){
        int now = q.front();
        q.pop();
        for(int i=head[now];i!=-1;i=e[i].nex){
            int w = e[i].w;
            int v = e[i].v;
            if(deep[v]!=0 || w<=0) continue;
            deep[v] = deep[now]+1;
            q.push(v);
        }
    }
    return deep[T];
}
int dfs(int now,int T,int maxflow){
    if(now==T) return maxflow;
    int all = 0;
    for(int i=head[now];i!=-1 && all<maxflow;i=e[i].nex){
        int v = e[i].v;
        int w = e[i].w;
        if(deep[v]!=deep[now]+1 || w<=0) continue;
        int tt = dfs(v,T,min(maxflow-all,w));
        e[i].w-=tt;
        e[i^1].w+=tt;
        all+=tt;
    }
    return all;
}
int dinic(int S,int T){
    int ret = 0;
    while(bfs(S,T)){
        ret+=dfs(S,T,0x3f3f3f3f);
    }
    return ret;
}
int main()
{
    int k,n;
    scanf("%d%d",&k,&n);
    memset(head,-1,sizeof(head));
    int m = 0;
    tot = 0;
    for(int i=1;i<=k;i++){
        int t;
        scanf("%d",&t);
        m+=t;
        addedge(n+i,n+k+1,t);
    }
    for(int i=1;i<=n;i++){
        addedge(0,i,1);
        int t;
        scanf("%d",&t);
        for(int j=0;j<t;j++){
            int tt;
            scanf("%d",&tt);
            addedge(i,n+tt,1);
        }
    }
    int ans = dinic(0,n+k+1);
    if(ans!=m){
        puts("No Solution!");
    }else{
        for(int i=1;i<=k;i++){
            printf("%d:",i);
            for(int j=head[i+n];j!=-1;j=e[j].nex){
                int v = e[j].v;
                int w = e[j].w;
                if(v<=n && v>=1 && w == 1){
                    printf(" %d",v);
                }

            }
            printf("\n");
        }
    }
    return 0;
}