Submission #17,168

2025-05-27 23:37:01

cpp17

Accepted 50/50

18ms

724 KiB

#include <iostream>

using namespace std;

int main()
{
    int N, K;
    cin >> N >> K;
    int* ret = new int[N];
    int viragdb=0;
    for (int i = 0;i < N;i++) {
        cin >> ret[i];
    }
    for (int i = 0;i < N;i++) {
        if (ret[i])viragdb++;
    }
    long long db=0;
    if (viragdb - K >= 0) {
        db = (long long)(viragdb - K + 2) * (viragdb - K+1) / 2;
    }
    cout << db;
    return 0;
}

//http://tehetseg.inf.elte.hu/nemes/2024/nt24-2f2.pdf
//2024, 2. korosztaly
//egydimenzios tombok

	Sum	Test	Verdict		Time	Memory
base	50/50
		1	Accepted	0/0	1ms	500 KiB
		2	Accepted	0/0	1ms	316 KiB
		3	Accepted	0/0	17ms	564 KiB
		4	Accepted	2/2	1ms	316 KiB
		5	Accepted	2/2	1ms	316 KiB
		6	Accepted	2/2	1ms	316 KiB
		7	Accepted	3/3	1ms	316 KiB
		8	Accepted	3/3	1ms	316 KiB
		9	Accepted	3/3	1ms	540 KiB
		10	Accepted	2/2	1ms	316 KiB
		11	Accepted	3/3	1ms	316 KiB
		12	Accepted	2/2	1ms	316 KiB
		13	Accepted	3/3	1ms	316 KiB
		14	Accepted	2/2	17ms	724 KiB
		15	Accepted	2/2	17ms	644 KiB
		16	Accepted	3/3	17ms	680 KiB
		17	Accepted	3/3	17ms	700 KiB
		18	Accepted	3/3	18ms	564 KiB
		19	Accepted	3/3	17ms	680 KiB
		20	Accepted	3/3	17ms	564 KiB
		21	Accepted	3/3	17ms	680 KiB
		22	Accepted	3/3	17ms	660 KiB