bzoj1834: [ZJOI2010]network 网络扩容

网络流+费用流

先把第一问跑出来，然后在残余网络加上超级原点和汇点，分别连1和n，然后流量为k，还要在每一条边建费用为w的边（残余网络当然费用是为零的），跑最小费用最大流（实际上最大流就是k了）就可以了。

注意一下边数的范围要开大一点。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
int n,m,K;
struct map
{
    int c,d;
    map(){c=0;d=2147483647;}
}mp[1100][1100];


struct node
{
    int x,y,c,d,next,other;
}a[21000];int len,last[1100];
void ins(int x,int y,int c,int d)
{
    int k1,k2;
    
    len++;k1=len;
    a[len].x=x;a[len].y=y;a[len].c=c;a[len].d=d;
    a[len].next=last[x];last[x]=len;
    
    len++;k2=len;
    a[len].x=y;a[len].y=x;a[len].c=0;a[len].d=d;
    a[len].next=last[y];last[y]=len;
    
    a[k1].other=k2;
    a[k2].other=k1;
}


int st,ed;
int h[1100];
int list[1100];
bool bt_h()
{
    int head=1,tail=2;list[1]=st;
    memset(h,0,sizeof(h));h[st]=1;
    while(head!=tail)
    {
        int x=list[head];
        for(int k=last[x];k;k=a[k].next)
        {
            int y=a[k].y;
            if(h[y]==0&&a[k].c>0)
            {
                h[y]=h[x]+1;
                list[tail]=y;
                tail++;
            }
        }
        head++;
    }
    if(h[ed]==0)return false;
    return true;
}
int findflow(int x,int f)
{
    if(x==ed)return f;
    int s=0;
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(h[y]==h[x]+1&&a[k].c>0&&s<f)
        {
            int t=findflow(y,min(a[k].c,f-s));
            s+=t;a[k].c-=t;a[a[k].other].c+=t;
        }
    }
    if(s==0)h[x]=0;
    return s;
}
void solve1()
{
    len=0;memset(last,0,sizeof(last));
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            if(mp[i][j].c!=0)
                ins(i,j,mp[i][j].c,0);
                
    st=1;ed=n;
    int ans=0;
    while(bt_h()==true)
    {
        ans+=findflow(st,999999999);
    }
    printf("%d ",ans);
}


int ans,S,E;
int c[1100],p[1100],d[1100];
bool v[1100];
bool bfs()
{
    memset(d,63,sizeof(d));d[S]=0;
    memset(v,false,sizeof(v));v[S]=true;
    int head=1,tail=2;
    list[1]=S;c[S]=2147483647;p[S]=0;
    while(head!=tail)
    {
        int x=list[head];
        for(int k=last[x];k;k=a[k].next)
        {
            int y=a[k].y;
            if(d[y]>d[x]+a[k].d&&a[k].c>0)
            {
                d[y]=d[x]+a[k].d;
                c[y]=min(c[x],a[k].c);
                p[y]=k;
                if(v[y]==false)
                {
                    v[y]=true;
                    list[tail]=y;
                    tail++;if(tail==1050)tail=1;
                }
            }
        }
        v[x]=false;
        head++;if(head==1050)head=1;
    }
    if(d[E]>999999999)return false;
    else
    {
        ans+=d[E]*c[E];
        int y=E;
        while(1)
        {
            int k=p[y];
            if(k==0)break;
            a[k].c-=c[E];a[a[k].other].c+=c[E];
            y=a[k].x;
        }
        return true;
    }
}
void solve2()
{
    S=n+1;E=n+2;
    ins(S,st,K,0);ins(ed,E,K,0);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            if(mp[i][j].c!=0)
                ins(i,j,K,mp[i][j].d);
    ans=0;
    while(bfs()==true);
    printf("%d
",ans);
}
int main()
{
    int x,y,c,dd;
    scanf("%d%d%d",&n,&m,&K);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d%d",&x,&y,&c,&dd);
        mp[x][y].c+=c;
        mp[x][y].d=min(mp[x][y].d,dd);
    }
    solve1();
    solve2();
    return 0;
}