CSES - Datatähti 2019 alku - Results
Submission details
Task:Taulukko
Sender:jaakkonen
Submission time:2018-10-08 17:30:47 +0300
Language:C++
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
#40
Test results
testverdicttimegroup
#10.02 s1details
#20.01 s1details
#30.01 s1details
#40.01 s1details
#50.01 s1details
#60.01 s1details
#70.02 s1details
#80.01 s1details
#90.03 s1details
#100.01 s1details
#110.03 s1details
#120.02 s1details
#130.03 s1details
#140.02 s1details
#150.02 s2details
#160.03 s2details
#170.03 s2details
#180.02 s2details
#190.02 s2details
#200.01 s2details
#210.01 s2details
#220.01 s2details
#230.01 s2details
#240.02 s2details
#250.01 s2details
#260.01 s2details
#270.01 s2details
#280.03 s2details
#290.02 s3details
#300.01 s3details
#310.01 s3details
#320.01 s3details
#330.02 s3details
#340.01 s3details
#350.01 s3details
#360.01 s3details
#370.01 s4details
#380.03 s4details
#390.02 s4details
#400.03 s4details
#410.01 s4details
#420.02 s4details
#430.01 s4details
#440.01 s4details
#450.01 s4details
#460.01 s4details
#470.02 s4details
#480.02 s4details
#490.02 s4details
#500.02 s4details
#510.02 s4details
#520.02 s4details

Compiler report

input/code.cpp: In function 'int main()':
input/code.cpp:20:19: warning: format '%ld' expects argument of type 'long int*', but argument 2 has type 'int*' [-Wformat=]
     scanf("%ld",&n);
                 ~~^
input/code.cpp:21:19: warning: format '%ld' expects argument of type 'long int*', but argument 2 has type 'int*' [-Wformat=]
     scanf("%ld",&k);
                 ~~^
input/code.cpp:25:26: warning: format '%ld' expects argument of type 'long int*', but argument 2 has type 'int*' [-Wformat=]
         scanf("%ld",&temp);
                     ~~~~~^
