CSES - Putka Open 2025 – 4/5

Submission details
Task:	Muutokset
Sender:	Laakeri
Submission time:	2025-11-07 21:19:37 +0200
Language:	C++ (C++11)
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.00 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.04 s	2, 3	details
#7	ACCEPTED	0.03 s	2, 3	details
#8	ACCEPTED	0.03 s	2, 3	details
#9	ACCEPTED	0.04 s	2, 3	details
#10	ACCEPTED	0.15 s	3	details
#11	ACCEPTED	0.14 s	3	details
#12	ACCEPTED	0.12 s	3	details
#13	ACCEPTED	0.11 s	3	details

Code

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N=1<<17;


void add(int i, ll x, ll* st){
	for (i+=N;i;i/=2){
		st[i]+=x;
	}
}

ll sum(int a, int b, ll* st){
	a+=N;
	b+=N;
	ll s=0;
	while (a<=b){
		if (a%2){
			s+=st[a++];
		}
		if (!(b%2)){
			s+=st[b--];
		}
		a/=2;
		b/=2;
	}
	return s;
}

void addcnt(int x, ll* st1, ll* st2, ll& cost){
	cost+=sum(1, x, st1);
	add(x, x, st1);
	cost+=(ll)x * (1ll + sum(x+1, N-1, st2));
	add(x, 1, st2);
}

void delcnt(int x, ll* st1, ll* st2, ll& cost){
	add(x, -x, st1);
	cost -= sum(1, x, st1);
	add(x, -1, st2);
	cost -= (ll)x * (1ll + sum(x+1, N-1, st2));
}


ll st1[2*N];
ll st2[2*N];
ll st3[2*N];
ll st4[2*N];

ll cost=0;

int a[N];

int c1[N];
int c2[N];

int main(){
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	int n,m;
	cin>>n>>m;
	for (int i=1;i<=m;i++){
		cin>>a[i];
		c1[a[i]]++;
	}
	ll cost1=0;
	ll cost2=0;
	for (int i=1;i<=n;i++){
		addcnt(c1[i], st1, st2, cost1);
		addcnt(c2[i], st3, st4, cost2);
	}
	for (int i=1;i<m;i++){
		delcnt(c1[a[i]], st1, st2, cost1);
		c1[a[i]]--;
		addcnt(c1[a[i]], st1, st2, cost1);

		delcnt(c2[a[i]], st3, st4, cost2);
		c2[a[i]]++;
		addcnt(c2[a[i]], st3, st4, cost2);

		cout<<cost1+cost2<<" ";
	}
	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 (Laakeri)