阿里笔试题 仓库均衡问题

题目：

一家卖黄金的淘宝商店，总共有n个仓库，每个仓库都存有一定量的黄金储备。现在要求进行仓库库存的均衡，使得所有仓库存有的黄金量相同。但是要求黄金只能在相邻的仓库之间进行运输。请给出一种最佳的解决方案，使得总的运输量最小。

解析：

首先，我们需要求出每个仓库在均衡之后，应该存有的黄金量。然后根据这个值来确定当前每个仓库的库存富余或短缺的量。为了使得总的运输量最小，我们需要让那些富余的仓库中的富余部分应该往左边和右边的运输量。因此整个过程应该如下：

1.计算仓库均衡后每个仓库应该存储的黄金量，并计算出当前仓库的富余和短缺情况；

2.找出富余最多的仓库位置机器富余量M

3.计算最富余仓库左边所有仓库的富余和短缺情况S

  （1）当S<0，且|S| < M，则此时最富余仓库应该往左边运输|S|，往右边运输M - |S|

  （2）当S<0，且|S| > M，则此时最富余仓库应该运输所有的富余到左边

  （3）当S>0，此时最富余的仓库应该运输所有的富余到右边

4.如果仓库为均衡，则循环步骤2和步骤3

有了上面的分析，则代码实现则不困难了：

//
//问题描述：
//设淘宝一个黄金卖家有n个仓库，每个仓库的存货量不同，现要平衡各个仓库间的
//存货量，使得每个仓库的存货量均相等，请设计算法，在各个仓库间移动货物，使
//得移动货物的代价最小（货物移动量最小），货物只能在相邻的仓库间移动。设计
//算法并代码实现。

#pragma once
#include <iostream>
#include <vector>
#include <numeric>
#include <algorithm>
#include <iterator>
using namespace std;

typedef vector<double>::iterator Iter;

//
//@brief : 计算最小代价
//@return : 返回最小代价
double minCost(vector<double>& barn);

int main()
{
	vector<double> barn;
	barn.reserve(10);
	for(int i = 0; i < 10; i++)
	{
		double x = 0.0;
		cin >> x;
		barn.push_back(x);
	}
	cout << minCost(barn) << endl;
	system("pause");
}

double _minCost(Iter& offset, Iter& first, Iter& last)
{
	if(first == last) return 0.0;

	for(Iter i = first; i != last; i++) cout << *i << " ";
	cout << endl;

	//最多多余储备的仓库往左边运送的黄金量
	double left = 0.0;
	//最多多余储备的仓库往右边运送的黄金量
	double right = 0.0;
	//当前最大仓库需要运送的黄金量
	double curr = 0.0;

	//当前区域拥有最多多余储备的仓库位置
	Iter max_iter = max_element(first, last);
	if(*max_iter < 0.0001) return 0.0;
	curr = *max_iter;
	//最多多余储备仓库应该往其左侧的仓库运送的黄金量
	double left_sum = accumulate(first,max_iter, 0.0);
	int k = distance(offset, max_iter);
	if(max_iter != first)
	{
		//用于最多多余储备的仓库不是第一个仓库
		if(left_sum < 0.0 && *max_iter + left_sum >= 0.0)
		{
			*(max_iter - 1) -= left_sum;
			*max_iter += left_sum;
			cout << "From barn " << k << " move " << 0.0 - left_sum << " to left." << endl;
			left = _minCost(offset, first, max_iter);
		}
		else if(left_sum < 0.0 && *max_iter + left_sum < 0.0)
		{
			*(max_iter - 1) += *max_iter;
			cout << "Frome barn " << k << " move " << *max_iter << " to left." << endl;
			*max_iter = 0.0;
			left = _minCost(offset, first, last);
		}
		else if(left_sum > 0.0)
		{
			cout << "From barn " << k << " move " << *max_iter << " to right." << endl;
			*(max_iter+1) += *max_iter;
			*max_iter = 0.0;
			left = _minCost(offset, first, last);
		}
	}
	if(max_iter != last - 1 && *max_iter >= 0.0)
	{
		*(max_iter + 1) += *max_iter;
		cout << "From barn " << k << " move " << *max_iter << " to right." << endl;
		*max_iter = 0.0;
		right = _minCost(offset, ++max_iter, last);
	}
	double ret  = left + right + curr;
	return ret;
}

double minCost(vector<double>& barn)
{
	if(barn.size() <= 1) return 0;
	//平均每个仓库应该存的黄金量
	double avr = accumulate(barn.begin(), barn.end(), 0.0) / barn.size();
	//计算每个仓库多余或缺的黄金量
	vector<double> t;
	t.reserve(10);
	transform(barn.begin(), barn.end(), back_inserter(t), bind2nd(minus<double>(), avr));
	cout << "Average " << avr <<  endl;

	return _minCost(t.begin(), t.begin(), t.end());
}

问题扩展：

此处假设仓库是直线型的，若假设仓库是一个环形的，即第一个仓库和第n个仓库是邻居仓库，它们之间可以相互运输，那么问题应该如何解决？

若假设每一个仓库之间是一个图的关系，即每一个仓库都与其中一些仓库相邻，则问题又该如何解决呢？