ccf试题1：分蛋糕

题目

问题描述
      小明今天生日，他有n块蛋糕要分给朋友们吃，这n块蛋糕（编号为1到n）的重量分别为a₁,a₂,…,a_n。小明想分给每个朋友至少重量为k的蛋糕。小明的朋友们已经排好队准备领蛋糕。对于每个朋友，小明总是先将自己手中编号最小的蛋糕分给他，当这个朋友所分得蛋糕的重量不到k时，再继续将剩下的蛋糕中编号最小的给他，直到小明的蛋糕分完或者这个朋友分到的蛋糕的总重量大于等于k。
      请问当小明的蛋糕分完时，总共有多少个朋友分到了蛋糕。
输入格式
      输入的第一行包含了两个整数n、k，意义如上所述。
      第二行包含n个正整数，依次表示a₁,a₂,…,a_n。
输出格式
      输出一个整数，表示有多少个朋友分到了蛋糕。
样例
输入：

6 9
2 6 5 6 3 5

输出：

3

说明：
      第一个朋友分到了前3块蛋糕，第二个朋友分到了第4、5块蛋糕，第三个朋友分到了最后一块蛋糕。
评测用例规模与约定
      对于所有评测用例，1 <= n <= 1000，1 <= k <=10000，1 <= a_i <= 1000。

分析

      1、计算有几人分到了蛋糕，即维护一个变量weight，weight表示正在被分配蛋糕的朋友手中蛋糕的重量。
      2、将输入的数据依次加到weight上，当weight >= k时，表示朋友A的蛋糕已经分配完成，此时分到蛋糕的人数加1，然后为下一个朋友B分配蛋糕，此时朋友B还没有蛋糕，所以将weight置为0，然后从剩下的蛋糕中编号最小的开始为朋友B分蛋糕。
      3、如果蛋糕已经分配完成，但此时weight不为0，说明有一位朋友C在最后分到了蛋糕，但他有点黑，分到的蛋糕重量小于k，但他依然得到了蛋糕，所以最后人数还要加1。
      4、输出人数。

代码

/*
20190817
ccf试题1：分蛋糕 
*/
#include <iostream>
using namespace std;

int main(){
	//接收输入数据 
	int n, k;
	cin >>n;
	cin >>k;
	int arr[n];
	for(int i=0; i<n; i++){
		cin >>arr[i];
	} 
	
	//计算人数
	int index = 0;
	int weight = 0;
	for(int i=0; i<n; i++){
		weight += arr[i];
		if(weight >= k){
			index++;
			weight = 0;
		}
	} 
	if(weight != 0){
		index += 1;
	}
	 
	//输出结果 
	cout <<index<<endl;
	return 0;
} 

总结

      题目不是太难，当然代码还可以再改进。显然此问题可以直接在接收数据时就计算蛋糕重量，从而计算出得到蛋糕的人数。用数组保存蛋糕的重量，即浪费了空间，同时两次for循环在时间上也有浪费。