[网络流24题 #10]餐巾计划问题

【问题分析】

网络优化问题，用最小费用最大流解决。

【建模方法】

把每天分为二分图两个集合中的顶点Xi,Yi，建立附加源S汇T。

1、从S向每个Xi连一条容量为ri，费用为0的有向边。
2、从每个Yi向T连一条容量为ri，费用为0的有向边。
3、从S向每个Yi连一条容量为无穷大，费用为p的有向边。
4、从每个Xi向Xi+1(i+1<=N)连一条容量为无穷大，费用为0的有向边。
5、从每个Xi向Yi+m(i+m<=N)连一条容量为无穷大，费用为f的有向边。
6、从每个Xi向Yi+n(i+n<=N)连一条容量为无穷大，费用为s的有向边。

求网络最小费用最大流，费用流值就是要求的最小总花费。

【建模分析】

这个问题的主要约束条件是每天的餐巾够用，而餐巾的来源可能是最新购买，也可能是前几天送洗，今天刚刚洗好的餐巾。每天用完的餐巾可以选择送到快洗部或慢洗部，或者留到下一天再处理。

经过分析可以把每天要用的和用完的分离开处理，建模后就是二分图。二分图X集合中顶点Xi表示第i天用完的餐巾，其数量为ri，所以从S向Xi连接容量为ri的边作为限制。Y集合中每个点Yi则是第i天需要的餐巾，数量为ri，与T连接的边容量作为限制。每天用完的餐巾可以选择留到下一天（Xi->Xi+1），不需要花费，送到快洗部(Xi->Yi+m)，费用为f，送到慢洗部(Xi->Yi+n)，费用为s。每天需要的餐巾除了刚刚洗好的餐巾，还可能是新购买的(S->Yi)，费用为p。

在网络上求出的最小费用最大流，满足了问题的约束条件（因为在这个图上最大流一定可以使与T连接的边全部满流，其他边只要有可行流就满足条件），而且还可以保证总费用最小，就是我们的优化目标。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
using namespace std;
const int MAXN=1001*2,MAXM=MAXN*6*2,INF=~0U>>1;
struct edge
{
	edge *next,*op;
	int t,c,v;
}*V[MAXN],ES[MAXM],*P[MAXN];
int N,S,T,EC,Ans,Costflow;
int dist[MAXN];
bool vis[MAXN];
inline void addedge(int a,int b,int c,int v)
{
	ES[++EC].next = V[a]; V[a]=ES+EC; V[a]->t=b; V[a]->c=c; V[a]->v=v;
	ES[++EC].next = V[b]; V[b]=ES+EC; V[b]->t=a; V[b]->c=0; V[b]->v=-v;
	V[a]->op = V[b]; V[b]->op = V[a];
}
void init()
{
	int i,c,newn,sd,sc,fd,fc;
	freopen("napk.in","r",stdin);
	freopen("napk.out","w",stdout);
	scanf("%d%d%d%d%d%d",&N,&newn,&sd,&sc,&fd,&fc);
	S=0; T=N+N+1;
	for (i=1;i<=N;i++)
	{
		scanf("%d",&c);
		addedge(S,i,c,0);
		addedge(i+N,T,c,0);
		addedge(S,i+N,INF,newn);
		if (i<N)
			addedge(i,i+1,INF,0);
		if (i+sd<=N)
			addedge(i,i+N+sd,INF,sc);
		if (i+fd<=N)
			addedge(i,i+N+fd,INF,fc);
	}
}
int Augment(int i,int flow)
{
	if (i==T)
	{
		Costflow += flow * dist[S];
		return flow;
	}
	vis[i]=true;
	for (edge *e=P[i];e;e=e->next)
	{
		int j=e->t;
		if (e->c && !vis[j] && dist[i] == dist[j] + e->v)
		{
			int delta = Augment(j,flow < e->c ? flow : e->c);
			if (delta)
			{
				e->c -= delta;
				e->op->c += delta;
				P[i] = e;
				return delta;
			}
		}
	}
	P[i] = 0;
	return 0;
}
bool ModifyDist()
{
	int delta=INF,i,j;
	for (i=S;i<=T;i++)
	{
		if (vis[i])
		{
			for (edge *e=V[i];e;e=e->next)
			{
				j=e->t;
				if (e->c && !vis[j] && dist[j] + e->v - dist[i] < delta)
					delta = dist[j] + e->v - dist[i];
			}
		}
	}
	if (delta == INF) return false;
	for (i=S;i<=T;i++)
	{
		if (vis[i])
		{
			dist[i] += delta;
			vis[i]=false;
		}
		P[i] = V[i];
	}
	return true;
}
void zkwFlow()
{
	memcpy(P,V,sizeof(V));
	do{
		while (Augment(S,INF));
	}while(ModifyDist());
}
int main()
{
	init();
	zkwFlow();
	printf("%d\n",Costflow);
	return 0;
}