Taulukko

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

#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 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: ACCEPTED

Test 4

Group: 1

Verdict: ACCEPTED

Test 5

Group: 1

Verdict: ACCEPTED

Test 6

Group: 1

Verdict: ACCEPTED

Test 7

Group: 1

Verdict: ACCEPTED

Test 8

Group: 1

Verdict: ACCEPTED

Test 9

Group: 1

Verdict: ACCEPTED

Test 10

Group: 1

Verdict: ACCEPTED

Test 11

Group: 1

Verdict: ACCEPTED

Test 12

Group: 1

Verdict: ACCEPTED

Test 13

Group: 1

Verdict: ACCEPTED

Test 14

Group: 1

Verdict: ACCEPTED

Test 15

Group: 2

Verdict: ACCEPTED

Test 16

Group: 2

Verdict: WRONG ANSWER

Test 17

Group: 2

Verdict: WRONG ANSWER

Test 18

Group: 2

Verdict: ACCEPTED

Test 19

Group: 2

Verdict: ACCEPTED

Test 20

Group: 2

Verdict: WRONG ANSWER

Test 21

Group: 2

Verdict: WRONG ANSWER

Test 22

Group: 2

Verdict: ACCEPTED

Test 23

Group: 2

Verdict: WRONG ANSWER

Test 24

Group: 2

Verdict: WRONG ANSWER

Test 25

Group: 2

Verdict: ACCEPTED

Test 26

Group: 2

Verdict: ACCEPTED

Test 27

Group: 2

Verdict: ACCEPTED

Test 28

Group: 2

Verdict: ACCEPTED

Test 29

Group: 3

Verdict: ACCEPTED

Test 30

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 31

Group: 3

Verdict: ACCEPTED

Test 32

Group: 3

Verdict: ACCEPTED

Test 33

Group: 3

Verdict: WRONG ANSWER

Test 34

Group: 3

Verdict: WRONG ANSWER

Test 35

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 36

Group: 3

Verdict: WRONG ANSWER

Test 37

Group: 4

Verdict: ACCEPTED

Test 38

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 39

Group: 4

Verdict: ACCEPTED

Test 40

Group: 4

Verdict: ACCEPTED

Test 41

Group: 4

Verdict: WRONG ANSWER

Test 42

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 43

Group: 4

Verdict: WRONG ANSWER

Test 44

Group: 4

Verdict: WRONG ANSWER

Test 45

Group: 4

Verdict: WRONG ANSWER

Test 46

Group: 4

Verdict: ACCEPTED

Test 47

Group: 4

Verdict: ACCEPTED

Test 48

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 49

Group: 4

Verdict: ACCEPTED

Test 50

Group: 4

Verdict: ACCEPTED

Test 51

Group: 4

Verdict: ACCEPTED

Test 52

Group: 4

Verdict: ACCEPTED