牛客2018 牛客国庆集训派对Day6 A.Birthday(最小费用流 板子题)

题目

宇扬在蛋糕上插了n(1<=n<=50)支蜡烛，并把蛋糕分为m(2<=m<=50)个区域。

因为某种原因，他必须把第i根蜡烛插在第ai个区域或第bi个区域。区域之间是不相交的。

宇扬在一个区域内同时摆放x支蜡烛就要花费的时间。

布置蛋糕所用的总时间是他在每个区域花的时间的和，求布置蛋糕最少时间。

思路来源

UESTC_Vici代码

题解

其余部分都是流量=1，代价=0这么连，

区域连汇点的时候，

考虑最后如果通过了一条边，代价为1，通过两条边，代价为4，以此类推

则将这个差值，给后续的边当费用，就可以等价了

由于最多选50个蜡烛在同一区域，故最多建50条连向汇点的边

代码

//Copyright(c)2018 Mstdream
#include<bits/stdc++.h>
using namespace std;
const int N=105;
const int M=100050;
int nxt[M],sz=1,to[M],cost[M],flow[M],exsit[N];
int fir[N],q[N*N+10],pre[N],dis[N],n,m,k;
void add1(int x,int y,int z,int w){
    nxt[++sz]=fir[x];fir[x]=sz;
    to[sz]=y;flow[sz]=z;cost[sz]=w;
}
void add(int x,int y,int z,int w){//无向图需正反调用两次add
    add1(x,y,z,w),add1(y,x,0,-w);
    //fprintf(stderr,"%d %d %d %d\n",x,y,z,w);
}
bool spfa(int S,int T){
    memset(dis,63,sizeof dis);
    int s=0,t=1;q[1]=S;dis[S]=0;
    while(s^t){
        int v=q[++s];exsit[v]=0;
        for(int u=fir[v];u;u=nxt[u]){
            if(flow[u]&&dis[to[u]]>dis[v]+cost[u]){
                dis[to[u]]=dis[v]+cost[u];pre[to[u]]=u;
                if(!exsit[to[u]]){
                    exsit[to[u]]=1;q[++t]=to[u];
                }
            }
        }
    }
    return dis[T]<=1000000000;//最短路的INF
}
int end(int S,int T){
    int now=T,mi=1000000000;
    while(now^S){
        mi=min(mi,flow[pre[now]]);
        now=to[pre[now]^1];
    }
    now=T;int ret=0;
    //ans+=mi ans为流量
    while(now^S){
        ret+=mi*cost[pre[now]];
        flow[pre[now]]-=mi;
        flow[pre[now]^1]+=mi;
        now=to[pre[now]^1];
    }
    return ret;
}
int mcmf(int S,int T){
    int ret=0;
    while(spfa(S,T)){
        ret+=end(S,T);
    }
    return ret;
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        add(n+m+1,i,1,0);
        int x,y;scanf("%d%d",&x,&y);
        add(i,n+x,1,0);
        add(i,n+y,1,0);
    }
    for(int i=1;i<=m;i++){
        for(int j=1;j<=50;j++){
            add(n+i,n+m+2,1,j*j-(j-1)*(j-1));
        }
    }
    printf("%d\n",mcmf(n+m+1,n+m+2));
}