CSES - Datatähti 2019 alku - Results
Submission details
Task:Taulukko
Sender:1
Submission time:2018-10-13 13:52:06 +0300
Language:C++
Status:READY
Result:30
Feedback
groupverdictscore
#1ACCEPTED12
#20
#3ACCEPTED18
#40
Test results
testverdicttimegroup
#1ACCEPTED0.02 s1details
#2ACCEPTED0.02 s1details
#3ACCEPTED0.01 s1details
#4ACCEPTED0.02 s1details
#5ACCEPTED0.02 s1details
#6ACCEPTED0.02 s1details
#7ACCEPTED0.01 s1details
#8ACCEPTED0.01 s1details
#9ACCEPTED0.01 s1details
#10ACCEPTED0.04 s1details
#11ACCEPTED0.01 s1details
#12ACCEPTED0.02 s1details
#13ACCEPTED0.01 s1details
#14ACCEPTED0.01 s1details
#15ACCEPTED0.03 s2details
#16ACCEPTED0.03 s2details
#17ACCEPTED0.02 s2details
#18ACCEPTED0.01 s2details
#19ACCEPTED0.02 s2details
#20ACCEPTED0.01 s2details
#21ACCEPTED0.01 s2details
#22--2details
#23--2details
#24--2details
#25ACCEPTED0.01 s2details
#26ACCEPTED0.14 s2details
#27ACCEPTED0.01 s2details
#28ACCEPTED0.01 s2details
#29ACCEPTED0.10 s3details
#30ACCEPTED0.23 s3details
#31ACCEPTED0.10 s3details
#32ACCEPTED0.10 s3details
#33ACCEPTED0.11 s3details
#34ACCEPTED0.10 s3details
#35ACCEPTED0.27 s3details
#36ACCEPTED0.10 s3details
#37ACCEPTED0.09 s4details
#38ACCEPTED0.23 s4details
#39ACCEPTED0.09 s4details
#40ACCEPTED0.10 s4details
#41ACCEPTED0.11 s4details
#42ACCEPTED0.27 s4details
#43--4details
#44--4details
#45--4details
#46--4details
#47--4details
#480.23 s4details
#49ACCEPTED0.10 s4details
#500.23 s4details
#51ACCEPTED0.10 s4details
#52ACCEPTED0.23 s4details

Compiler report

