题目
思路
《惊!网络瘤24题里有dp》——OI日报
小学5年级时,这个拆环dp用时1周,如今竟是网络瘤,是我太年轻
垂死病中惊坐起,小丑竟是我自己
我们考虑建2个新点s和t,对于每一个仓库,如果该运出就从s连一条容量为运出数量,费用0的边,该运入就向t连一条容量为运入数量,费用0的边,仓库之间按题意,连容量inf,费用1的双向边。
接下来就是最小费用最大流。
code:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;
int n,m,s,t,x,y,tot=2,head[5001],dep[5001],fy;
bool book[5001];
long long w,ans,in[5001],ans2;
long long mn(long long x,long long y)
{
return (x>y?y:x);
}
struct f{
int to,net;
long long w,fy;
} a[100011];
int b[100011];
void add(int x,int y,long long w,long long fy)
{
a[tot].to=y,a[tot].w=w,a[tot].net=head[x],a[tot].fy=fy,head[x]=tot++;
return;
}
bool spfa()
{
memset(dep,0x3f,sizeof(dep));
memset(book,0,sizeof(book));
queue<int> u;
u.push(s);
book[s]=1,dep[s]=0,in[s]=0x7fffffff;
while (u.size())
{
y=u.front();
u.pop();
book[y]=0;
for (int j=head[y];j;j=a[j].net)
{
if (a[j].w!=0&&dep[y]+a[j].fy<dep[a[j].to])
{
dep[a[j].to]=dep[y]+a[j].fy;//记录单价和
in[a[j].to]=min(in[y],a[j].w);
b[a[j].to]=j;
if (!book[a[j].to]) book[a[j].to]=1,u.push(a[j].to);
}
}
}
return dep[t]!=1061109567;
}
void dfs()
{
x=t;
ans+=in[t];
ans2+=dep[t]*in[t];
while (x!=s)
{
a[b[x]].w-=in[t],a[b[x]^1].w+=in[t],x=a[b[x]^1].to;
}//清空增广路
return;
}
int main()
{
scanf("%d",&n);
s=n+1,t=n+2;
for (int i=1;i<=n;i++)
{
scanf("%d",&dep[i]);
ans+=dep[i];
add(x,y,w,1);
add(y,x,0,-1);
}
ans/=n;
for (int i=1;i<=n;i++)
{
add(i,(i==n?1:i+1),1e30,1);
add((i==n?1:i+1),i,0,-1);
add((i==n?1:i+1),i,1e30,1);
add(i,(i==n?1:i+1),0,-1);
if (dep[i]>ans)
{
add(s,i,dep[i]-ans,0);
add(i,s,0,0);
}
if (dep[i]<ans)
{
add(i,t,ans-dep[i],0);
add(t,i,0,0);
}
}
ans=0;
while (spfa())
{
dfs();
}
cout<<ans2;
return 0;
}