CCF 201703-1分蛋糕

CCF 201703-1分蛋糕

问题描述
　　小明今天生日，他有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。

//
// Created by asimov on 2020/3/24.
//

#ifndef CSP_201703_1_H
#define CSP_201703_1_H

#endif //CSP_201703_1_H

#include "bits/stdc++.h"

using namespace std;
int n, k, friends = 0, wcake = 0;
vector<int> cakes;

int fun() {

    cin >> n >> k;
    for (int i = 0; i < n; i++) {
        int temp;
        cin >> temp;
        cakes.push_back(temp);
    }
    for (int i = 0; i < cakes.size(); i++) {
        if (cakes[i] >= k)
            friends++;
        else if (wcake + cakes[i] >= k) {
            wcake = 0;
            friends++;
        } else if (i == cakes.size() - 1)
            friends++;
        else
            wcake = wcake+cakes[i];

    }
    cout << friends;
    return 0;
}

/*
6 9
2 6 5 6 3 5
 */