有n堆石子,每堆的石子数为a1,a2,...,an。现在想把它变成b1,b2,...,bn。
如果不能,输出-1。如果能,输出最小移动次数。
每次只能将一个石子从该堆移动到相邻堆上,且空堆不能跳过。如假设5为空堆,则从4移到6视为2次而不是1次。
思想:我是贪心做的,先算两个序列的和,不相等就输出-1。
然后对于ai和bi,把较小的变为0,较大的变为abs(ai-bi)。就是只需要把相差的那部分移走或者移进来就行。这点可以证明,如果你移动的为x+abs(ai-bi),则因为最后要得到b那个序列,所以如果你多移动了x个,则还要把这x个给移回来才能得到bi。那么这x个石子所进行的移动次数则完全是多余的。或者你把这x个石子移动到了m堆上,然后从n堆上得到了x个式子,则相当于直接从n堆上移动x个石子到m堆上,移动次数是一样的。
所以,先进行了这个预处理。然后就从0到n-1依次处理每堆,每堆上,ai和bi至少有一个为0。对于第i堆,如果ai不为0,则这ai个石子肯定会移动到其他堆上,暂且不管。我们只处理bi不为0即可(其实处理哪个都可以,是对称的,只不过按题意是处理bi),bi不为0表明要从其他堆上移动bi个石子到第i堆上,然后从左边第一个ai不为0的堆上开始移动就行,依次扫描,直到把得到了bi个石子为止。如此,得到的就是 最小移动次数
代码:
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <vector>
#include <string>
#include <map>
#include <set>
#include <iostream>
#include <cassert>
using namespace std;
class MoveStonesEasy
{
public:
int get(vector <int> a, vector <int> b)
{
int ans = 0, sum1 = 0, sum2 = 0;
int a1[55], b1[55];
memset(a1, 0, sizeof(a1));
memset(b1, 0, sizeof(b1));
int n = a.size();
for (int i = 0; i < n; ++i)
{
a1[i] = a[i];
b1[i] = b[i];
sum1 += a1[i];
sum2 += b1[i];
}
if (sum1 != sum2)
return -1;
for (int i = 0; i < n; ++i)
{
if (b1[i] >= a1[i])
{
b1[i] -= a1[i];
a1[i] = 0;
}
else
{
a1[i] -= b1[i];
b1[i] = 0;
}
}
int last = 0;
for (int i = 0; i < n; ++i)
{
for (int j = last; j < n; ++j)
{
if (a1[j] == 0)
continue;
if (a1[j] < b1[i])
{
b1[i] -= a1[j];
ans += a1[j] * (int)abs(i - j);
a1[j] = 0;
}
else
{
a1[j] -= b1[i];
ans += b1[i] * (int)abs(i - j);
b1[i] = 0;
last = j;
break;
}
}
}
return ans;
}
};