CSES - Datatähti 2021 alku - Results
Submission details
Task:Alitaulukot
Sender:mikkopal
Submission time:2020-10-08 18:46:12 +0300
Language:C++ (C++17)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
Test results
testverdicttimegroup
#1ACCEPTED0.01 s1, 2, 3details
#20.01 s1, 2, 3details
#3ACCEPTED0.01 s1, 2, 3details
#40.01 s1, 2, 3details
#5ACCEPTED0.01 s1, 2, 3details
#6ACCEPTED0.01 s2, 3details
#70.01 s2, 3details
#8ACCEPTED0.01 s2, 3details
#90.01 s2, 3details
#10ACCEPTED0.01 s2, 3details
#11ACCEPTED0.02 s3details
#120.02 s3details
#130.02 s3details
#140.03 s3details
#15ACCEPTED0.03 s3details
#160.02 s3details
#170.02 s3details

Code

#include<bits/stdc++.h>
#include <iostream>
#include <vector>
#include <iterator>
using namespace std;

 
typedef long long ll; 
 
const int mx = 1e5;
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
#ifndef ONLINE_JUDGE
	freopen("input.txt","r",stdin);
	freopen("output.txt","w",stdout);
#endif
 
	ll n,m; cin >> n >>m;
	ll a[mx];
	ll ans=n;
	ll last=0;

	for (int i=0; i<n; i++){
		cin >> a[i];
	}

 	deque<ll> minQ;
 	deque<ll> maxQ;

 	ll cnt=0;

 	ll start=0;
 	ll end=1;

 	minQ.push_back(start);
 	maxQ.push_back(start);
 
 	while (end<n){

 		ll num=a[end];

 		while (!minQ.empty() && a[minQ.back()]>=num){
 			minQ.pop_back();
 		}

 		minQ.push_back(end);

 		while (!maxQ.empty() && a[maxQ.back()]<=num){
 			maxQ.pop_back();
 		}

 		maxQ.push_back(end);

 		//minQ alussa välin minimi
 		//maxQ alussa välin maksimi

 		ll minIdx=minQ.front();
 		ll maxIdx=maxQ.front();
 		ll minVal=a[minIdx];
 		ll maxVal=a[maxIdx];


 		if (maxVal-minVal>m){
 			cnt++;
 			start++;
 			if (start>minQ.front()){
 				minQ.pop_front();
 			}
 			if (start>maxQ.front()){
 				maxQ.pop_front();
 			}

 		} else {
 			ans++;
 			if (cnt){
 				if (start==end){
 					ans--;
 				}
 				ans+=((cnt*cnt-3*cnt)/2+1);
 				ans+=((end-last-cnt)*(cnt-1));
 				if (end-start>1){
 					ans++;
 				}
 			}
 			cnt=0;
 			last=start;
 			end++;
 		}

 	}
 	ans+=((end-start)*(end-start)-3*(end-start))/2+1;

 	cout << ans;



}

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:

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

correct output
317

user output
316

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
3936

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:

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

correct output
6340

user output
6299

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:

input
2000 999000000
746120950 772769620 721488968 ...

correct output
1287776

user output
1287103

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:

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

correct output
317066

user output
314092

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
123286

Test 14

Group: 3

Verdict:

input
100000 999990000
460235639 963048588 47270983 3...

correct output
4946886742

user output
4946875803

Test 15

Group: 3

Verdict: ACCEPTED

input
100000 1000000000
885457070 18257718 927615960 3...

correct output
5000050000

user output
5000050000

Test 16

Group: 3

Verdict:

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

correct output
3750075000

user output
1250224998

Test 17

Group: 3

Verdict:

input
100000 50000
100000 99999 99998 99997 99996...

correct output
3750075000

user output
1250224998