题目大意:
你负责反应堆的冷却系统。反应堆冷却系统包括大量管道和流过的特殊冷却液体。管道连接在特殊的点,叫做端点,每个管道有起始点和终止点。液体必须从开始点流到终止点并且不能相反。
端点编号1到N。冷却系统必须设计成液体是循环的并且到达每个端点的液体的量和离开液体的量是相同的。也就是说,如果我们指定从第i个节点到第j个节点的液体的量是fij(如果没有管道从节点i到节点j,我们设fij=0),对于每个i下面的情况必须满足:
fi,1+fi,2+…+fi,N = f1,i+f2,i+…+fN,i
每个管道有容量,因此fij<=cij。为了提供足够的冷却,液体的量至少是lij,满足fij>=lij.
给定所有管道的cij和lij,知道满足上面的fij
输入:第一行包含数字N,节点的个数和M,管道的数量。接下来的M行包含4个整数,i,j,lij,cij。没有管道连接到自己。如果有一个管道从i连到j,那么就没有管道从j连到i
输出:如果有办法启动反应堆,第一行输出YES,否则,第一行输出NO。第一种情况下,将输出M个整数,第K个是第K个管道的液体量。
解题思路:
无源汇流量有上下界网络的可行流
代码如下:
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define INF 99999999
int head[210],cnt;
int n,m;
int cap[21000];
struct node
{
int y,next;
int w,cp;
}a[50010];
void ad(int x,int y,int w)
{
a[cnt].next=head[x];
a[cnt].y=y;
a[cnt].w=w;
head[x]=cnt++;
a[cnt].next=head[y];
a[cnt].y=x;
a[cnt].w=0;
head[y]=cnt++;
}
int st[210],ed[210];
int S,T,sum;
int id[210];
int v[210];
int dfs(int x,int mfw)
{
if(x==T) return mfw;
int ans=0;
for(int i=head[x];i;i=a[i].next)
{
if(a[i].w>0&&id[x]==id[a[i].y]+1)
{
int f=dfs(a[i].y,min(mfw-ans,a[i].w));
a[i].w-=f;
a[i^1].w+=f;
ans+=f;
if(ans==mfw) return ans;
}
}
if(id[S]>=n+2) return ans;
v[id[x]]--;
if(v[id[x]]==0) id[S]=n+2;
id[x]++;
v[id[x]]++;
return ans;
}
int ISAP()
{
int ans=0;
memset(id,0,sizeof(id));
memset(v,0,sizeof(v));
v[0]=n+2;
while(id[S]<n+2)
{
ans+=dfs(S,INF);
}
return ans;
}
bool check()
{
int i,j;
int ans=ISAP();
if(ans==sum) return true;
return false;
}
int main()
{
int i,j,Case;
int x,y,bl,cp;
scanf("%d",&Case);
while(Case--)
{
memset(st,0,sizeof(st));
memset(ed,0,sizeof(ed));
memset(head,0,sizeof(head));
cnt=2;
scanf("%d%d",&n,&m);
S=0;T=n+1;sum=0;
for(i=1;i<=m;i++)
{
scanf("%d%d%d%d",&x,&y,&bl,&cp);
ed[y]+=bl;st[x]+=bl;
ad(x,y,cp-bl);
sum+=bl;
cap[i]=cp;
}
for(i=1;i<=n;i++)
{
ad(S,i,ed[i]);
ad(i,T,st[i]);
}
if(check())
{
printf("YES\n");
for(i=1;i<=m;i++)
{
printf("%d\n",cap[i]-a[i*2].w);
}
}
else
printf("NO\n\n");
}
return 0;
}