input/code.cpp: In function 'ul countDynamic(std::vector<long unsigned int>*, int, int, ul)':
input/code.cpp:37:22: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (ul j = 0; j < size; j++) {
                    ~~^~~~~~
input/code.cpp:41:30: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     if (countNonZero(&items) <= k) {
         ~~~~~~~~~~~~~~~~~~~~~^~~~
input/code.cpp:47:34: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         if (countNonZero(&items) <= k) {
             ~~~~~~~~~~~~~~~~~~~~~^~~~
input/code.cpp: In function 'ul countK2(std::vector<long unsigned int>*, ul)':
input/code.cpp:78:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
                 if (T != -1) {
                     ~~^~~~~
input/code.cpp:89:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         if...

Code

#include <iostream>
#include <vector>
#include <set>

using namespace std;

typedef vector <unsigned long> lVect;
typedef unsigned long ul;

ul countNonZero(vector <lVect>* items) {
    ul n = 0;
    for (ul i = 0; i < (*items).size(); i++) {
        if ((*items)[i][1] != 0) {
            n += 1;
        }
    }
    return n;
}

ul maxValue(vector <ul>* seq) {
    ul maxFound = 0;
    for (ul i = 0; i < (*seq).size(); i++) {
        if ((*seq)[i] > maxFound) {
            maxFound = (*seq)[i];
        }
    }
    return maxFound;
}

ul countDynamic(vector <ul>* seq, int size, int k, ul maxVal) {
    ul validSublists = 0;
    vector <lVect> items(maxVal + 1);
    for (ul i = 0; i < items.size(); i++) {
        vector <ul> c = {i, 0};
        items[i] = c;
    }
    for (ul j = 0; j < size; j++) {
        ul index = (*seq)[j];
        items[index][1] += 1;
    }
    if (countNonZero(&items) <= k) {
        validSublists += 1;
    }
    for (ul q = size; q < (*seq).size(); q++) {
        items[(*seq)[q]][1] += 1;
        items[(*seq)[q - size]][1] -= 1;
        if (countNonZero(&items) <= k) {
            validSublists += 1;
        }
    }
    return validSublists;
}

ul triangleNumber(ul n) {
    return (n * n + n) / 2;
}

ul countK2(vector <ul>* r, ul size) {
    ul n = 0;
    set <ul> S;
    ul T = -1;
    ul crdS = 0;
    ul prev_s_end = 0;
    ul s_start = 0;
    ul t_start = 0;
    for (ul i = 0; i < size; i++) {
        ul ith = (*r)[i];
        if (crdS < 2) {
            if (S.count(ith) == 0) {
                crdS += 1;
            }
            S.insert(ith);
        } else if (crdS == 2) {
            if (S.count(ith) == 0) {
                n += triangleNumber(i - s_start) - triangleNumber(prev_s_end - s_start);
                S.erase(S.begin(), S.end());
                crdS = 0;
                if (T != -1) {
                    S.insert(T);
                }
                prev_s_end = i;
                s_start = t_start;
                T = -1;
                t_start = i;
                S.insert(ith);
                crdS = S.size();
            }
        }
        if (T == -1) {
            T = ith;
        } else {
            if (ith != T) {
                t_start = i;
                T = ith;
            }
        }
        if (i == size - 1) {
            n += triangleNumber(i + 1 - s_start) - triangleNumber(prev_s_end - s_start);
        }
    }
    return n;
}

ul countK1(vector <ul>* r, ul size) {
    ul n = 0;
    ul val = (*r)[0];
    ul start = 0;
    for (ul i = 0; i <= size; i++) {
        ul ith;
        if (i == size) {
            ith = -1;
        } else {
            ith = (*r)[i];
        }
        if (ith != val) {
            n += triangleNumber(i - start);
            start = i;
            val = ith;
        }
    }
    return n;
}

int main(int argc, const char * argv[]) {
    int arrayLength;
    int k;
    cin >> arrayLength;
    cin >> k;
    vector <ul> list(arrayLength);
    for (int i = 0; i < arrayLength; i++) {
        int next;
        cin >> next;
        list[i] = next;
    }
    if (k > 2) {
        ul sum = 0;
        for (int n = 1; n < k + 1; n++) {
            sum += list.size() - n + 1;
        }
        ul maxVal = maxValue(&list);
        for (int m = k + 1; m < list.size() + 1; m++) {
            sum += countDynamic(&list, m, k, maxVal);
        }
        cout << sum;
    } else if (k == 2) {
        ul sum = countK2(&list, list.size());
        cout << sum;
    } else {
        ul sum = countK1(&list, list.size());
        cout << sum;
    }
    return 0;
}

Test details

Test 1

Group: 1

Verdict: ACCEPTED

input
100 1
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

Verdict: ACCEPTED

input
100 1
1 1 1 1 2 2 1 1 2 2 2 2 2 1 1 ...

correct output
190

user output
190

Test 3

Group: 1

Verdict: ACCEPTED

input
100 1
5 9 9 6 9 8 1 4 7 7 8 9 5 5 6 ...

correct output
110

user output
110

Test 4

Group: 1

Verdict: ACCEPTED

input
100 2
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
5050

user output
5050

Test 5

Group: 1

Verdict: ACCEPTED

input
100 2
2 1 2 1 2 2 1 1 2 1 1 2 1 2 1 ...

correct output
5050

user output
5050

Test 6

Group: 1

Verdict: ACCEPTED

input
100 2
2 3 1 1 2 2 3 1 2 1 1 1 3 3 1 ...

correct output
379

user output
379

Test 7

Group: 1

Verdict: ACCEPTED

input
100 2
4 6 10 8 6 8 10 8 4 7 8 9 6 2 ...

correct output
245

user output
245

Test 8

Group: 1

Verdict: ACCEPTED

input
100 5
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
5050

user output
5050

Test 9

Group: 1

Verdict: ACCEPTED

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

correct output
5050

user output
5050

Test 10

Group: 1

Verdict: ACCEPTED

input
100 5
5 3 4 1 1 1 1 4 5 5 4 6 6 3 3 ...

correct output
1488

user output
1488

Test 11

Group: 1

Verdict: ACCEPTED

input
100 5
10 1 1 9 1 6 9 4 3 10 9 2 4 2 ...

correct output
743

user output
743

Test 12

Group: 1

Verdict: ACCEPTED

input
100 10
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
5050

user output
5050

Test 13

Group: 1

Verdict: ACCEPTED

input
100 10
9 4 9 1 9 1 5 2 10 10 9 3 9 6 ...

correct output
5050

user output
5050

Test 14

Group: 1

Verdict: ACCEPTED

input
100 10
80 59 58 87 28 83 83 93 96 24 ...

correct output
994

user output
994

Test 15

Group: 2

Verdict: ACCEPTED

input
5000 1
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
12502500

user output
12502500

Test 16

Group: 2

Verdict: ACCEPTED

input
5000 1
2 2 2 1 2 1 1 1 2 2 1 2 1 1 1 ...

correct output
10148

user output
10148

Test 17

Group: 2

Verdict: ACCEPTED

input
5000 1
434568 634119 102509 107238 72...

correct output
5000

user output
5000

Test 18

Group: 2

Verdict: ACCEPTED

input
5000 2
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
12502500

user output
12502500

Test 19

Group: 2

Verdict: ACCEPTED

input
5000 2
2 1 1 1 2 2 2 2 1 1 1 1 2 1 2 ...

correct output
12502500

user output
12502500

Test 20

Group: 2

Verdict: ACCEPTED

input
5000 2
1 3 2 2 3 3 1 2 1 2 1 2 3 1 2 ...

correct output
22451

user output
22451

Test 21

Group: 2

Verdict: ACCEPTED

input
5000 2
132968 414421 468358 432744 43...

correct output
9999

user output
9999

Test 22

Group: 2

Verdict:

input
5000 100
93 76 87 71 93 74 69 35 92 96 ...

correct output
12502500

user output
(empty)

Test 23

Group: 2

Verdict:

input
5000 100
2 57 51 4 35 76 5 40 51 55 20 ...

correct output
2669737

user output
(empty)

Test 24

Group: 2

Verdict:

input
5000 100
657 823 1773 1801 2785 4107 25...

correct output
498978

user output
(empty)

Test 25

Group: 2

Verdict: ACCEPTED

input
5000 4999
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
12502500

user output
12502500

Test 26

Group: 2

Verdict: ACCEPTED

input
5000 4999
865706 910619 695192 183574 92...

correct output
12502500

user output
12502500

Test 27

Group: 2

Verdict: ACCEPTED

input
5000 5000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
12502500

user output
12502500

Test 28

Group: 2

Verdict: ACCEPTED

input
5000 5000
774752 159472 183796 654476 69...

correct output
12502500

user output
12502500

Test 29

Group: 3

Verdict: ACCEPTED

input
500000 1
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
125000250000

user output
125000250000

Test 30

Group: 3

Verdict: ACCEPTED

input
500000 1
28107328 613212742 298033960 7...

correct output
500000

user output
500000

Test 31

Group: 3

Verdict: ACCEPTED

input
500000 2
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
125000250000

user output
125000250000

Test 32

Group: 3

Verdict: ACCEPTED

input
500000 2
1 1 2 1 1 2 2 1 2 1 2 1 1 2 1 ...

correct output
125000250000

user output
125000250000

Test 33

Group: 3

Verdict: ACCEPTED

input
500000 2
3 2 2 3 2 2 2 1 1 2 2 3 2 3 2 ...

correct output
2245355

user output
2245355

Test 34

Group: 3

Verdict: ACCEPTED

input
500000 2
1 1 1 2 2 2 1 1 2 1 1 2 1 2 1 ...

correct output
1570486882

user output
1570486882

Test 35

Group: 3

Verdict: ACCEPTED

input
500000 2
318961563 84011941 882177798 1...

correct output
999999

user output
999999

Test 36

Group: 3

Verdict: ACCEPTED

input
500000 2
1 1 2 1 2 1 2 1 2 2 2 2 1 2 1 ...

correct output
483793169

user output
483793169

Test 37

Group: 4

Verdict: ACCEPTED

input
500000 1
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
125000250000

user output
125000250000

Test 38

Group: 4

Verdict: ACCEPTED

input
500000 1
709486749 350496125 796065873 ...

correct output
500000

user output
500000

Test 39

Group: 4

Verdict: ACCEPTED

input
500000 2
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
125000250000

user output
125000250000

Test 40

Group: 4

Verdict: ACCEPTED

input
500000 2
1 2 2 2 2 2 1 1 2 2 2 2 2 2 1 ...

correct output
125000250000

user output
125000250000

Test 41

Group: 4

Verdict: ACCEPTED

input
500000 2
3 3 3 3 2 1 3 2 1 3 1 2 1 1 1 ...

correct output
2255047

user output
2255047

Test 42

Group: 4

Verdict: ACCEPTED

input
500000 2
414942284 362802746 108881396 ...

correct output
999999

user output
999999

Test 43

Group: 4

Verdict:

input
500000 100
59 14 100 74 49 43 91 84 31 16...

correct output
260458849

user output
(empty)

Test 44

Group: 4

Verdict:

input
500000 1000
505 511 86 321 780 495 106 330...

correct output
3694803834

user output
(empty)

Test 45

Group: 4

Verdict:

input
500000 10000
4956 8463 8582 980 9278 1747 2...

correct output
41655235436

user output
(empty)

Test 46

Group: 4

Verdict:

input
500000 100000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
125000250000

user output
(empty)

Test 47

Group: 4

Verdict:

input
500000 100000
40198 78364 3724 51802 378 130...

correct output
125000250000

user output
(empty)

Test 48

Group: 4

Verdict:

input
500000 100000
320513086 811766509 339605137 ...

correct output
45001856854

user output
(empty)

Error:
terminate called after throwing an instance of 'std::bad_alloc'
  what():  std::bad_alloc

Test 49

Group: 4

Verdict: ACCEPTED

input
500000 499999
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
125000250000

user output
125000250000

Test 50

Group: 4

Verdict:

input
500000 499999
995132060 587162982 59723733 8...

correct output
125000250000

user output
(empty)

Error:
terminate called after throwing an instance of 'std::bad_alloc'
  what():  std::bad_alloc

Test 51

Group: 4

Verdict: ACCEPTED

input
500000 500000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
125000250000

user output
125000250000

Test 52

Group: 4

Verdict: ACCEPTED

input
500000 500000
72083718 753463162 730560260 6...

correct output
125000250000

user output
125000250000