网络流24题07:试题库问题

网络流24题07：试题库问题

题目描述

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

对于给定的组卷要求，计算满足要求的组卷方案。

输入

第1行有2个正整数k和n (2 <=k<= 20, k<=n<= 1000)

k 表示题库中试题类型总数，n 表示题库中试题总数。第2 行有k 个正整数，第i 个正整数表示要选出的类型i的题数。这k个数相加就是要选出的总题数m。接下来的n行给出了题库中每个试题的类型信息。每行的第1 个正整数p表明该题可以属于p类，接着的p个数是该题所属的类型号。

输出

第i 行输出 “i：”后接类型i的题号。如果有多个满足要求的方案，只要输出1个方案。如果问题无解，则输出“No Solution!”。

---

Solution

建立一个源s,把s向每个试题连一条权值为1的边
建立一个汇t,把每种试题类型向t连一条权值为c[i]的边(c[i]为这种类型试题的数目)
将每个试题向所属的每个试题类型连一条权值为1的边
跑一边最大流匹配一下
如果最大流!=m则无解，否则输出答案
洛谷上可以测

#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
#define maxn 1000010
int n,k,s,t;
const int inf=1e9+7;
inline int read(){
    int ff=1,ret=0;
    char ch=getchar();
    while(ch<'0'||ch>'9'){
        if(ch=='-') ff=-ff;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9'){
        ret=ret*10+ch-'0';
        ch=getchar();
    }
    return ret;
}
struct Edge{
    int u,v,w,next;
}E[maxn];
int dis[maxn];
int ecnt=0,head[maxn];
void addedge(int u,int v,int w){
    E[++ecnt].u=u;
    E[ecnt].v=v;
    E[ecnt].w=w;
    E[ecnt].next=head[u];
    head[u]=ecnt;
}
void Addedge(int u,int v,int w){
    addedge(u,v,w);
    addedge(v,u,0);
}
bool bfs(){
    queue<int>q;
    for(int i=1;i<=t;++i) dis[i]=inf;
    dis[s]=1;
    q.push(s);
    while(!q.empty()){
        int x=q.front();
        q.pop();
        for(int i=head[x];i;i=E[i].next){
            int v=E[i].v;
            if(dis[v]!=inf||!E[i].w) continue;
            dis[v]=dis[x]+1;
            q.push(v);
        }
    }
    return dis[t]!=inf;
}
int dfs(int x,int nar){
    if(x==t) return nar;
    int used=0;
    for(int i=head[x];i;i=E[i].next){
        int v=E[i].v;
        if(dis[v]!=dis[x]+1||!E[i].w) continue;
        int tmp=nar-used;
        int flow=dfs(v,min(E[i].w,tmp));
        used+=flow;
        E[i].w-=flow;
        E[i+1].w+=flow;
        if(used==nar) return nar;
    }
    if(!used) dis[x]=-1;
    return used;
}
int Dinic(){
    int ans=0;
    while(bfs())ans+=dfs(s,inf);
    return ans;
}
int w[maxn];
int main(){
    freopen("luogu2763.in","r",stdin);
    k=read(),n=read();
    s=n+k+1,t=s+1;
    int sum=0;
    for(int i=1;i<=k;++i){
        w[i]=read();
        Addedge(i+n,t,w[i]);
        sum+=w[i];
    }
    for(int i=1;i<=n;++i){
        int x=read();
        Addedge(s,i,1);
        for(int j=1;j<=x;++j){
            int c=read();
            Addedge(i,c+n,1);
        }
    }
    if(Dinic()==sum){
        for(int i=n+1;i<=n+k;++i){
            printf("%d:",i-n);
            for(int j=head[i];j;j=E[j].next){
                int v=E[j].v;
                if(E[j].w){
                    printf(" %d",v);
                }
            }
            puts("");
        }
    }
    else printf("No Solution!\n");
    return 0;
}