Taulukko

input/code.cpp: In function 'int main()':
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 signed and unsigned integer expressions [-Wsign-compare]
             if(active_nums.size() > k){
                ~~~~~~~~~~~~~~~~~~~^~~
input/code.cpp:20:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d",&n);
     ~~~~~^~~~~~~~~
input/code.cpp:21:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d",&k);
     ~~~~~^~~~~~~~~
inp...

#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("%d",&n);
    scanf("%d",&k);
    
    int temp;
    for(int i=0;i<n;i++){
        scanf("%d",&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 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: ACCEPTED

Test 17

Group: 2

Verdict: ACCEPTED

Test 18

Group: 2

Verdict: ACCEPTED

Test 19

Group: 2

Verdict: ACCEPTED

Test 20

Group: 2

Verdict: ACCEPTED

Test 21

Group: 2

Verdict: ACCEPTED

Test 22

Group: 2

Verdict: ACCEPTED

Test 23

Group: 2

Verdict: ACCEPTED

Test 24

Group: 2

Verdict: ACCEPTED

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

Test 31

Group: 3

Verdict: ACCEPTED

Test 32

Group: 3

Verdict: ACCEPTED

Test 33

Group: 3

Verdict: ACCEPTED

Test 34

Group: 3

Verdict: ACCEPTED

Test 35

Group: 3

Verdict: ACCEPTED

Test 36

Group: 3

Verdict: ACCEPTED

Test 37

Group: 4

Verdict: ACCEPTED

Test 38

Group: 4

Verdict: ACCEPTED

Test 39

Group: 4

Verdict: ACCEPTED

Test 40

Group: 4

Verdict: ACCEPTED

Test 41

Group: 4

Verdict: ACCEPTED

Test 42

Group: 4

Verdict: ACCEPTED

Test 43

Group: 4

Verdict: ACCEPTED

Test 44

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 45

Group: 4

Verdict: TIME LIMIT EXCEEDED

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