网络流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;
}