acwing 3242. 分蛋糕 C++

小明今天生日，他有 n 块蛋糕要分给朋友们吃，这 n 块蛋糕（编号为 1 到 n）的重量分别为 a1,a2,…,an。

小明想分给每个朋友至少重量为 k 的蛋糕。

小明的朋友们已经排好队准备领蛋糕，对于每个朋友，小明总是先将自己手中编号最小的蛋糕分给他，当这个朋友所分得蛋糕的重量不到 k 时，再继续将剩下的蛋糕中编号最小的给他，直到小明的蛋糕分完或者这个朋友分到的蛋糕的总重量大于等于 k。

请问当小明的蛋糕分完时，总共有多少个朋友分到了蛋糕。

输入格式

输入的第一行包含了两个整数 n,k，意义如上所述。

第二行包含 n 个正整数，依次表示 a1,a2,…,an。

输出格式

输出一个整数，表示有多少个朋友分到了蛋糕。

数据范围

对于所有评测用例，1≤n≤1000，1≤k≤10000，1≤ai≤1000。

输入样例：

6 9
2 6 5 6 3 5

输出样例：

3

样例解释

第一个朋友分到了前 3 块蛋糕，第二个朋友分到了第4、5 块蛋糕，第三个朋友分到了最后一块蛋糕。

思路

每一小块蛋糕重量累加，当累加总重量>=k时，让结果（有多少个朋友分到了蛋糕）加一，并初始化蛋糕重量累加的结果，循环重新累加操作。还要考虑到：当最后一个人获得的蛋糕总重小于k时，也要让结果加一。

代码

#include <iostream>

using namespace std;

const int N = 1010;
int a[N];
int n, k;
int res; // 统计结果（有多少个朋友分到了蛋糕）
int r; // 一个人分到的蛋糕总重量（每一小块蛋糕重量的累加）

int main() {
    cin >> n >> k;
    for (int i = 1; i <= n; i++) {
        cin >> a[i]; // 将每一块蛋糕的重量录入到数组中
    }
    for (int i = 1; i <= n; i++) {
        r += a[i]; // 统计一个人分到的蛋糕重量
        if (r >= k) { // 如果分到的蛋糕总重量>=k，结果加一并进行初始化操作
            res++;
            r = 0; // 对 r 初始化，重新统计下一个人分到的蛋糕重量
        }
    }
    if (r > 0) { // 最后一个人，如果分到的蛋糕重量没有到达 k，仍要让结果 res 加一
        res++;
    }
    cout << res << endl;
    return 0;
}