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