CSES - Datatähti 2021 alku - Results
Submission details
Task:Alitaulukot
Sender:Ender
Submission time:2020-09-29 13:38:03 +0300
Language:C++ (C++11)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
Test results
testverdicttimegroup
#10.01 s1, 2, 3details
#20.01 s1, 2, 3details
#3ACCEPTED0.01 s1, 2, 3details
#40.01 s1, 2, 3details
#50.01 s1, 2, 3details
#60.01 s2, 3details
#70.01 s2, 3details
#80.01 s2, 3details
#90.01 s2, 3details
#100.01 s2, 3details
#110.04 s3details
#120.03 s3details
#130.02 s3details
#140.07 s3details
#150.07 s3details
#16ACCEPTED0.04 s3details
#17ACCEPTED0.04 s3details

Code

#include <bits/stdc++.h>
#define ll long long int

using namespace std;


int main()
{
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);

	int n,k; cin >> n >> k;

	set<pair<int,int>> q;
	vector<int> e(n+1);

	for(int i = 0; i<n; i++)
	{
		int a; cin >> a;	
		q.insert(make_pair(a,i));

		while(a-q.begin()->first>k)
		{
			e[q.begin()->second]=i;
			q.erase(q.begin());
		}
		while(q.rbegin()->first-a>k)
		{
			e[prev(q.end())->second]=i;
			q.erase(prev(q.end()));
		}

	}


	if(!e[n-1]) e[n]=n;

	ll ans = 0;
	int mem = 0;
	int m = 0;
	for(int i = 0; i<=n; i++)
	{
		if(!e[i]&&!m) m = i;
		if(!e[i]||e[i]<=mem)	continue;

		ll x = e[i]-i;
		ll y = mem-i;
		if(i>mem)
			y = 0;
		if(m)
		{
			x = e[i]-m;
			y = mem-m;
			if(m>mem)
				y = 0;
			m = 0;
		}
		ans += (x*(1+x))/2;
		ans -= (y*(1+y))/2;

		if(e[i]==n) break;
		mem = e[i];
	}

	cout << ans << "\n";

	return 0;
}

Test details

Test 1

Group: 1, 2, 3

Verdict:

input
100 0
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
5050

user output
4950

Test 2

Group: 1, 2, 3

Verdict:

input
100 2
5 5 2 4 3 5 3 4 3 2 3 4 5 4 4 ...

correct output
317

user output
877

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

input
100 10
71 60 61 96 25 10 10 9 84 85 1...

correct output
119

user output
119

Test 4

Group: 1, 2, 3

Verdict:

input
100 990000000
111122929 961821360 578238211 ...

correct output
4006

user output
3935

Test 5

Group: 1, 2, 3

Verdict:

input
100 1000000000
553190572 453407680 667300705 ...

correct output
5050

user output
4950

Test 6

Group: 2, 3

Verdict:

input
2000 0
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
2001000

user output
1999000

Test 7

Group: 2, 3

Verdict:

input
2000 2
4 4 1 4 2 3 1 2 1 3 5 2 2 4 4 ...

correct output
6340

user output
13858

Test 8

Group: 2, 3

Verdict:

input
2000 10
65 88 33 88 41 10 17 38 22 3 8...

correct output
2413

user output
2467

Test 9

Group: 2, 3

Verdict:

input
2000 999000000
746120950 772769620 721488968 ...

correct output
1287776

user output
1287181

Test 10

Group: 2, 3

Verdict:

input
2000 1000000000
621947980 510355354 756705418 ...

correct output
2001000

user output
1999000

Test 11

Group: 3

Verdict:

input
100000 0
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
5000050000

user output
4999950000

Test 12

Group: 3

Verdict:

input
100000 2
3 3 1 3 3 1 1 5 1 2 5 4 1 3 1 ...

correct output
317066

user output
651068

Test 13

Group: 3

Verdict:

input
100000 10
10 3 6 3 43 60 5 48 15 27 86 4...

correct output
123292

user output
126278

Test 14

Group: 3

Verdict:

input
100000 999990000
460235639 963048588 47270983 3...

correct output
4946886742

user output
4946787343

Test 15

Group: 3

Verdict:

input
100000 1000000000
885457070 18257718 927615960 3...

correct output
5000050000

user output
4999950000

Test 16

Group: 3

Verdict: ACCEPTED

input
100000 50000
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

correct output
3750075000

user output
3750075000

Test 17

Group: 3

Verdict: ACCEPTED

input
100000 50000
100000 99999 99998 99997 99996...

correct output
3750075000

user output
3750075000