loj 6006「网络流 24 题」试题库

(http://www.elijahqi.win/2017/12/15/loj-6006%E3%80%8C%E7%BD%91%E7%BB%9C%E6%B5%81-24-%E9%A2%98%E3%80%8D%E8%AF%95%E9%A2%98%E5%BA%93/%20%E2%80%8E)
题目描述

假设一个试题库中有 n n n 道试题。每道试题都标明了所属类别。同一道题可能有多个类别属性。现要从题库中抽取 m m m 道题组成试卷。并要求试卷包含指定类型的试题。试设计一个满足要求的组卷算法。
输入格式

第 1 1 1 行有 2 2 2 个正整数 k k k 和 n n n。k k k 表示题库中试题类型总数，n n n 表示题库中试题总数。第 2 2 2 行有 k k k 个正整数，第 i i i 个正整数表示要选出的类型 i i i 的题数。这 k k k 个数相加就是要选出的总题数 m m m。

接下来的 n n n 行给出了题库中每个试题的类型信息。每行的第 1 1 1 个正整数 p p p 表明该题可以属于 p p p 类，接着的 p p p 个数是该题所属的类型号。
输出格式

第 i i i 行输出 i: 后接类型 i i i 的题号。如果有多个满足要求的方案，只要输出一个方案。如果问题无解，则输出 No Solution!。
样例
样例输入

3 15
3 3 4
2 1 2
1 3
1 3
1 3
1 3
3 1 2 3
2 2 3
2 1 3
1 2
1 2
2 1 2
2 1 3
2 1 2
1 1
3 1 2 3

样例输出

1: 1 6 8
2: 7 9 10
3: 2 3 4 5

数据范围与提示

2≤k≤20,k≤n≤1000 2 \leq k \leq 20, k \leq n \leq 1000 2≤k≤20,k≤n≤1000

试题向类型建权值为1 的边 源点向试题建权值为1的边 类型向汇点建需要题数的边 然后跑最大流看是否满流 输出方案

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 1100 
#define inf 0x3f3f3f3f
using namespace std;
inline char gc(){
    static char now[1<<16],*S,*T;
    if (T==S){T=(S=now)+fread(now,1,1<<16,stdin);if (T==S) return EOF;}
    return *S++; 
}
inline int read(){
    int x=0;char ch=gc();
    while(ch<'0'||ch>'9') ch=gc();
    while(ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=gc();}
    return x;
}
struct node{
    int y,z,next;
}data[N*20*2];
int num=1,h[N],level[N],k,n,T;
inline void insert1(int x,int y,int z){
    data[++num].y=y;data[num].z=z;data[num].next=h[x];h[x]=num;
    data[++num].y=x;data[num].z=0;data[num].next=h[y];h[y]=num;
}
inline bool bfs(){
    memset(level,0,sizeof(level));level[0]=1;queue<int>q;q.push(0);
    while(!q.empty()){
        int x=q.front();q.pop();
        for (int i=h[x];i;i=data[i].next){
            int y=data[i].y,z=data[i].z;
            if (level[y]||!z) continue;level[y]=level[x]+1;q.push(y);if (y==T) return 1; 
        }
    }return 0;
}
inline int dfs(int x,int s){
    if (x==T) return s;int ss=s;
    for (int i=h[x];i;i=data[i].next){
        int y=data[i].y,z=data[i].z;
        if(level[x]+1==level[y]&&z){
            int xx=dfs(y,min(s,z));if(!xx) level[y]=0;
            s-=xx;data[i].z-=xx;data[i^1].z+=xx;if(!s) return ss;
        }
    }return ss-s;
}
int main(){
    freopen("2763.in","r",stdin);
    k=read();n=read();int sum=0;T=k+n+1;int p;
    for (int i=1;i<=k;++i) p=read(),sum+=p,insert1(n+i,T,p);
    for (int i=1;i<=n;++i){
        int pp=read();for (int j=1;j<=pp;++j) insert1(i,n+read(),1);
        insert1(0,i,1);
    }int ans=0;while(bfs()) 
    ans+=dfs(0,inf);
    if (ans!=sum) printf("No Solution!");else{
        for (int i=1;i<=k;++i){
            printf("%d: ",i);
            for (int j=h[i+n];j;j=data[j].next){
                int y=data[j].y,z=data[j].z;
                if (y<=n&&z) printf("%d ",y);
            }puts("");
        }
    }
    return 0;
}