CSES - Putka Open 2025 – 4/5

Submission details
Task:	Muutokset
Sender:	ollpu
Submission time:	2025-11-07 19:35:32 +0200
Language:	C++ (C++20)
Status:	READY
Result:	100

Feedback
group	verdict	score
#1	ACCEPTED	20
#2	ACCEPTED	30
#3	ACCEPTED	50

Test results
test	verdict	time	group
#1	ACCEPTED	0.01 s	1, 2, 3	details
#2	ACCEPTED	0.01 s	1, 2, 3	details
#3	ACCEPTED	0.01 s	1, 2, 3	details
#4	ACCEPTED	0.01 s	1, 2, 3	details
#5	ACCEPTED	0.01 s	1, 2, 3	details
#6	ACCEPTED	0.01 s	2, 3	details
#7	ACCEPTED	0.01 s	2, 3	details
#8	ACCEPTED	0.01 s	2, 3	details
#9	ACCEPTED	0.01 s	2, 3	details
#10	ACCEPTED	0.05 s	3	details
#11	ACCEPTED	0.05 s	3	details
#12	ACCEPTED	0.05 s	3	details
#13	ACCEPTED	0.05 s	3	details

Code

#include <bits/stdc++.h>
using namespace std;
struct S {
    int amt;
    long sum, cost;
};
S operator+(S l, S r) {
    return {l.amt + r.amt, l.sum + r.sum, l.cost + r.cost + l.sum * r.amt};
}
const int N = 1<<17;
S sp[2*N];
void sp_clear() {
    for (int i = 0; i < 2*N; ++i) sp[i] = {0, 0, 0};
}
void sp_add(int x, int v) {
    if (!x) return;
    int y = x + N;
    sp[y].amt += v;
    sp[y].sum = long(x) * sp[y].amt;
    sp[y].cost = long(x) * sp[y].amt*(sp[y].amt+1)/2;
    while (y /= 2) {
        sp[y] = sp[2*y] + sp[2*y+1];
    }
}
int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    int n, m;
    cin >> n >> m;
    int q[m];
    for (int i = 0; i < m; ++i) cin >> q[i];
    long pref[m], suff[m];
    int z[n+1] {};
    for (int i = 0; i < m; ++i) {
        sp_add(z[q[i]], -1);
        z[q[i]]++;
        sp_add(z[q[i]], 1);
        pref[i] = sp[1].cost;
    }
    sp_clear();
    for (int i = 0; i < n+1; ++i) z[i] = 0;
    for (int i = m-1; i >= 0; --i) {
        sp_add(z[q[i]], -1);
        z[q[i]]++;
        sp_add(z[q[i]], 1);
        suff[i] = sp[1].cost;
    }
    for (int i = 0; i < m-1; ++i) {
        cout << pref[i] + suff[i+1] << " ";
    }
    cout << endl;
}

Test details

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

input
`1 1 1` view save

correct output
(empty)

user output
(empty)

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

input
`100 1000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...` view save

correct output
`1000 1000 1000 1000 1000 1000 ...` view save

user output
`1000 1000 1000 1000 1000 1000 ...` view save

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

input
`100 1000 1 2 2 2 1 1 1 1 1 1 1 1 1 2 1 ...` view save

correct output
`1488 1488 1487 1486 1487 1488 ...` view save

user output
`1488 1488 1487 1486 1487 1488 ...` view save

Test 4

Group: 1, 2, 3

Verdict: ACCEPTED

input
`100 1000 7 8 2 4 8 3 3 10 9 7 7 6 8 7 2...` view save

correct output
`5107 5107 5103 5099 5098 5102 ...` view save

user output
`5107 5107 5103 5099 5098 5102 ...` view save

Test 5

Group: 1, 2, 3

Verdict: ACCEPTED

input
`100 1000 23 85 3 99 63 79 38 37 67 28 7...` view save

correct output
`41676 41672 41621 41587 41589 ...` view save

user output
`41676 41672 41621 41587 41589 ...` view save

Test 6

Group: 2, 3

Verdict: ACCEPTED

input
`100000 1000 2 1 2 1 1 2 2 2 1 2 2 2 2 2 2 ...` view save

correct output
`1484 1485 1484 1485 1485 1485 ...` view save

user output
`1484 1485 1484 1485 1485 1485 ...` view save

Test 7

Group: 2, 3

Verdict: ACCEPTED

input
`100000 1000 10 8 4 4 3 8 4 7 5 9 7 1 5 3 4...` view save

correct output
`5279 5275 5277 5276 5272 5268 ...` view save

user output
`5279 5275 5277 5276 5272 5268 ...` view save

Test 8

Group: 2, 3

Verdict: ACCEPTED

input
`100000 1000 82 63 63 58 57 46 58 40 23 47 ...` view save

correct output
`41444 41381 41301 41284 41279 ...` view save

user output
`41444 41381 41301 41284 41279 ...` view save

Test 9

Group: 2, 3

Verdict: ACCEPTED

input
`100000 1000 50231 31135 38003 12048 55578 ...` view save

correct output
`498503 497507 496513 495521 49...` view save

user output
`498503 497507 496513 495521 49...` view save

Test 10

Group: 3

Verdict: ACCEPTED

input
`100000 100000 1 2 1 2 1 1 2 1 1 1 1 2 1 1 2 ...` view save

correct output
`149988 149988 149988 149988 14...` view save

user output
`149988 149988 149988 149988 14...` view save

Test 11

Group: 3

Verdict: ACCEPTED

input
`100000 100000 4 2 6 2 2 2 2 5 9 5 3 5 1 9 1 ...` view save

correct output
`547169 547164 547157 547151 54...` view save

user output
`547169 547164 547157 547151 54...` view save

Test 12

Group: 3

Verdict: ACCEPTED

input
`100000 100000 15 9 37 82 72 45 42 96 67 99 6...` view save

correct output
`4959855 4959845 4959773 495975...` view save

user output
`4959855 4959845 4959773 495975...` view save

Test 13

Group: 3

Verdict: ACCEPTED

input
`100000 100000 30941 71402 45742 82863 71830 ...` view save

correct output
`2378217706 2378191367 23781909...` view save

user output
`2378217706 2378191367 23781909...` view save

Putka Open 2025 – 4/5

Muutokset

Code

Test details

Test 1

Test 2

Test 3

Test 4

Test 5

Test 6

Test 7

Test 8

Test 9

Test 10

Test 11

Test 12

Test 13

Tasks

Submissions (ollpu)