1. 程式人生 > >codevs 1098 均分紙牌

codevs 1098 均分紙牌

題目描述 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;
}