U41570 War1(網路流)
阿新 • • 發佈:2018-12-14
題目大意:
ENLIGHTENED總部有NN個Portal,編號為11~NN,編號為ii的Portal初始能量值為A[i]A[i],在Portal之間有MM條LINK,每條LINK著連線著兩個不同Portal,被連線著的兩個Portal可以相互傳輸能量,每個Portal最多總共只能向其連線著的Portal傳輸A[i]A[i]點能量,現在ENLIGHTENED行動指揮想讓每第ii個Portal的能量值變為B[i]B[i],但他不知道著可不可行,所以找到了你,如果可行的話,需要你求出一種可行的能量傳輸方案。
能量只能之間傳輸不能間接傳輸
思路:
這題很明顯的網路流,考慮怎麼建圖,我們每個點自身建一次點,然後分別連向原點和匯點,然後如果X,Y可以傳輸能量就X-Y*,Y-X*連邊,然後直接最大流就好了,最後如果滿流就是可以,然後每條邊的迴流就是傳輸了多少流量。
程式:
#include<cstdio> #include<cstdlib> #include<iostream> #include<algorithm> #include<queue> #include<cstring> using namespace std; const int N=505; const int INF=1e9; int x,y,cnt,sum,ans,s,t,n,m; int a[N],b[N],last[N],dep[N],cur[N],f[N][N]; queue<int>q; struct data{int to,w,next;}e[N*100]; void add(int x,int y,int w){ e[++cnt].to=y; e[cnt].next=last[x]; e[cnt].w=w; last[x]=cnt; e[++cnt].to=x; e[cnt].next=last[y]; e[cnt].w=0; last[y]=cnt; } bool bfs(){ while (!q.empty()) q.pop(); q.push(s); memset(dep,0,sizeof(dep)); dep[s]=1; while (!q.empty()){ int x=q.front(); q.pop(); for (int i=last[x];i;i=e[i].next) if (e[i].w&&!dep[e[i].to]){ dep[e[i].to]=dep[x]+1; q.push(e[i].to); if (e[i].to==t) return 1; } } return 0; } int dfs(int x,int maxf){ if (x==t||!maxf) return maxf; int ret=0; for (int &i=cur[x];i;i=e[i].next) if (e[i].w&&dep[e[i].to]==dep[x]+1){ int f=dfs(e[i].to,min(maxf-ret,e[i].w)); e[i].w-=f; e[i^1].w+=f; ret+=f; if (ret==maxf) break; } return ret; } void dinic(){ while (bfs()){ for (int i=s;i<=t;i++) cur[i]=last[i]; ans+=dfs(s,INF); } } int main(){ scanf("%d%d",&n,&m); s=0; t=2*n+1; cnt=1; for (int i=1;i<=n;i++){ scanf("%d",&a[i]); add(s,i,a[i]); } for (int i=1;i<=n;i++){ scanf("%d",&b[i]); add(i+n,t,b[i]); sum=sum+b[i]; } for (int i=1;i<=n;i++) add(i,n+i,INF); for (int i=1;i<=m;i++){ scanf("%d%d",&x,&y); add(x,n+y,INF); add(y,n+x,INF); } dinic(); for (int i=1;i<=n;i++) for (int j=last[i];j;j=e[j].next) if (e[j].to>n&&e[j].to<=2*n) f[i][e[j].to-n]=e[j^1].w; if (ans!=sum) printf("NO\n"); else { printf("YES\n"); for (int i=1;i<=n;i++){ for (int j=1;j<=n;j++) printf("%d ",f[i][j]); printf("\n"); } } }