CSES - Datatähti 2021 alku - Results
Submission details
Task:Alitaulukot
Sender:lady-stardust
Submission time:2020-09-30 17:37:38 +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
#30.01 s1, 2, 3details
#4ACCEPTED0.01 s1, 2, 3details
#5ACCEPTED0.01 s1, 2, 3details
#6ACCEPTED0.01 s2, 3details
#70.01 s2, 3details
#80.01 s2, 3details
#90.01 s2, 3details
#10ACCEPTED0.01 s2, 3details
#11ACCEPTED0.02 s3details
#120.02 s3details
#130.02 s3details
#14ACCEPTED0.03 s3details
#15ACCEPTED0.03 s3details
#160.02 s3details
#170.02 s3details

Code

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

int main() {
  ios_base::sync_with_stdio(0);
  cin.tie(0);
  
  int n, k; cin >> n >> k;

  vector<int> v(n);
  for (int i = 0; i < n; i++) cin >> v[i];

  vector<int> next_min(n);
  vector<int> next_max(n);
  queue<pair<int,int>> q_min;
  queue<pair<int,int>> q_max;

  for (int i = n-1; i>=0; --i) {
    while (q_min.size() != 0 && q_min.front().first >= v[i]) q_min.pop();
    while (q_max.size() != 0 && q_max.front().first <= v[i]) q_max.pop();
    if (q_min.size() != 0) next_min[i] = q_min.front().second; else next_min[i] = n;
    if (q_max.size() != 0) next_max[i] = q_max.front().second; else next_max[i] = n;
    q_min.push({v[i], i});
    q_max.push({v[i], i});
  }

  // for (int i = 0; i < n; i++) {
  //   cout << "next_min: " << next_min[i] << "\n";
  //   cout << "next_max: " << next_max[i] << "\n";
  // }
  
  ll t = 0;

  for (int s = 0; s < n; s++) {
    int i = s;
    int mini = i;
    int maxi = i;

    while (i < n) {
      int mult = v[maxi]-v[mini] <= k;

      if (next_min[mini] < next_max[maxi]) {
        t += mult * (next_min[mini] - i);
        i = next_min[mini];
        mini = next_min[mini];
      } else {
        t += mult * (next_max[maxi]- i);
        i = next_max[maxi];
        maxi = next_max[maxi];
      }
    }
  }

  cout << t;
}

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
671

Test 3

Group: 1, 2, 3

Verdict:

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

correct output
119

user output
1568

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:

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

correct output
6340

user output
11101

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
97638

Test 9

Group: 2, 3

Verdict:

input
2000 999000000
746120950 772769620 721488968 ...

correct output
1287776

user output
1303212

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
549043

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
5862709

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:

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

correct output
3750075000

user output
5000000001

Test 17

Group: 3

Verdict:

input
100000 50000
100000 99999 99998 99997 99996...

correct output
3750075000

user output
5000000001