P1031 均分纸牌

题目描述

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

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

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

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

① 99 ② 88 ③ 1717 ④ 66

移动 33 次可达到目的：

从 ③ 取 44 张牌放到 ④ （ 9,8,13,109,8,13,10 ）-> 从 ③ 取 33 张牌放到 ②（ 9,11,10,109,11,10,10 ）-> 从 ② 取 11 张牌放到①（ 10,10,10,1010,10,10,10 ）。

输入输出格式

输入格式：

 

两行

第一行为： NN （ NN 堆纸牌， 1 \le N \le 1001≤N≤100 ）

第二行为： A_1,A_2, … ,A_nA1​,A2​,…,An​ （ NN 堆纸牌，每堆纸牌初始数， l \le A_i \le 10000l≤Ai​≤10000 ）

 

输出格式：

 

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

 

输入输出样例

输入样例#1： 复制

4
9 8 17 6

输出样例#1： 复制

3

贪心，移动时只考虑从右向左移动，不考虑右方是否有多余的卡牌，因为反正最后总会有牌移过来，这样只要此堆卡牌数不是平均值就移动一次。

#include <iostream>
#include <string.h>
#include <algorithm>
using namespace std;
int main() {
	int n,sum = 0,ans = 0;
	cin >> n;
	int num[n];
	for(int i = 0; i < n; i ++){
		cin >> num[i];
		sum += num[i];
	}
	sum = sum/n;
	for(int i = 0; i < n-1; i ++){
		if(num[i] != sum){
			ans ++;
			if(num[i] < sum){
				num[i+1] -= (sum - num[i]);
				num[i] = sum;
			}
			else{
				num[i+1] += num[i] - sum;
				num[i] = sum; 
			}
		}
	}
	if(num[n-1] != sum)
		ans ++;
	cout << ans;
	return 0;
}