试题库问题

问题描述

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

编程任务

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

数据输入

    由文件input.txt提供输入数据。 文件第1行有2个正整数n和k (2 <=k<= 20, k<=n<= 1000)k 表示题库中试题类型总数，n 表示题库中试题总数。
第 2 行有 k 个正整数，第 i 个正整数表示要选出的类型i的题数。这k个数相加就是要选出的总题数m。接下来的n行给出了题库中每个试题的类型信
息。每行的第1 个正整数p表明该题可以属于 p类，接着的 p个数是该题所属的类型号。

结果输出

    程序运行结束时，将组卷方案输出到文件 output.txt 中。文件第 i 行输出 “i：”后接类型i的题号。如果有多个满足要求的方案，只要输出1
个方案。如果问题无解，则输出“No Solution!”。 

输入文件示例

input.txt 
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 

输出文件示例

output.txt
1: 1 6 8  
2: 7 9 10 
3: 2 3 4 5 

题解

这题和圆桌问题一样水。
使用二分图的策略，在对应的要求和试题之间连容量为1的边，分别向源点和汇点连边，dinic一遍，不需要返回值，从源点开始判断所有出边，若有容量未满的边就输出”No Solution!”，否则像圆桌问题一样将所得答案排序后输出。
我的程序在COGS上能过，但是在洛谷上只有90分，好迷啊

CODE：

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int INF=1e9;
struct edge
{
    int next,to,remain;
}a[20000];
int head[3001];
int deep[3001];
int ans[1001];
int n,k,x,y,num=1,S,T,tot;
queue<int>q;
inline int min(int a,int b){return a<b?a:b;}
inline void add(int x,int y,int cap)
{
    a[++num].next=head[x],a[num].to=y,a[num].remain=cap,head[x]=num;
    a[++num].next=head[y],a[num].to=x,head[y]=num;
}
inline bool bfs()
{
    memset(deep,0x3f,sizeof(deep));
    q.push(S),deep[S]=0;
    while(!q.empty())
    {
        int tmp=q.front();q.pop();
        for(int i=head[tmp];i;i=a[i].next)
          if(deep[a[i].to]>INF&&a[i].remain) q.push(a[i].to),deep[a[i].to]=deep[tmp]+1;
    }
    return deep[T]<INF;
}
int dfs(int now,int limit)
{
    if(now==T||!limit) return limit;
    int flow=0,f;
    for(int i=head[now];i;i=a[i].next)
      if(deep[a[i].to]==deep[now]+1&&a[i].remain&&(f=dfs(a[i].to,min(limit,a[i].remain))))
      {
        flow+=f,limit-=f,a[i].remain-=f,a[i^1].remain+=f;
        if(!limit) return flow;
      }
    deep[now]=-1;return flow;
}
inline void dinic(){while(bfs()) dfs(S,INF);}
int main()
{
    scanf("%d%d",&k,&n);
    //1~k:试题类型 k+1~n:试题 
    S=k+(n<<1)+1,T=S+1; 
    for(int i=1;i<=k;i++)
      scanf("%d",&x),add(S,i,x);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&y);
        for(int j=1;j<=y;j++)
          scanf("%d",&x),add(x,k+i,1);
    }
    for(int i=1;i<=n;i++)
      add(k+i,T,1);
    dinic();
    for(int i=head[S];i;i=a[i].next)
      if(a[i].remain) {printf("No Solution!");return 0;}
    for(int i=1;i<=k;i++)
    {
        tot=0;
        for(int j=head[i];j;j=a[j].next)
          if(!a[j].remain) ans[++tot]=a[j].to-k;
        sort(ans+1,ans+tot+1);
        printf("%d:",i);
        for(int j=1;j<=tot;j++)
          printf(" %d",ans[j]);
        printf("\n");
    }
    return 0;
}