[洛谷P2604] [ZJOI2010]网络扩容（最大流+费用流）

https://www.luogu.org/problemnew/show/P2604

先直接跑一边最大流，为了第二问令每条边的费用为0。

然后继续用残量网络，在原先有边的点之间再建一条容量为inf，费用为原边扩容费用的边。因为残量网络中的边费用为0，所以会优先走残量网络的。为了保证将·容量扩大k，我们再建一个新汇点，从原汇点向新汇点连容量为k费用为0的边。然后跑费用流就可以了。

#include
#include
#include
#include
using namespace std;
const int N=1010,M=5010,inf=0x7fffffff;
struct edge{
	int y,f,c,next;
}data[M*4];
int n,m,k,num,num1,s,t1,t2,h[N],c[M];
int cur[N],dep[N],cost[N],pre[N],path[N],fmax[N];
bool vis[N];
queue q;
inline int read(){
	int x=0,f=0;
	char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return f?-x:x;
}
inline void add(int x,int y,int f,int c){
	data[++num].y=y;data[num].f=f;data[num].c=c;
	data[num].next=h[x];h[x]=num;
	data[++num].y=x;data[num].f=0;data[num].c=-c;
	data[num].next=h[y];h[y]=num;
}
bool bfs(int s,int t){
	memset(dep,0,sizeof dep);
	for(int i=s;i<=t1;i++)cur[i]=h[i];
	while(!q.empty())q.pop();
	q.push(s);dep[s]=1;
	while(!q.empty()){
		int u=q.front();q.pop();
		for(int i=h[u];i!=-1;i=data[i].next){
			int v=data[i].y;
			if(!dep[v]&&data[i].f>0){
				dep[v]=dep[u]+1;
				q.push(v);
			}
		}
	}
	return dep[t];
}
int dfs(int u,int t,int lim){
	if(u==t)return lim;
	int res=0;
	for(int& i=cur[u];i!=-1;i=data[i].next){
		int v=data[i].y;
		if(dep[v]==dep[u]+1&&data[i].f>0){
			int flow=dfs(v,t,min(lim-res,data[i].f));
			if(flow>0){
				data[i].f-=flow;data[i^1].f+=flow;
				res+=flow;
				if(lim==res)return res;
			}
		}
	}
	return res;
}
int dinic(int s,int t){
	int mf=0;
	while(bfs(s,t))mf+=dfs(s,t,inf);
	return mf;
}
bool spfa(int s,int t){
	memset(vis,false,sizeof vis);
	memset(cost,0x3f,sizeof cost);
	while(!q.empty())q.pop();
	q.push(s);vis[s]=true;cost[s]=0;fmax[s]=inf;pre[t]=-1;
	while(!q.empty()){
		int u=q.front();q.pop();vis[u]=false;
		for(int i=h[u];i!=-1;i=data[i].next){
			int v=data[i].y;
			if(data[i].f>0&&cost[v]>cost[u]+data[i].c){
				cost[v]=cost[u]+data[i].c;
				fmax[v]=min(fmax[u],data[i].f);
				pre[v]=u;path[v]=i;
				if(!vis[v]){vis[v]=true;q.push(v);}
			}
		}
	}
	return pre[t]!=-1;
}
int mcmf(int s,int t){
	int mc=0;
	while(spfa(s,t)){
		mc+=cost[t]*fmax[t];
		for(int i=t;i!=s;i=pre[i]){
			data[path[i]].f-=fmax[t];
			data[path[i]^1].f+=fmax[t];
		}
	}
	return mc;
}
int main(){
	n=read();m=read();k=read();s=1;t1=n;t2=n+1;
	memset(h,-1,sizeof h);num=-1;
	for(int x,y,f,i=1;i<=m;i++){
		x=read();y=read();f=read();c[i]=read();
		add(x,y,f,0);
	}
	num1=num;
	printf("%d ",dinic(s,t1));
	add(t1,t2,k,0);
	for(int i=0,j=1;i<=num1;i+=2,j++){add(data[i^1].y,data[i].y,inf,c[j]);
	}
	printf("%d",mcmf(s,t2));
	return 0;
}