1. 程式人生 > >U41570 War1(網路流)

U41570 War1(網路流)

題目大意:

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");
				}
			}
}