CSES - Datatähti 2019 alku - Results
Submission details
Task:Taulukko
Sender:borkbork
Submission time:2018-10-06 20:01:07 +0300
Language:C++
Status:READY
Result:12
Feedback
groupverdictscore
#1ACCEPTED12
#20
#30
#40
Test results
testverdicttimegroup
#1ACCEPTED0.03 s1details
#2ACCEPTED0.01 s1details
#3ACCEPTED0.02 s1details
#4ACCEPTED0.01 s1details
#5ACCEPTED0.01 s1details
#6ACCEPTED0.01 s1details
#7ACCEPTED0.02 s1details
#8ACCEPTED0.01 s1details
#9ACCEPTED0.01 s1details
#10ACCEPTED0.01 s1details
#11ACCEPTED0.01 s1details
#12ACCEPTED0.02 s1details
#13ACCEPTED0.02 s1details
#14ACCEPTED0.03 s1details
#15ACCEPTED0.01 s2details
#160.01 s2details
#170.03 s2details
#18ACCEPTED0.01 s2details
#19ACCEPTED0.02 s2details
#200.02 s2details
#210.04 s2details
#22ACCEPTED0.02 s2details
#230.02 s2details
#240.05 s2details
#25ACCEPTED0.02 s2details
#26ACCEPTED0.02 s2details
#27ACCEPTED0.02 s2details
#28ACCEPTED0.03 s2details
#29ACCEPTED0.06 s3details
#30--3details
#31ACCEPTED0.05 s3details
#32ACCEPTED0.06 s3details
#330.12 s3details
#340.13 s3details
#35--3details
#360.18 s3details
#37ACCEPTED0.05 s4details
#38--4details
#39ACCEPTED0.06 s4details
#40ACCEPTED0.06 s4details
#410.12 s4details
#42--4details
#430.11 s4details
#440.11 s4details
#450.12 s4details
#46ACCEPTED0.06 s4details
#47ACCEPTED0.13 s4details
#48--4details
#49ACCEPTED0.05 s4details
#50ACCEPTED0.30 s4details
#51ACCEPTED0.06 s4details
#52ACCEPTED0.30 s4details

Compiler report

input/code.cpp: In function 'int main()':
input/code.cpp:45:24: warning: unused variable 'count' [-Wunused-variable]
     unsigned long long count=0;
                        ^~~~~

Code

#include<iostream>
#include<unordered_set>
#include<unordered_map>

int main(){
    std::ios_base::sync_with_stdio(0);std::cin.tie(NULL);
    unsigned int n,k;
    std::cin>>n>>k;
    unsigned int d[500100];
    unsigned int input;
    std::unordered_set<unsigned int> s;
    for(size_t i=0;i<n;i++){std::cin>>input;s.insert(input);d[i]=input;}
    unsigned int uniq=s.size();
    unsigned long long res=n;
    for(size_t i=1;i<k;i++){res+=(n-i);}
    if(n==k){std::cout<<res;return 0;}
    if(uniq<1+k){for(size_t i=k;i<n;i++){res+=(n-i);}std::cout<<res;return 0;}
    if(n*3/4<k||n<101){

    unsigned long long count=0;
    std::unordered_map<unsigned int,unsigned int> map;
    
    for(size_t i=0;i<n-k+1;i++){
        count=1;
        map.clear();
        map[d[i]]++;
        for(size_t j=1;j<n-i;j++){
            
            if(!map[d[i+j]]){
                count++;
                map[d[i+j]]++;
            }
            if(count>k){break;}
            if(k<j+1){
                if(count<1+k){
                    res++;
                    
                }
            }
        }
    }

    }
    else{
    unsigned long long count=0;
    std::unordered_map<unsigned int,unsigned int> map;
    size_t i=0;
    size_t size=n-2;
    for(;i<n;){
        if(n-k-1<i){break;}
        if(n-i<size){size=n-i;}
        map.clear();
        for(size_t j=0;j<size;j++){map[d[i+j]]++;}
        if(map.size()<k+1){res+=((size+1)*size/2);i=i+size-k;}
        else{
            while(k<size){
                if(map[d[i+size]]<2){map.erase(d[i+size]);}
                else{map[d[i+size]]--;}
                size--;
                if(map.size()<k+1){res+=((size+1)*size/2);i=i+size-k;break;}
            }
        }


        i++;
    }
    }
    std::cout<<res;
}

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:

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

correct output
10148

user output
10009

Test 17

Group: 2

Verdict:

input
5000 1
434568 634119 102509 107238 72...

correct output
5000

user output
9998

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:

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

correct output
22451

user output
24993

Test 21

Group: 2

Verdict:

input
5000 2
132968 414421 468358 432744 43...

correct output
9999

user output
24990

Test 22

Group: 2

Verdict: ACCEPTED

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

correct output
12502500

user output
12502500

Test 23

Group: 2

Verdict:

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

correct output
2669737

user output
1917157

Test 24

Group: 2

Verdict:

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

correct output
498978

user output
25240050

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:

input
500000 1
28107328 613212742 298033960 7...

correct output
500000

user output
(empty)

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:

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

correct output
2245355

user output
2500012

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
2703834

Test 35

Group: 3

Verdict:

input
500000 2
318961563 84011941 882177798 1...

correct output
999999

user output
(empty)

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
2543064

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:

input
500000 1
709486749 350496125 796065873 ...

correct output
500000

user output
(empty)

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:

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

correct output
2255047

user output
2500002

Test 42

Group: 4

Verdict:

input
500000 2
414942284 362802746 108881396 ...

correct output
999999

user output
(empty)

Test 43

Group: 4

Verdict:

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

correct output
260458849

user output
157234927

Test 44

Group: 4

Verdict:

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

correct output
3694803834

user output
2213989748

Test 45

Group: 4

Verdict:

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

correct output
41655235436

user output
27399493696

Test 46

Group: 4

Verdict: ACCEPTED

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

correct output
125000250000

user output
125000250000

Test 47

Group: 4

Verdict: ACCEPTED

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

correct output
125000250000

user output
125000250000

Test 48

Group: 4

Verdict:

input
500000 100000
320513086 811766509 339605137 ...

correct output
45001856854

user output
(empty)

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: ACCEPTED

input
500000 499999
995132060 587162982 59723733 8...

correct output
125000250000

user output
125000250000

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