1. 程式人生 > 實用技巧 >【NOIP 2002提高】均分紙牌-貪心

【NOIP 2002提高】均分紙牌-貪心

Description

  有 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)。

Input

  第一行為一個整數N(N 堆紙牌,1 <= N <= 100)
  第二行為n個用空格分開的整數,依次為A1 A2 … An (N 堆紙牌,每堆紙牌初始數,l<= Ai <=10000)

Output

  所有堆均達到相等時的最少移動次數。

Sample Input

4
9 8 17 6

Sample Output

3


思路

我們採用“移動一次使得一堆牌數達到平均值”的貪心策略:先把每堆的牌數減去平均數,然後由左而右的順序移動紙牌。
若第i堆紙牌的張數a[i]不為0,則將值移動到下一堆,即a[i+1]+=a[i],並且ans++。


程式碼

#include <iostream>
#include <cstdio>
using namespace std;
int n,a[105],sum,ans;
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;++i) scanf("%d",&a[i]),sum+=a[i];
	sum/=n;
	for(int i=1;i<=n;++i) a[i]-=sum;
	for(int i=1;i<=n;++i)
		if(a[i]!=0) a[i+1]+=a[i],++ans;
	printf("%d\n",ans);
	return 0;
}