[模板练习]最小费用最大流

原题链接

大体上就是
在找最大流的时候
顺便找个最短路
基本上就是
改约束条件
注意不要重复约束

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>
#include<queue>
#define MAX 1e12 
using namespace std;

struct nico
{
    int to,val,cost,bk;
};

struct maki
{
    int p,d;
}pre[50005];

int n,m,s,t,flow,discost,dis[50005],f[50005],fl[50005];
vector<nico>edge[50005];
queue<int>q;

void add(int u,int v,int w,int f)
{
    edge[u].push_back((nico){v,w,f,edge[v].size()});
    edge[v].push_back((nico){u,0,-f,edge[u].size()-1}); 
}

int bfs(int sp,int tp)
{
    int i; 

    memset(dis,127,sizeof(dis));
    memset(f,0,sizeof(f));

    for(i=1;i<=n;i++)
    {
        pre[i].p=-1;
        pre[i].d=-1;        
    }

    while(!q.empty())
        q.pop();

    dis[sp]=0;
    f[sp]=1;
    q.push(sp);
    pre[sp].p=0;
    fl[sp]=MAX;


    while(!q.empty())
    {
        int poi=q.front();
        f[poi]=0;
        q.pop();

        for(i=0;i<edge[poi].size();i++)
        {
            nico &tmp=edge[poi][i];

            if(tmp.val&&dis[tmp.to]>dis[poi]+tmp.cost)
            {
                dis[tmp.to]=dis[poi]+tmp.cost;
                fl[tmp.to]=min(fl[poi],tmp.val);
                pre[tmp.to].p=poi;
                pre[tmp.to].d=i;
                if(!f[tmp.to]) f[tmp.to]=1,q.push(tmp.to);

            }
        }
    }

    if(pre[t].p==-1) return 0;
    return fl[t];
}

void maxflow(int sp,int tp)
{
    int k,last,incr;

    while(bfs(sp,tp))
    {
        k=t;

        while(k!=sp)
        {
            last=pre[k].p;          
            nico &tmp=edge[last][pre[k].d];

            tmp.val-=fl[tp];
            edge[k][tmp.bk].val+=fl[tp];

            k=last;
        }

        flow+=fl[tp];
        discost+=dis[tp]*fl[tp];
    }
}

int main()
{
    int u,v,w,f,i;

    scanf("%d%d%d%d",&n,&m,&s,&t);

    for(i=1;i<=m;i++)
    {
        scanf("%d%d%d%d",&u,&v,&w,&f);
        add(u,v,w,f);
    }   

    maxflow(s,t);

    printf("%d %d",flow,discost);

    return 0;
}