题目描述
G 公司有n个沿铁路运输线环形排列的仓库,每个仓库存储的货物数量不等。如何用最少搬运量可以使 n 个仓库的库存数量相同。搬运货物时,只能在相邻的仓库之间搬运。
输入格式:
文件的第 1 行中有 1个正整数 n,表示有 n 个仓库。
第 2 行中有 n 个正整数,表示 n 个仓库的库存量。
输出格式:
输出最少搬运量。
说明
1001≤n≤100
题目分析
建立超级源点
向每个仓库连边,容量为仓库库存量,费用为0
保证每个仓库一开始的库存量
然后每两个相邻的仓库连边,容量为inf,费用为1
这里每一次搬运费用就加一
所以最后的搬运量就是总费用
最后每个仓库向超级汇点连边
容量为最后平均的库存量,费用为0
保证每个仓库的库存量相等且总和相等
注意仓库环状排布,第n个仓库与第1个仓库要连边
仓库可以互相搬运,图是无向的
#include<iostream>
#include<iostream>
#include<vector>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
using namespace std;
int read()
{
int f=1,x=0;
char ss=getchar();
while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
return f*x;
}
const int inf=1128481603;
int k,n;
int s,t;
struct node{int v,f,c,nxt;}E[1000010];
int head[100010],tot=1;
int dis[100010],vis[100010];
int incf[100010],pre[100010],sum;
int fee;
void add(int u,int v,int f,int c)
{
E[++tot].nxt=head[u];
E[tot].f=f;
E[tot].v=v;
E[tot].c=c;
head[u]=tot;
}
bool bfs()
{
memset(dis,67,sizeof(dis)); dis[s]=0;
queue<int> q; q.push(s); incf[s]=inf;
while(!q.empty())
{
int u=q.front();
q.pop(); vis[u]=0;
for(int i=head[u];i;i=E[i].nxt)
{
int v=E[i].v;
if(E[i].f&&dis[v]>dis[u]+E[i].c)
{
dis[v]=dis[u]+E[i].c;
incf[v]=min(incf[u],E[i].f); pre[v]=i;
if(!vis[v]) q.push(v),vis[v]=1;
}
}
}
return dis[t]!=inf;
}
void dfs()
{
int u=t;
while(u!=s)
{
int i=pre[u];
E[i].f-=incf[t]; E[i^1].f+=incf[t];
u=E[i^1].v;
}
fee+=dis[t]*incf[t];
}
int main()
{
n=read(); t=n+1;
for(int i=1;i<=n;i++)
{
int f=read(); sum+=f;
add(s,i,f,0);add(i,s,0,0);
}
sum/=n;
for(int i=1;i<=n;i++)
{
add(i,t,sum,0);add(t,i,0,0);
if(i!=n)
{
add(i,i+1,inf,1);add(i+1,i,0,-1);
add(i+1,i,inf,1);add(i,i+1,0,-1);
}
else if(i==n)
{
add(i,1,inf,1);add(1,i,0,-1);
add(1,i,inf,1);add(i,1,0,-1);
}
}
while(bfs())dfs();
cout<<fee;
return 0;
}