CSES - Datatähti 2021 alku - Results
Submission details
Task:Alitaulukot
Sender:jusola
Submission time:2020-10-09 18:38:13 +0300
Language:C++17
Status:READY
Result:37
Feedback
groupverdictscore
#1ACCEPTED11
#2ACCEPTED26
#30
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.02 s3details
#13ACCEPTED0.03 s3details
#14ACCEPTED0.05 s3details
#15ACCEPTED0.05 s3details
#16--3details
#17--3details

Code

#include <iostream>
#include <vector>
#include <algorithm>
#include <typeinfo>

#define ll long long

using namespace std;

ll int s(ll int n){
    return (n*(n+1))/2;
}

int main(){
    int n;
    ll int k;
    int c = 0;
    ll int r = 0;
    cin >> n >> k;
    vector<ll int> a(n);
    for(int i = 0; i < n; i++){
        cin >> a[i];
    }

    ll int min = a[0];
    ll int max = a[0];
    int distmax = 0;
    int distmin = 0;
    int distbreak = 0;
    for (auto it = a.begin(); it != a.end(); ++it){
        if(*it > max){
            //cout << "new max "<< *it << " dist from last: " << distmax <<"\n";
            distmax = 0;
            max = *it;
            if(max - min > k){
                auto oit = it;
                r += s(c);
                //cout << " broken by new max " << max << " going back to break: back " << distbreak << ": " << *(oit - distbreak) << "\n"; 
                //cout << " adding c " << c << "\n";
                c = 0; // start here as following loop is executed atleast once
                min = *it;
                while(max - *(it-1) <= k && (it-1) != (oit-distbreak)){
                    c++;
                    it--;
                    if(*it < min){
                        min = *it;
                    }
                    //cout << " while: " << *it << "\n";
                }
                //cout << " after while ended up in " << *it << " with c of " << c << " and max of " << max << " and min of " << min << " and removing of " << c << "\n";
                //distmin = 0;
                //distmax = 0;
                it += c;
                r -= s(c); // remove already added c
                //cout << " ended in " << *it << "\n";
                distbreak = c;
            }
        }
        if(*it < min){
            //cout << "new min "<< *it << " dist from last: " << distmin <<"\n";
            distmin = 0;
            min = *it;
            if(max - min > k){
                auto oit = it;
                r += s(c);
                //cout << " broken by new min " << min << " going back to break: back " << distbreak << ": " << *(oit - distbreak) << "\n"; 
                //cout << " adding c " << c << "\n";
                c = 0;
                max = *it;
                while(*(it-1) - min <= k && (it-1) != (oit-distbreak)){
                    c++;
                    it--;
                    if(*it > max){
                        max = *it;
                    }
                    //cout << " while: " << *it << "\n";
                }
                if(it == (oit - distbreak)){
                    //cout << " reached oit\n";
                }
                //cout << " after while ended up in " << *it << " with c of " << c << " and max of " << max << " and min of " << min << " and removing of " << c << "\n";
                it += c;
                r -= s(c); // remove already added c
                //cout << " ended in " << *it << "\n";
                distbreak = c;
            }
        }
        distbreak++;
        distmax++;
        distmin++;
        c++;
    }
    //cout << "remainder: " << c << "\n";
    r += s(c);

    cout << r;

    return 0;
}

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:

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

correct output
3750075000

user output
(empty)

Test 17

Group: 3

Verdict:

input
100000 50000
100000 99999 99998 99997 99996...

correct output
3750075000

user output
(empty)