CSES - Datatähti 2021 alku - Results
Submission details
Task:Alitaulukot
Sender:Nanohenry
Submission time:2020-09-28 10:14:59 +0300
Language:C++11
Status:READY
Result:100
Feedback
groupverdictscore
#1ACCEPTED11
#2ACCEPTED26
#3ACCEPTED63
Test results
testverdicttimegroup
#1ACCEPTED0.01 s1, 2, 3details
#2ACCEPTED0.01 s1, 2, 3details
#3ACCEPTED0.01 s1, 2, 3details
#4ACCEPTED0.01 s1, 2, 3details
#5ACCEPTED0.01 s1, 2, 3details
#6ACCEPTED0.01 s2, 3details
#7ACCEPTED0.01 s2, 3details
#8ACCEPTED0.01 s2, 3details
#9ACCEPTED0.01 s2, 3details
#10ACCEPTED0.01 s2, 3details
#11ACCEPTED0.02 s3details
#12ACCEPTED0.03 s3details
#13ACCEPTED0.03 s3details
#14ACCEPTED0.05 s3details
#15ACCEPTED0.05 s3details
#16ACCEPTED0.04 s3details
#17ACCEPTED0.04 s3details

Code

#include <iostream>
#include <deque>

using namespace std;

#define min(a, b) ((a < b)? a : b)
#define max(a, b) ((a > b)? a : b)

#define SM sm.front()
#define LG lg.front()

#define BCM (b < n - 1 && max(LG, x[b + 1]) - min(SM, x[b + 1]) <= k)

typedef long long ll;

ll n, k, a, b, x[100001], i, j, s, c;

deque<ll> sm, lg;

inline void rm(ll v) {
	if (v == SM) {
		sm.pop_front();
	}
	if (v == LG) {
		lg.pop_front();
	}
}

inline void add(ll v) {
	while (sm.size() > 0 && v < sm.back()) {
		sm.pop_back();
	}
	sm.push_back(v);
	while (lg.size() > 0 && v > lg.back()) {
		lg.pop_back();
	}
	lg.push_back(v);
}

int main() {
	cin >> n >> k;
	for (; i < n; i++) {
		cin >> x[i];
	}
	add(x[0]);
	while (a < n) {
		//cout << "===" << endl;
		//cout << "a " << a << " b " << b << endl;
		//cout << "max " << LG << " min " << SM << endl;
		//cout << "b move? " << BCM << endl;
		if (BCM) {
			b++;
			//cout << "new a " << a << " b " << b << endl;
			add(x[b]);
			//cout << "new max " << LG << " min " << SM << endl;
		} else if (a < b) {
			c += b - a + 1;
			//cout << "new count " << c << endl;
			rm(x[a]);
			a++;
			//cout << "new a " << a << " b " << b << endl;
			//cout << "new max " << LG << " min " << SM << endl;
		} else {
			b++;
			add(x[b]);

			c++;
			rm(x[a]);
			a++;
			//cout << "new a " << a << " b " << b << endl;
			//cout << "new max " << LG << " min " << SM << endl;
		}
	}
	cout << c << endl;
}

Test details

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

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

correct output
5050

user output
5050

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

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

correct output
317

user output
317

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: ACCEPTED

input
100 990000000
111122929 961821360 578238211 ...

correct output
4006

user output
4006

Test 5

Group: 1, 2, 3

Verdict: ACCEPTED

input
100 1000000000
553190572 453407680 667300705 ...

correct output
5050

user output
5050

Test 6

Group: 2, 3

Verdict: ACCEPTED

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

correct output
2001000

user output
2001000

Test 7

Group: 2, 3

Verdict: ACCEPTED

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

correct output
6340

user output
6340

Test 8

Group: 2, 3

Verdict: ACCEPTED

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

correct output
2413

user output
2413

Test 9

Group: 2, 3

Verdict: ACCEPTED

input
2000 999000000
746120950 772769620 721488968 ...

correct output
1287776

user output
1287776

Test 10

Group: 2, 3

Verdict: ACCEPTED

input
2000 1000000000
621947980 510355354 756705418 ...

correct output
2001000

user output
2001000

Test 11

Group: 3

Verdict: ACCEPTED

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

correct output
5000050000

user output
5000050000

Test 12

Group: 3

Verdict: ACCEPTED

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

correct output
317066

user output
317066

Test 13

Group: 3

Verdict: ACCEPTED

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

correct output
123292

user output
123292

Test 14

Group: 3

Verdict: ACCEPTED

input
100000 999990000
460235639 963048588 47270983 3...

correct output
4946886742

user output
4946886742

Test 15

Group: 3

Verdict: ACCEPTED

input
100000 1000000000
885457070 18257718 927615960 3...

correct output
5000050000

user output
5000050000

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