网络流二十四题 ————(十五)、P1251 餐巾计划问题 dinic+spfa 跑费用流 飞快

突然发现费用流可以用dinic写,虽然快不了太多,但能快一点是一点。比如这题是用时最短的。

这题本身很简单:

左部图点表示每天剩的衣服,右部图表示每天要用的衣服。

源点连向每个点,容量为r[i],花费0,容量表示每天剩的脏衣服总量。

源点连向左部图点p,容量inf,花费0,p连向右部图每个点,容量inf,花费P,表示新买衣服。

右部图每个点相连汇点t,容量r[i],花费0,表示每天需要r[i]个新衣服

左部图每个点i,连向i+N,容量inf,花费S,表示慢洗。

左部图每个点i,连向i+M,容量inf,花费F,表示快洗。

左部图每个点i,连向i+1,容量inf,花费0,表示衣服攒到第二天(不然多洗的衣服没法存下来!!!)。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1000005;
const int M=1000005;
const ll inf=0x7fffffff;
struct MCMF{
	struct EDGE {
		int nxt, to, cost, flow;
	}ee[M << 1];
	int n,s,t;
	int cur[N],head[N],vis[N],a[N];
	ll dis[N];
	ll mincost=0,maxflow=0;
	inline void AD(int from, int to, int flow, int cost) {
		ee[++cnt].nxt = head[from];
		ee[cnt].to = to;
		ee[cnt].cost = cost;
		ee[cnt].flow = flow;
		head[from] = cnt;
	}
	int cnt=1;
	inline void add(int u,int v,int flow,int cost){
	    AD(u,v,flow,cost);
	    AD(v,u,0,-cost);
	}
	inline int spfa()
	{
	    for(int i=1;i<=n;++i)dis[i]=inf;
	    queue<int>q;q.push(s);
	    dis[s]=0;vis[s]=1;
	    while(!q.empty()){
	        int u=q.front();
	        vis[u]=0;q.pop();
	        for(int i=head[u];i;i=ee[i].nxt)
	        {
	            int v=ee[i].to;
	            if(!ee[i].flow)continue;
	            if(ee[i].flow&&dis[v]>dis[u]+ee[i].cost){
	                dis[v]=dis[u]+ee[i].cost;
	                if(!vis[v]){
	                    vis[v]=1;
	                    q.push(v);
	                }
	            }
	        }
	    }
	    return dis[t]!=inf;
	}
	inline ll dfs(int u,ll flow)
	{
	    if(u==t)return flow;
	    ll rest=flow;
	    vis[u]=1;
	    for(int i=cur[u];i&&rest;i=ee[i].nxt){
	        cur[u]=i;
	        int v=ee[i].to;
	        ll w=ee[i].cost,flow=ee[i].flow;
	        if(ee[i].flow&&(dis[v]==dis[u]+w)&&!vis[v]){
	            ll k=dfs(v,min(flow,rest));
	            if(k){
	            	maxflow+=w;
	                mincost+=k*w;
	                ee[i].flow-=k;
	                ee[i^1].flow+=k;
	                rest-=k;
	            }
	        }
	    }
	    vis[u]=0;
	    return flow-rest;
	}
	void gao(){
		while(spfa()){
	        for(int i=1;i<=n;++i)cur[i]=head[i];
	        dfs(s,inf);
	    }
	}
	void init(int nn,int ss,int tt){
		n=nn,s=ss,t=tt;
		for(int i=0;i<=n;i++)head[i]=0;
	}
}mc;
int a[2121];
int main() 
{
//	freopen("1.in","r",stdin);
	int n,s,t,p;
	//p买衣服的点 
	cin>>n;
	s=n*2+1,t=n*2+2,p=n*2+3;
	mc.init(n*2+3,s,t);
	long long inf = 0x3f3f3f3f3f3f3f3f;
	//cout<<inf<<endl;
	mc.add(s,p,inf,0);
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	int P,M,F,N,S;
	cin>>P>>M>>F>>N>>S;
	for(int i=1;i<=n;i++){
		mc.add(s,i,a[i],0);
		mc.add(i+n,t,a[i],0);
		mc.add(p,i+n,inf,P);
		if(i+M<=n)mc.add(i,i+n+M,inf,F);
		if(i+N<=n)mc.add(i,i+n+N,inf,S);
		if(i+1<=n)mc.add(i,i+1,inf,0);
	}
	mc.gao();
	cout<<mc.mincost<<endl;
	return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值