bzoj1834 [ZJOI2010]network 网络扩容（费用流）

bzoj1834 [ZJOI2010]network 网络扩容

原题地址：http://www.lydsy.com/JudgeOnline/problem.php?id=1834

题意：

给定一张有向图，每条边都有一个容量C和一个扩容费用W。这里扩容费用是指将容量扩大1所需的费用。

求：

1、在不扩容的情况下，1到N的最大流；

2、将1到N的最大流增加K所需的最小扩容费用。

数据范围

N<=1000，M<=5000，K<=10

题解：

1.最大流

2.残量网络上跑费用流

注意，2要新建源汇以向1节点补充K的流量，且保证n到T的流量最大流为K。

数据范围是错的，n<=6000，m<=20000左右。而且测出来不是RE是WA，为此查错了一中午。

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int N=6115;
const int M=50115;
const int inf=0x3f3f3f3f;
queue<int> Q;
int n,m,K,S,preS,T,preT;
int head[N],nxt[4*N],w[4*M],to[4*N],c[4*M],num=1,U[M],V[M],C[M],W[M],ans=0,dis[N],pre[N][2],dep
;
bool inq
,vis
;
void build(int u,int v,int cc,int ww)
{
num++;
to[num]=v; nxt[num]=head[u];
w[num]=ww; c[num]=cc; head[u]=num;
num++;
to[num]=u; nxt[num]=head[v];
w[num]=0; c[num]=-cc; head[v]=num;
}
bool bfs()
{
memset(vis,0,sizeof(vis)); dis[S]=1;
vis[S]=1; Q.push(S);
while(!Q.empty())
{
int u=Q.front(); Q.pop();
for(int i=head[u];i;i=nxt[i])
{
int v=to[i];
if(vis[v]||w[i]<=0) continue;
vis[v]=1; dep[v]=dep[u]+1;
Q.push(v);
}
}
return vis[T];
}
int dfs(int u,int d)
{
if(u==T||d==0) return d; int ret=0;
for(int i=head[u];i;i=nxt[i])
{
int v=to[i];
if(dep[v]!=dep[u]+1||w[i]<=0) continue;
int flow=dfs(v,min(d,w[i]));
d-=flow; ret+=flow;
w[i]-=flow; w[i^1]+=flow;
if(!d) break;
}
if(ret==0) dep[u]=-1;
return ret;
}
bool spfa()
{
memset(dis,0x3f,sizeof(dis));
memset(inq,0,sizeof(inq));
dis[S]=0; inq[S]=1; Q.push(S);
while(!Q.empty())
{
int u=Q.front(); Q.pop(); inq[u]=0;
for(int i=head[u];i;i=nxt[i])
{
if(w[i]<=0) continue;
int v=to[i];
if(dis[v]>dis[u]+c[i])
{
dis[v]=dis[u]+c[i];
pre[v][0]=u,pre[v][1]=i;
if(!inq[v]) inq[v]=1,Q.push(v);
}
}
}
return dis[T]<inf;
}
int get()
{
int flow=inf; int tmp=T; int ret=0;
while(tmp!=S)
{
flow=min(flow,w[pre[tmp][1]]);
tmp=pre[tmp][0];
}
tmp=T;
while(tmp!=S)
{
ret+=flow*c[pre[tmp][1]];
w[pre[tmp][1]]-=flow;
w[pre[tmp][1]^1]+=flow;
tmp=pre[tmp][0];
}
return ret;
}
int main()
{
scanf("%d%d%d",&n,&m,&K); S=1,T=n,preS=n+1,preT=n+2;
for(int i=1;i<=m;i++)
scanf("%d%d%d%d",&U[i],&V[i],&W[i],&C[i]),build(U[i],V[i],0,W[i]);
while(bfs()) ans+=dfs(S,inf);
printf("%d ",ans);
swap(T,preT); swap(S,preS); build(preT,T,0,K); build(S,preS,0,K); ans=0;
for(int i=1;i<=m;i++) build(U[i],V[i],C[i],K);
while(spfa()) ans+=get();
printf("%d\n",ans);
return 0;
}