wikioi1237 餐巾计划问题

费用流。

把每天分为二分图两个集合中的顶点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的有向边。
求网络最小费用最大流，费用流值就是要求的最小总花费。

#include <cstdio>
#include <queue>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
#define MAXN 1100000
#define MAXM 1100000
#define INF 0x3f3f3f3f
struct node
{
	int u,v,f,c,next;
}e[MAXM];
int value[1100],need[1100],p,pf,ps,tf,ts;
int k,head[MAXN],pre[MAXN],dist[MAXN],vis[MAXN];
int en,s,t,maxflow,mincost,m,n; //s源点，t汇点
void add(int u,int v,int c,int f)
{
	e[en].u=u;
	e[en].v=v;
	e[en].c=c;
	e[en].f=f;
	e[en].next=head[u];
	head[u]=en++;
	e[en].u=v;
	e[en].v=u;
	e[en].c=-c;
	e[en].f=0;
	e[en].next=head[v];
	head[v]=en++;
}
int spfa()
{
	int i,u,v;
	for(i=0;i<=t;i++)
		pre[i]=-1,vis[i]=0,dist[i]=INF;
	dist[s]=0;
	vis[s]=1;
	queue<int>q;
	q.push(s);
	while(!q.empty())
	{
		u=q.front();
		q.pop();
		for(i=head[u];i!=-1;i=e[i].next)
		{
			v=e[i].v;
			if(e[i].f>0&&dist[u]+e[i].c<dist[v])
			{
				dist[v]=dist[u]+e[i].c;
				pre[v]=i;
				if(!vis[v])
				{
					vis[v]=1;
					q.push(v);
				}
			}
		}
		vis[u]=0;
	}
	if(dist[t]==INF)
		return 0;
	return 1;
}
void add2()
{
	int v;
	int maxf=INF;
	for(v=pre[t];~v;v=pre[e[v].u])
		maxf=min(maxf,e[v].f);
	for(v=pre[t];~v;v=pre[e[v].u])
	{
		e[v].f-=maxf;
		e[v^1].f+=maxf;
		maxflow+=maxf;
	}
	mincost+=maxf*dist[t];
}

void init()
{
    maxflow=0;
    s=0;
    t=n*2+1;
    memset(head,-1,sizeof(head));
}

int main()
{
	int i,j,a,b,c;
	scanf("%d%d%d%d%d%d",&n,&p,&tf,&pf,&ts,&ps);
	init();
	for(int i=1;i<=n;i++)
    {
        scanf("%d",&need[i]);
    }
    for(int i=1;i<=n;i++)
    {
        add(s,i,0,need[i]);
        add(s,i+n,p,0x3f3f3f3f);
        add(i+n,t,0,need[i]);
    }
    for(int i=1;i<n;i++)
    {
        add(i,i+1,0,0x3f3f3f3f);
    }
    for(int i=1;i+ts<=n;i++)
    {
        add(i,i+n+ts,ps,0x3f3f3f3f);
    }
    for(int i=1;i+tf<=n;i++)
    {
        add(i,i+n+tf,pf,0x3f3f3f3f);
    }
    while(spfa())
        add2();
    printf("%d",mincost);
	return 0;
}
/*
3 10 2 3 3 2
5
6
7
*/