TC SRM 683 div2 MoveStonesEasy 贪心

有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;
	}
};