input/code.cpp:39:49: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     if(set<int> (arr.begin(), arr.end()).size() <= k){
        ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~
input/code.cpp:44:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     while (index != arr.size()){
            ~~~~~~^~~~~~~~~~~~~
input/code.cpp:53:35: warning: comparison between...

Code

#include<iostream>
#include<sstream>
#include<deque>
#include<string>
#include<algorithm>
#include<map>
#include<set>
#include<list>
using namespace std;
int main(){
    //ios_base::sync_with_stdio(false);

    deque<int> arr;
    int k;
    int n;
    //k = 3;
    //n = 8;
    //arr = {1, 2, 1, 1, 1, 3, 1, 4};
    
    scanf("%ld",&n);
    scanf("%ld",&k);
    
    int temp;
    for(int i=0;i<n;i++){
        scanf("%ld",&temp);
        arr.push_back(temp);
    }
    
    arr.resize(n);
    list<int> active_nums;
    list<int>::iterator it;
    int subarrays = 0;
    int index = 0;
    int curr_num;
    int num_to_remove;
    int removed_index;
    map<int, int> last_index;

    if(set<int> (arr.begin(), arr.end()).size() <= k){
        cout << arr.size()*(arr.size()+1)/2 << endl;
        return 0;
    }

    while (index != arr.size()){
        curr_num = arr[index];
        it = find(active_nums.begin(), active_nums.end(), curr_num);
        last_index[curr_num] = index;
        if(it != active_nums.end()){
            active_nums.erase(it);
            active_nums.push_back(curr_num);
        }else{
            active_nums.push_back(curr_num);
            if(active_nums.size() > k){
                num_to_remove = active_nums.front();
                //for(int x : active_nums) {num_to_remove = x; break;}
                //cout << "For-loop: " << num_to_remove << endl;
                active_nums.pop_front();
                removed_index = last_index[num_to_remove];
                //removed_index = arr.size() - (find(arr.rbegin() + arr.size() - index, arr.rend(), num_to_remove) - arr.rbegin()) - 1;
                //cout << "Current: " << removed_index << endl << "Magic: " << last_index[num_to_remove] << endl;
                subarrays += (removed_index*(2*index-removed_index-1)+2*index)/2;
                arr.erase(arr.begin(), arr.begin()+removed_index+1);
                last_index.clear();
                index = 0;
            }
        }
        index++;
    }
    subarrays += (arr.size()+1)*arr.size()/2;
    cout << subarrays << endl;
}

Test details

Test 1

Group: 1

Verdict:

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

correct output
5050

user output
0

Test 2

Group: 1

Verdict:

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

correct output
190

user output
0

Test 3

Group: 1

Verdict:

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

correct output
110

user output
0

Test 4

Group: 1

Verdict:

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

correct output
5050

user output
0

Test 5

Group: 1

Verdict:

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

correct output
5050

user output
0

Test 6

Group: 1

Verdict:

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

correct output
379

user output
0

Test 7

Group: 1

Verdict:

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

correct output
245

user output
0

Test 8

Group: 1

Verdict:

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

correct output
5050

user output
0

Test 9

Group: 1

Verdict:

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

correct output
5050

user output
0

Test 10

Group: 1

Verdict:

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

correct output
1488

user output
0

Test 11

Group: 1

Verdict:

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

correct output
743

user output
0

Test 12

Group: 1

Verdict:

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

correct output
5050

user output
0

Test 13

Group: 1

Verdict:

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

correct output
5050

user output
0

Test 14

Group: 1

Verdict:

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

correct output
994

user output
0

Test 15

Group: 2

Verdict:

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

correct output
12502500

user output
0

Test 16

Group: 2

Verdict:

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

correct output
10148

user output
0

Test 17

Group: 2

Verdict:

input
5000 1
434568 634119 102509 107238 72...

correct output
5000

user output
0

Test 18

Group: 2

Verdict:

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

correct output
12502500

user output
0

Test 19

Group: 2

Verdict:

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

correct output
12502500

user output
0

Test 20

Group: 2

Verdict:

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

correct output
22451

user output
0

Test 21

Group: 2

Verdict:

input
5000 2
132968 414421 468358 432744 43...

correct output
9999

user output
0

Test 22

Group: 2

Verdict:

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

correct output
12502500

user output
0

Test 23

Group: 2

Verdict:

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

correct output
2669737

user output
0

Test 24

Group: 2

Verdict:

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

correct output
498978

user output
0

Test 25

Group: 2

Verdict:

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

correct output
12502500

user output
0

Test 26

Group: 2

Verdict:

input
5000 4999
865706 910619 695192 183574 92...

correct output
12502500

user output
0

Test 27

Group: 2

Verdict:

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

correct output
12502500

user output
0

Test 28

Group: 2

Verdict:

input
5000 5000
774752 159472 183796 654476 69...

correct output
12502500

user output
0

Test 29

Group: 3

Verdict:

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

correct output
125000250000

user output
0

Test 30

Group: 3

Verdict:

input
500000 1
28107328 613212742 298033960 7...

correct output
500000

user output
0

Test 31

Group: 3

Verdict:

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

correct output
125000250000

user output
0

Test 32

Group: 3

Verdict:

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

correct output
125000250000

user output
0

Test 33

Group: 3

Verdict:

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

correct output
2245355

user output
0

Test 34

Group: 3

Verdict:

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

correct output
1570486882

user output
0

Test 35

Group: 3

Verdict:

input
500000 2
318961563 84011941 882177798 1...

correct output
999999

user output
0

Test 36

Group: 3

Verdict:

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

correct output
483793169

user output
0

Test 37

Group: 4

Verdict:

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

correct output
125000250000

user output
0

Test 38

Group: 4

Verdict:

input
500000 1
709486749 350496125 796065873 ...

correct output
500000

user output
0

Test 39

Group: 4

Verdict:

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

correct output
125000250000

user output
0

Test 40

Group: 4

Verdict:

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

correct output
125000250000

user output
0

Test 41

Group: 4

Verdict:

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

correct output
2255047

user output
0

Test 42

Group: 4

Verdict:

input
500000 2
414942284 362802746 108881396 ...

correct output
999999

user output
0

Test 43

Group: 4

Verdict:

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

correct output
260458849

user output
0

Test 44

Group: 4

Verdict:

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

correct output
3694803834

user output
0

Test 45

Group: 4

Verdict:

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

correct output
41655235436

user output
0

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
0

Test 47

Group: 4

Verdict:

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

correct output
125000250000

user output
0

Test 48

Group: 4

Verdict:

input
500000 100000
320513086 811766509 339605137 ...

correct output
45001856854

user output
0

Test 49

Group: 4

Verdict:

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

correct output
125000250000

user output
0

Test 50

Group: 4

Verdict:

input
500000 499999
995132060 587162982 59723733 8...

correct output
125000250000

user output
0

Test 51

Group: 4

Verdict:

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

correct output
125000250000

user output
0

Test 52

Group: 4

Verdict:

input
500000 500000
72083718 753463162 730560260 6...

correct output
125000250000

user output
0