【NOIP 2002提高】均分紙牌-貪心
阿新 • • 發佈:2020-07-18
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; }