Network flow problems, a detailed look at the code
1 #include<bits/stdc++.h> 2 using namespace std; 3 4 int n,m,tot=-1,h[3005],ans=0,sum=0; 5 struct node{ 6 int from,next,to,rest,full; 7 int last; 8 }e[100005]; 9 void add(int x,int y,int z){ 10 tot++; 11 e[tot].next=h[x]; 12 h[x]=tot; 13 e[tot].from=x; 14 e[tot].to=y; 15 e[tot].rest=z; 16 e[tot].full=z; 17 } 18 19 int dis[3005],g[3005],flow[3005]; 20 bool vis[3005]; 21 22 int bfs(int s,int t){ 23 queue<int>q; 24 dis[s]=0; 25 q.push(s);vis[s]=true; 26 while(!q.empty()){ 27 int u=q.front();vis[u]=false;q.pop(); 28 for(int i=h[u];i!=(-1);i=e[i].next){ 29 if(dis[e[i].to]>dis[u]+1&&g[e[i].to]==(-1)&&e[i].rest>0){ 30 g[e[i].to]=i; 31 flow[e[i].to]=min(flow[u],e[i].rest); 32 dis[e[i].to]=dis[u]+1; 33 if(vis[e[i].to]==false){ 34 vis[e[i].to]=true; 35 q.push(e[i].to); 36 } 37 } 38 } 39 } 40 } 41 42 int EK(int s,int t){ 43 while(1){ 44 memset(vis,false,sizeof(vis)); 45 memset(dis,0x7f,sizeof(dis)); 46 memset(flow,0x7f,sizeof(flow)); 47 memset(g,-1,sizeof(g)); 48 bfs(s,t); 49 if(g[t]==(-1))return 0; 50 ans+=flow[t]; 51 for(int p=t;p!=(s);p=e[g[p]].from){ 52 e[g[p]].rest-=flow[t]; 53 e[g[p]^1].rest+=flow[t]; 54 } 55 56 } 57 } 58 59 int main(){ 60 memset(h,-1,sizeof(h)); 61 cin>>m>>n; 62 for(int i=1;i<=m;i++){ 63 int co;cin>>co;sum+=co; 64 add(0,i,co); 65 add(i,0,0); 66 for(int j=1;j<=n;j++){ 67 add(i,j+m,1); 68 add(j+m,i,0); 69 } 70 } 71 for(int i=1;i<=n;i++){ 72 int co;cin>>co; 73 add(i+m,n+m+1,co); 74 add(n+m+1,i+m,0); 75 } 76 EK(0,n+m+1); 77 if(sum>ans){ 78 cout<<0<<endl; 79 exit(0); 80 } 81 cout<<1<<endl; 82 for(int i=1;i<=m;i++){ 83 for(int j=h[i];j!=(-1);j=e[j].next){ 84 if(e[j].to!=0&&e[j].rest==0){ 85 cout<<e[j].to-m<<" "; 86 } 87 } 88 cout<<endl; 89 } 90 }