好久之前做的,现在才写博客……
填一下网络流24题的坑
一图胜千言,做了一张图
怎么建图应该一目了然
因为很久之前就做过这题,所以码风清奇,请见谅
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
struct zzz{
int t,len,nex;
}e[5010<<2]; int head[2100],tot;
void add(int x,int y,int z){
e[++tot].t=y;
e[tot].len=z;
e[tot].nex=head[x];
head[x]=tot;
}
int dis[2100],s,t;
bool bfs(){
memset(dis,0,sizeof(dis));
queue <int> q; q.push(s);
dis[s]=1;
while(!q.empty()){
int k=q.front(); q.pop();
for(int i=head[k];i;i=e[i].nex){
int to=e[i].t;
if(dis[to]&&!e[i].len) continue;
dis[to]=dis[k]+1;
if(to==t) return 1;
q.push(to);
}
}
return dis[t];
}
int dfs(int now,int flow){
if(now==t||!flow) return flow;
int rest=0,fl;
for(int i=head[now];i;i=e[i].nex){
int to=e[i].t;
if(dis[to]==dis[now]+1&&(fl=dfs(to,min(e[i].len,flow)))){
e[i].len-=fl, e[i^1].len+=fl, rest+=fl;
if(rest==flow) return rest;
}
}
if(rest<flow) dis[now]=0;
return rest;
}
int dinic(){
int ans=0;
if(bfs()) ans+=dfs(s,1<<30);
return ans;
}
int read(){
int k=0; char c=getchar();
for(;c<'0'||c>'9';) c=getchar();
for(;c>='0'&&c<='9';c=getchar())
k=k*10+c-48;
return k;
}
int main(){
int k=read(),n=read(),m=0; t=2*n+k+1;
for(int i=1;i<=k;i++){
int kind=read(); m+=kind;
add(s,i,kind); add(i,s,0);
}
for(int i=1;i<=n;i++){
int num=read();
for(int j=1;j<=num;j++){
int kind=read();
add(kind,i+k,1); add(i+k,kind,0);
}
add(i+k,i+k+n,1); add(i+k+n,i+k,0);
add(i+k+n,t,1); add(t,i+k+n,0);
}
int ans=dinic();
if(ans!=m){
printf("No Solution!"); return 0;
}
for(int i=1;i<=k;i++){
printf("%d: ",i);
for(int j=head[i];j;j=e[j].nex){
int to=e[j].t;
if(to!=s&&!e[j].len){
printf("%d ",to-k);
}
}
printf("\n");
}
return 0;
}