网络流界的一股清流啊……终于没那么变态了……
考虑一下怎么建图。对于每一个类型,我们从$S$向他连边,容量为它所需的题数,表明它要可以有这么多题,对于每一道题目,我们从它对应的类型向他连边,容量为$1$,表明他可以被对应类型选中,且只能选一次,然后在把每道题目向$T$连容量为$1$的边,表明每一道题目只能被一个类型选中。然后跑一遍最大流,对于每一个类型看一看有哪几条边是有流的,那么对应的点就是这个类型选中的点
还有注意一下……输出格式有点那啥……
1 //minamoto 2 #include<iostream> 3 #include<cstdio> 4 #include<cstring> 5 #include<queue> 6 #define inf 0x3f3f3f3f 7 using namespace std; 8 #define getc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++) 9 char buf[1<<21],*p1=buf,*p2=buf; 10 inline int read(){ 11 #define num ch-'0' 12 char ch;bool flag=0;int res; 13 while(!isdigit(ch=getc())) 14 (ch=='-')&&(flag=true); 15 for(res=num;isdigit(ch=getc());res=res*10+num); 16 (flag)&&(res=-res); 17 #undef num 18 return res; 19 } 20 const int N=1105,M=50005; 21 int head[N],Next[M],ver[M],edge[M],cur[N],tot=1; 22 int dep[N],s,t,n,m,sum; 23 queue<int> q; 24 inline void add(int u,int v,int e){ 25 ver[++tot]=v,Next[tot]=head[u],head[u]=tot,edge[tot]=e; 26 ver[++tot]=u,Next[tot]=head[v],head[v]=tot,edge[tot]=0; 27 } 28 bool bfs(){ 29 while(!q.empty()) q.pop(); 30 memset(dep,-1,sizeof(dep)); 31 for(int i=0;i<=n+m+1;++i) cur[i]=head[i]; 32 q.push(s),dep[s]=0; 33 while(!q.empty()){ 34 int u=q.front();q.pop(); 35 for(int i=head[u];i;i=Next[i]){ 36 int v=ver[i]; 37 if(dep[v]<0&&edge[i]){ 38 dep[v]=dep[u]+1,q.push(v); 39 if(v==t) return true; 40 } 41 } 42 } 43 return false; 44 } 45 int dfs(int u,int limit){ 46 if(!limit||u==t) return limit; 47 int flow=0,f; 48 for(int i=cur[u];i;i=Next[i]){ 49 int v=ver[i];cur[u]=i; 50 if(dep[v]==dep[u]+1&&(f=dfs(v,min(limit,edge[i])))){ 51 flow+=f,limit-=f; 52 edge[i]-=f,edge[i^1]+=f; 53 if(!limit) break; 54 } 55 } 56 return flow; 57 } 58 int dinic(){ 59 int flow=0; 60 while(bfs()) flow+=dfs(s,inf); 61 return flow; 62 } 63 int main(){ 64 //freopen("testdata.in","r",stdin); 65 n=read(),m=read(); 66 s=0,t=n+m+1; 67 for(int i=1;i<=n;++i){ 68 int e=read();sum+=e; 69 add(s,i,e); 70 } 71 for(int i=n+1;i<=n+m;++i){ 72 add(i,t,1); 73 int p=read(); 74 for(int j=1;j<=p;++j){ 75 int k=read(); 76 add(k,i,1); 77 } 78 } 79 if(dinic()!=sum) return puts("No Solution!"),0; 80 for(int i=1;i<=n;++i){ 81 printf("%d: ",i); 82 for(int j=head[i];j;j=Next[j]) 83 if(!edge[j]&&ver[j]) 84 printf("%d ",ver[j]-n); 85 putchar(10); 86 } 87 return 0; 88 }