裸题,只要找准建边的方法就好了,跑一遍最大流,枚举从试题编号指向类型相连的单向边,如果其flow(flt[ ])的值为0,那么说明这道题就选了是不是;
洛谷OJ是Special Judge,随便输出哪一个正确的就好了。
#include<bits/stdc++.h>
const int N=1000+4,M=1000000+10,K=30+5;
using namespace std;
queue<int> q;
int head[M],flt[M],nxt[M],to[M],cn=1;
int ans[K][N],dis[N],num[K];
int n,k,x,tmp,src,sink,inf=1e8+7;
bool vis[N];
int minx(int a,int b){
return a < b ? a : b ;
}
void create(int u,int v,int f){
cn++;
to[cn]=v;
flt[cn]=f;
nxt[cn]=head[u];
head[u]=cn;
cn++;
to[cn]=u;
flt[cn]=0;
nxt[cn]=head[v];
head[v]=cn;
}
bool bfs(){
int v;
memset(dis,0,sizeof(dis));
memset(vis,false,sizeof(vis));
q.push(src);
vis[src]=true;
while(!q.empty()){
tmp=q.front();
for(int i=head[tmp];i;i=nxt[i]){
v=to[i];
if(flt[i] && !vis[v]){
dis[v]=dis[tmp]+1;
q.push(v);
vis[v]=true;
}
}
q.pop();
}
return dis[sink]>0;
}
int dinic(int u,int delta){
if(u==sink) return delta;
int res=0,v;
for(int i=head[u];i && delta;i=nxt[i]){
v=to[i];
if(flt[i] && dis[v]==dis[u]+1){
int dd=dinic(v,minx(flt[i],delta));
flt[i]-=dd;
delta-=dd;
flt[i^1]+=dd;
res+=dd;
}
}
return res;
}
int main(){
ios::sync_with_stdio(false);
cin>>k>>n;
for(int i=1+n;i<=k+n;i++)
cin>>num[i];
src=0;sink=k+n+1;
for(int i=1;i<=n;i++)
create(src,i,1);
for(int i=1;i<=n;i++){
cin>>x;
for(int j=1;j<=x;j++){
cin>>tmp;
create(i,tmp+n,1);
}
}
for(int i=n+1;i<=n+k;i++)
create(i,sink,num[i]);
while(bfs())
dinic(src,inf);
for(int i=1;i<=n;i++){
for(int j=head[i];j;j=nxt[j]){
int v=to[j];
if(!flt[j])
ans[v][++ans[v][0]]=i;
}
}
for(int i=1;i<=k;i++){
cout<<i<<": ";
for(int j=1;j<=ans[i+n][0];j++)
cout<<ans[i+n][j]<<" ";
cout<<endl;
}
return 0;
}