codevs 1098 均分紙牌
阿新 • • 發佈:2018-12-22
題目描述 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 Description
第一行N(N 堆紙牌,1 <= N <= 100)
第二行A1 A2 … An (N 堆紙牌,每堆紙牌初始數,l<= Ai <=10000)
輸出描述 Output Description
輸出至螢幕。格式為:
所有堆均達到相等時的最少移動次數。‘
整體思路就是先計算n堆牌的平均數s,然後從第一堆開始向後遍歷,用start記錄起點的牌堆編號,同時用變數now記錄start到i牌堆的總牌數,如果到第i堆時now<s*(i-start+1),那麼說明這start到i堆需要從之外的牌堆拿牌,則繼續向後遍歷;若now=s*(i-start+1),則說明第i堆的牌比其應有的牌數多,則最後一堆得牌應該向前傳,此時該組需要傳遞的次數就是i-start;若now>s*(i-start+1),則將多餘的牌now-s*((i-start+1))向後傳,然後再向前傳,此時需要的次數是i-start+1。同時start=i+1,now=now-s*((i-start+1));
程式碼如下
#include<cstdio> #include<iostream> using namespace std; int cards[101]; int main() { int start=1,now=0,ans=0,s=0,n;scanf("%d",&n); for(int i=1;i<=n;i++) { scanf("%d",&cards[i]); s+=cards[i]; } s/=n; for(int i=1;i<=n;i++) { now+=cards[i]; if(now==s*(i-start+1)) { ans+=i-start; start=i+1; now=0; } if(now>s*(i-start+1)) { ans+=i-start+1; now=now-s*(i-start+1); start=i+1; } } printf("%d",ans); return 0; }