[最大流]试题库问题

题目描述

假设一个试题库中有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!”。

输入输出样例

输入样例
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

分析

将源点和类别连边，流量为所需题数
将符合题的类别和题连边，流量为∞
将题和汇点连边，流量为1
找方案时要判是否流向源点
需要注意的是，一道题是可以当作p类而不是作为p类的一部分（就是说一题不多用，仅属于一个类别集合）

#include <cstdio>
#include <queue>
#include <memory.h>
#define rep(i,a,b) for (i=a;i<=b;i++)
#define inf 2147483647
using namespace std;
struct edge{
    int u,v,c,nx;
}g[30011];
int list[2011],d[2011],cur[2011],w[21];
int cnt=1;
int s,t;
int ans;
int k,n,m;

void Add(int u,int v,int w)
{
    g[++cnt].u=u;g[cnt].v=v;g[cnt].c=w;g[cnt].nx=list[u];list[u]=cnt;
    g[++cnt].u=v;g[cnt].v=u;g[cnt].c=0;g[cnt].nx=list[v];list[v]=cnt;
}

bool Bfs(){
    queue<int> q;
    int i;
    memset(d,0,sizeof(d));
    while (!q.empty()) q.pop();
    q.push(s);
    d[s]=1;
    while (!q.empty()){
        int x=q.front();q.pop();
        i=list[x];
        while (i>0){
            if (g[i].c&&!d[g[i].v]){
                d[g[i].v]=d[x]+1;
                if (g[i].v==t) return 1;
                q.push(g[i].v);
            }
            i=g[i].nx;
        }
    }
    return 0;
}

int Mcf(int x,int minf){
    int fpoint,fout=0;
    if (x==t||!minf) return minf;
    for (int &i=cur[x];i;i=g[i].nx)
    if (d[x]+1==d[g[i].v]&&g[i].c){
        fpoint=Mcf(g[i].v,min(g[i].c,minf-fout));
        g[i].c-=fpoint;
        g[i^1].c+=fpoint;
        fout+=fpoint;
        if (minf<=fout) break;
    }
    return fout;
}

void Dinic(){
    int i;
    while (Bfs()){
        rep(i,1,2000) cur[i]=list[i];
        ans+=Mcf(s,inf);
    }
}

int main()
{
    int i,j;
    int p,a;
    scanf("%d%d",&k,&n);
    s=1999;
    t=2000;
    rep(i,1,k)
    {
        scanf("%d",&w[i]);
        Add(s,i,w[i]);
        m+=w[i];
    }
    rep(i,1,n)
    {
        scanf("%d",&p);
        rep(j,1,p)
        {
            scanf("%d",&a);
            Add(a,k+i,1);
        }
        Add(k+i,t,1);
    }
    Dinic();
    if (ans!=m)
    {
        printf("No Solution!");
        return 0;
    }
    rep(i,1,k)
    {
        printf("%d:",i);
        j=list[i];
        while (j)
        {
            if (!g[j].c&&g[j].v!=s)
            printf(" %d",g[j].v-k);
            j=g[j].nx;
        }
        printf("\n");
    }
}