排序--线性均分纸牌

题目概述

有N堆纸牌，编号分别为 1,2,…,N。每堆上有若干张，但纸牌总数必为N的倍数。可以在任一堆上取若干张纸牌，然后移动。

移牌规则为：在编号为1堆上取的纸牌，只能移到编号为2的堆上；在编号为N的堆上取的纸牌，只能移到编号为 N-1 的堆上；其他堆上取的纸牌，可以移到相邻左边或右边的堆上。

现在要求找出一种移动方法，用最少的移动次数使每堆上纸牌数都一样多。

例如N=4，4堆纸牌数分别为：

①9②8③17④6

移动3次可达到目的：

从 ③ 取4张牌放到 ④ (9,8,13,10)-> 从 ③ 取3张牌放到 ②(9,11,10,10) -> 从 ② 取1张牌放到①(10,10,10,10)。

输入格式

第一行为：N（N 堆纸牌，1≤N≤100）

第二行为：A₁ ，A₂，…，A_n ( N堆纸牌，每堆纸牌初始数1 ≤ A_i ≤ 10000）

输出格式

一行：即所有堆均达到相等时的最少移动次数。

输入

4
9 8 17 6

输出

3

解题思路

设共 M 人，手中分别有 C[1]，C[2]，…，C[M] 张牌，总数为 T 。
在“均分纸牌”问题中，当所有人手持的纸牌总数 T 能被 M 整除时有解。我们先考虑第一个人：
1、若 C[1] > T/M，则第一个人要给第二个人C[1] - T/M 张纸牌，即C[2]要加上 C[1]-T/M。
2、若 C[1] < T/M，则第一个人要拿第二个人T/M - C[1] 张纸牌，即C[2]要减去 T/M - C[1]。

以此类推，达到目标所需要的最少步骤为：

其中G是C的前缀和，每个“前缀”最初有 G[i] 张，最终有 i*T/M 张

我们设 A[i] = C[i] - T/M，即最终让每个人手里都恰好有 0 张牌，答案显然不变。

S是A的前缀和，即：

最后一个人一定是躺赢的，S[M] 必然等于 0。
前缀和向前推，为0则不用移动，不为0则需要移动。

代码

#include<iostream>
using namespace std;
#include<vector>

void main()
{
	int N;		// 牌堆数
	cin >> N;
	vector<int>A;		// 每堆牌数
	int a;
	vector<int>S(100,0);		// 前缀和
	int sum = 0;
	for (int i = 0; i < N; i++)
	{
		cin >> a;
		A.push_back(a);
		sum += A[i];
	}
	int avg = sum / N;
	S[N - 1] = 0;
	int ans = 0;
	for (int i = N - 2; i >= 0; i--)
	{
		S[i] = S[i + 1] - (A[i] - avg);
		// 如果当前的前缀和不为0，移动数+1
		if (S[i])
			ans++;

	}
	cout << ans << endl;

}