superoj441 餐巾计划

题目描述

输入格式

输入文件第 1 行有 6 个正整数 N，p，m，f，n，s。其中 N 是要安排餐巾使用计划的天数；p 是每块新餐巾的费用；m 是快洗部洗一块餐巾需用天数；f 是快洗部洗一块餐巾需要的费用；n 是慢洗部洗一块餐巾需用天数；s 是慢洗部洗一块餐巾需要的费用。数据范围如下：1≤N≤800，1≤p≤50，1≤m≤20， 1≤f≤20，1≤n≤20，1≤s≤10。

接下来的 N 行是餐厅在相继的 N 天里，每天需用的餐巾数（每个数不超过500）。

输出格式

输出餐厅在相继的 N 天里使用餐巾的最小总花费。

样例数据 1

输入　 [复制]

3 10 2 3 3 2 

5 

6 

7
输出

145

费用流

用类似于可重复背包删边

注意这坑题慢洗可能比快洗要快（坑哭我）

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>

using namespace std;
struct node
{
int to;
int next;
int liu;
int cost;
int from;
};
node bian[1300000];
int first[40010];
int dis[40010];
int xuqiu[40010];
int n,p,m,f,z,s;  //p新餐巾费用 m快洗time f快洗费用 z慢洗time s慢洗费用
int size;
queue <int> q;
bool exsit[40000];
int zcost,zflow;
int pre[40000];

void inser(int a,int b,int liu,int cost)
{
size++;
bian[size].to=b;
bian[size].next=first[a];
first[a]=size;
bian[size].liu=liu;
bian[size].from=a;
bian[size].cost=cost;
}

void init()
{
int i,j;
scanf("%d%d%d%d%d%d",&n,&p,&m,&f,&z,&s);
for(i=1;i<=n;i++)
{
scanf("%d",&xuqiu[i]);
}
for(i=1;i<=n;i++)
{
inser(0,i,xuqiu[i],0);
inser(i,0,0,0);

}
for(i=1;i<=n;i++)
{
inser(n+i,2*n+1,xuqiu[i],0);
inser(2*n+1,n+i,0,0);
}

for(i=1;i<n;i++)
{
inser(i,i+1,10000000,0);
inser(i+1,i,0,0);
}

for(i=1;i<=n;i++)
{
if(i+m<=n)
{
inser(i,i+n+m,10000000,f);
inser(i+n+m,i,0,-f);
}
if(i+z<=n)
{
inser(i,i+n+z,10000000,s);
inser(i+n+z,i,0,-s);
}
inser(0,n+i,10000000,p);
inser(n+i,0,0,-p);
}
inser(4*n,0,100000000,0);
}

bool spfa()
{
while(!q.empty()) q.pop();
memset(dis,127,sizeof(dis));
memset(exsit,false,sizeof(exsit));
memset(pre,-1,sizeof(pre));
pre[0]=size;
dis[0]=0;
exsit[0]=true;
q.push(0);

while(!q.empty())
{
int u=q.front();
q.pop();
exsit[u]=false;
for(int i=first[u];i!=0;i=bian[i].next)
{
int d=bian[i].to;
if(bian[i].liu!=0&&dis[d]>dis[u]+bian[i].cost)
{
dis[d]=dis[u]+bian[i].cost;
pre[d]=i;
if(!exsit[d])
{
exsit[d]=true;
// if(exsit[n*2+1]) return true;
q.push(d);
}
}
}
}
if(dis[2*n+1]>1000000000) return false;
return true;
}

int flow_()
{
int flow;
while(spfa())
{
flow=0x7fffffff;
for(int i=pre[2*n+1];bian[i].from!=4*n;i=pre[bian[i].from])
{
if(bian[i].liu<flow) flow=bian[i].liu;
}
for(int i=pre[2*n+1];bian[i].from!=4*n;i=pre[bian[i].from])
{
bian[i].liu-=flow;
bian[i+1].liu+=flow;
zcost+=flow*bian[i].cost;
}
zflow+=flow;
}

return zcost;
}

int main()
{
// freopen("lx.in","r",stdin);
//  freopen("lx.out","w",stdout);
int i,j,k;

init();
int ans=flow_();
cout<<ans;
//cout<<zflow<<" ";
// cout<<size;
return 0;
}