CSP201703-1分蛋糕(C++100分)
问题描述
小明今天生日,他有n块蛋糕要分给朋友们吃,这n块蛋糕(编号为1到n)的重量分别为a1, a2, …, an。小明想分给每个朋友至少重量为k的蛋糕。小明的朋友们已经排好队准备领蛋糕,对于每个朋友,小明总是先将自己手中编号最小的蛋糕分给他,当这个朋友所分得蛋糕的重量不到k时,再继续将剩下的蛋糕中编号最小的给他,直到小明的蛋糕分完或者这个朋友分到的蛋糕的总重量大于等于k。
请问当小明的蛋糕分完时,总共有多少个朋友分到了蛋糕。
输入格式
输入的第一行包含了两个整数n, k,意义如上所述。
第二行包含n个正整数,依次表示a1, a2, …, an。
输出格式
输出一个整数,表示有多少个朋友分到了蛋糕。
样例输入
6 9
2 6 5 6 3 5
样例输出
3
样例说明
第一个朋友分到了前3块蛋糕,第二个朋友分到了第4、5块蛋糕,第三个朋友分到了最后一块蛋糕。
评测用例规模与约定
对于所有评测用例,1 ≤ n ≤ 1000,1 ≤ k ≤ 10000,1 ≤ ai ≤ 1000。
思路
用一个标记位kg计算每个人分到的蛋糕重量,数组a[i]保存n块蛋糕。
每次循环相当于拿一块蛋糕给一个人,并把这块蛋糕的重量累加到kg上,随机判断kg是否大于k,如果是,此人将不能继续分到蛋糕,将kg清零,分到蛋糕的人数加一即可。
需要注意的是,在for循环中的else if判断时需要满足i!=n-1,因为当i=n-1时,说明此时判断的是分到最后一块蛋糕的那个人,这个人不必加到sum中,我们在初始化sum时令sum为1,实际上就预先计算了分到最后一块蛋糕的这个人。
为什么要像上段这样处理呢?如果我们令sum初始化为0,并将else if (i != n - 1)替换为else,会遗漏一种情况,假如最后一个人分到的蛋糕重量不到k,而此时所有的蛋糕都分完了,那么这个人是不会进入else循环加到sum中的,有兴趣的话可以自己试一下,得不到满分,遗漏的就是这种情况,因此我们不妨预先计算分到最后一块蛋糕的那个人,即令sum初始化为1,然后进入else if分支时保证此时加的不是分到最后一块蛋糕的那个人即可。
#include<iostream>
#include<algorithm>
using namespace std;
int a[1000] = { 0 };
int main()
{
int n,k;
cin >> n >> k;
for (int i = 0; i < n; i++)
cin >> a[i];
int sum = 1;
int kg = 0;
for (int i = 0; i < n; i++)
{
kg += a[i];
if (kg < k)
continue;
else if (i != n - 1)
{
sum++;
kg = 0;
}
}
cout << sum;
}