[ZJOI2010]网络扩容

题目描述
给定一张有向图，每条边都有一个容量C和一个扩容费用W。这里扩容费用是指将容量扩大1所需的费用。求： 1、 在不扩容的情况下，1到N的最大流； 2、 将1到N的最大流增加K所需的最小扩容费用。

输入格式：
输入文件的第一行包含三个整数N,M,K，表示有向图的点数、边数以及所需要增加的流量。 接下来的M行每行包含四个整数u,v,C,W，表示一条从u到v，容量为C，扩容费用为W的边。

输出格式：
输出文件一行包含两个整数，分别表示问题1和问题2的答案。

问题1 裸的最大流
问题2 在残量网络上直接增加扩容边，然后在建一个新的S连向之前的S，容量为新增的。
（如果不在残量网络上增加扩容边可能会出事情），因为你最大流增加k后的最大流方案不再是之前那个了。
所以我觉得如果算出最大流以后重建图，在加一个最大流+k的边应该也能跑过？

#include<bits/stdc++.h>
using namespace std;

const int MAXN=1e5+5;
const int INF=1e9+7;
int n,m,k,s,t;


struct edge{
    int u,to,next,w,c;
}e[MAXN<<1],e2[MAXN<<1];

int head[MAXN],cur[MAXN],cnt=1;
inline void add1(int u,int v,int w,int cost){
    e[++cnt]=(edge){u,v,head[u],w,cost},head[u]=cnt;
    e[++cnt]=(edge){v,u,head[v],0,-cost},head[v]=cnt;
}

int head2[MAXN],cnt2=1;
inline void add2(int u,int v,int w,int cost){
    e2[++cnt2]=(edge){u,v,head2[u],w,cost},head2[u]=cnt2;
} 

queue<int>q;
int dep[MAXN];
bool bfs(int x){
    memset(dep,0,sizeof(dep));
    q.push(x);dep[x]=1;
    while(q.size()){
        int u=q.front();q.pop();
        for(int i=head[u];i;i=e[i].next){
            int v=e[i].to,w=e[i].w;
            if(!dep[v]&&w){
                dep[v]=dep[u]+1;q.push(v);
            }
        }
    }
    if(!dep[t])return 0;
    return 1; 
}

int dfs(int u,int flow){
    if(u==t||flow==0)return flow;
    for(int &i=cur[u];i;i=e[i].next){
        int v=e[i].to,w=e[i].w;
        if(dep[v]==dep[u]+1&&w){
            int tem=dfs(v,min(w,flow));
            if(tem){
                e[i].w-=tem;
                e[i^1].w+=tem;
                return tem;
            }
        }
    }
    return 0;
}

int dinic(){
    int ans=0;
    while(bfs(s)){
        for(int i=s;i<=t;i++)cur[i]=head[i];
        while(int d=dfs(s,INF))ans+=d;
    }
    return ans;
}

bool vis[MAXN];
int dis[MAXN],pre[MAXN];

bool SPFA(int x){
    memset(pre,0,sizeof(pre));
    for(int i=s;i<=t;i++)dis[i]=INF;
    q.push(x);dis[x]=0;vis[x]=1;
    while(q.size()){
        int u=q.front();q.pop();
        vis[u]=0;
        for(int i=head2[u];i;i=e2[i].next){
            int v=e2[i].to,w=e2[i].c;
            if(e2[i].w&&dis[u]+w<dis[v]){
                dis[v]=dis[u]+w;pre[v]=i;
                if(!vis[v]){
                    vis[v]=1;q.push(v);
                }
            }
        }   
    }
    if(dis[t]==INF)return 0;
    return 1;
}
int flow=0;

int MC(){
    int ans=0;
    while(SPFA(s)){
        int tem=INF;
        for(int i=pre[t];i;i=pre[e2[i].u]){
            tem=min(e2[i].w,tem);
        }
        for(int i=pre[t];i;i=pre[e2[i].u]){
            e2[i].w-=tem;
            e2[i^1].w+=tem;
            ans+=e2[i].c*tem;
        }
        flow+=tem;
    }
    return ans;
}

int rongliang[MAXN];

int main(){
    scanf("%d%d%d",&n,&m,&k);
    s=1,t=n;
    for(int i=1;i<=m;i++){
        int tem1,tem2,tem3,tem4;
        scanf("%d%d%d%d",&tem1,&tem2,&tem3,&tem4);
        rongliang[cnt+1]=tem3;
        add1(tem1,tem2,tem3,tem4);
    }
    printf("%d ",dinic());
    for(int i=2;i<=cnt;i+=2){
        if(e[i].w>0)add2(e[i].u,e[i].to,e[i].w,0),add2(e[i].to,e[i].u,rongliang[i]-e[i].w,0);
        if(e[i].w<k)add2(e[i].u,e[i].to,k-e[i].w,e[i].c),add2(e[i].to,e[i].u,0,-e[i].c);
    }
    t++;
    add2(t-1,t,k,0);
    printf("%d\n",MC());
    return 0;
}