84222024-01-15 20:14:18kukkermanVirágos rét (50 pont)cpp17Accepted 50/5020ms4772 KiB
#include <iostream>
#include <vector>
#include <algorithm>

void beolvas(std::istream &be, int &k, std::vector<int> &retek) {
    int n;
    be >> n >> k;

    retek.resize(n);
    for (auto &r : retek) {
        be >> r;
    }
}

void feldolgoz(int k, const std::vector<int> &retek) {
    const int64_t n = std::count(retek.cbegin(), retek.cend(), 1) - k + 1;
    std::cout << (n > 0 ? n * (n + 1) / 2 : 0) << '\n';
}

int main() {
    int k;
    std::vector<int> retek;
    beolvas(std::cin, k, retek);

    feldolgoz(k, retek);

    return 0;
}
SubtaskSumTestVerdictTimeMemory
base50/50
1Accepted0/03ms1680 KiB
2Accepted0/03ms1932 KiB
3Accepted0/020ms2764 KiB
4Accepted2/23ms2276 KiB
5Accepted2/23ms2484 KiB
6Accepted2/23ms2572 KiB
7Accepted3/33ms2804 KiB
8Accepted3/33ms2892 KiB
9Accepted3/33ms3024 KiB
10Accepted2/23ms3120 KiB
11Accepted3/33ms3336 KiB
12Accepted2/23ms3524 KiB
13Accepted3/33ms3744 KiB
14Accepted2/218ms4672 KiB
15Accepted2/218ms4712 KiB
16Accepted3/319ms4676 KiB
17Accepted3/318ms4672 KiB
18Accepted3/320ms4672 KiB
19Accepted3/319ms4768 KiB
20Accepted3/319ms4676 KiB
21Accepted3/320ms4772 KiB
22Accepted3/319ms4772 KiB