Taulukko

input/code.cpp: In function 'std::vector<std::pair<int, int> > howMany(int)':
input/code.cpp:23:7: warning: unused variable 'toimiko' [-Wunused-variable]
  bool toimiko = false;
       ^~~~~~~
input/code.cpp: In function 'int main()':
input/code.cpp:105:16: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for(int i=0;i<testi.size();i++){
               ~^~~~~~~~~~~~~
input/code.cpp:122:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=1;i<kaikkiYhteydet.size();i++){
              ~^~~~~~~~~~~~~~~~~~~~~~

#include<bits/stdc++.h>
using namespace std;

int numberCount;
int maxDifferent;
int number;
vector<int> numbers;
int badPlaces[1010101];
vector<pair<int,int>> howMany(int size){
	int beginning = 0;
	vector<pair<int,int>> kohdat;
	int end = size-1;
	map<int,int> used = {};
	int counterFunct = 0;
	for(int i=0;i<size;i++){
		if(!used[numbers[i]]){
			used[numbers[i]] = 1;
			counterFunct++;
		}else{
			used[numbers[i]]++;
		}
	}
	bool toimiko = false;
	while(end < numberCount){
//		cout << beginning << " " << end << " " << counterFunct << " JUTTU\n";
		if(badPlaces[beginning]){
			beginning=badPlaces[beginning];
			end=beginning + size - 1;
			continue;
		}
		if(counterFunct <= maxDifferent){
			kohdat.push_back({beginning,end});
		}
		used[numbers[beginning]]--;
		if(used[numbers[beginning]] == 0){
			counterFunct--;
		}
		used[numbers[end+1]]++;
		if(used[numbers[end+1]]==1){
			counterFunct++;
		}
		beginning++;
		end++;
	}
	return kohdat;
}

bool checking(int size){
	
	int beginning = 0;
	int end = size-1;
	map<int,int> used = {};
	int counterFunct = 0;
	for(int i=0;i<size;i++){
		if(!used[numbers[i]]){
			used[numbers[i]] = 1;
			counterFunct++;
		}else{
			used[numbers[i]]++;
		}
	}
	while(end < numberCount){
//		cout << beginning << " " << end << " " << counterFunct << " "  << badPlaces[beginning] << " TESTI \n";
		if(badPlaces[beginning]){
			beginning=badPlaces[beginning];
			end=beginning+size-1;
			continue;
		}
		if(counterFunct <= maxDifferent){
			return true;	
		}
		used[numbers[beginning]]--;
		if(used[numbers[beginning]] == 0){
			counterFunct--;
		}
	 	used[numbers[end+1]]++;
		if(used[numbers[end+1]]==1){
			counterFunct++;
		}
		beginning++;
		end++;
	}

	return false;
}
vector<int> laskujutut;
int main(){
	cin >> numberCount >> maxDifferent;
	for(int i=0;i<numberCount;i++){
		cin >> number;
		numbers.push_back(number);
	}
	int k = numberCount;
	//cout << "\n\n";
	long long takapakki = 0;
	vector<pair<int,int>> kaikkiYhteydet;
	while(k!=0){
		int x = -1;
		for(int b = k; b>=1; b/=2){
			while(checking(x+b) == true){
				x += b;
			}
		}
		vector<pair<int,int>> testi = howMany(x);
		for(int i=0;i<testi.size();i++){
			//cout << i << " " << testi.size() << " " << testi[i].first << " " << testi[i].second <<  ":   " << testi[i-1].first << " " << testi[i-1].second << "\n";
			badPlaces[testi[i].first] = testi[i].second;
			laskujutut.push_back(testi[i].second-testi[i].first+1);
	 	
		}
		//cout << k << " " << x-1 << " VÄLISSÄ\n";
		k = x-1;
		//cout << k << "\n";
		kaikkiYhteydet.insert(kaikkiYhteydet.end(),testi.begin(),testi.end());
	}
	long long summa = 0;
	for(auto u : laskujutut){
		summa += u*(u+1)/2;
	}
	kaikkiYhteydet.push_back({-500,-500});
	sort(kaikkiYhteydet.begin(),kaikkiYhteydet.end());
	for(int i=1;i<kaikkiYhteydet.size();i++){
		//§cout << kaikkiYhteydet[i].first << " " << kaikkiYhteydet[i-1].second << " SECOND\n";
		if(kaikkiYhteydet[i].first <= kaikkiYhteydet[i-1].second){
			//cout << (kaikkiYhteydet[i].second-kaikkiYhteydet[i].first+1) << " " << kaikkiYhteydet[i].second-kaikkiYhteydet[i].first+1 << " JES\n";
			takapakki += (kaikkiYhteydet[i-1].second-kaikkiYhteydet[i].first+1)*(kaikkiYhteydet[i-1].second-kaikkiYhteydet[i].first+2)/2;
		}
	}
//	cout << kaikkiYhteydet.size() << " Kaikki\n";
	cout << summa - takapakki<< "\n";
//	cout << get<0>(checking(x-1)) << " JA ULOS TULTIIN " << x-1 << " CHECKING0\n";
//	cout << "VALTTERI\n";
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: WRONG ANSWER

Test 3

Group: 1

Verdict: WRONG ANSWER

Test 4

Group: 1

Verdict: ACCEPTED

Test 5

Group: 1

Verdict: ACCEPTED

Test 6

Group: 1

Verdict: WRONG ANSWER

Test 7

Group: 1

Verdict: WRONG ANSWER

Test 8

Group: 1

Verdict: ACCEPTED

Test 9

Group: 1

Verdict: ACCEPTED

Test 10

Group: 1

Verdict: WRONG ANSWER

Test 11

Group: 1

Verdict: WRONG ANSWER

Test 12

Group: 1

Verdict: ACCEPTED

Test 13

Group: 1

Verdict: ACCEPTED

Test 14

Group: 1

Verdict: WRONG ANSWER

Test 15

Group: 2

Verdict: ACCEPTED

Test 16

Group: 2

Verdict: WRONG ANSWER

Test 17

Group: 2

Verdict: ACCEPTED

Test 18

Group: 2

Verdict: ACCEPTED

Test 19

Group: 2

Verdict: ACCEPTED

Test 20

Group: 2

Verdict: WRONG ANSWER

Test 21

Group: 2

Verdict: ACCEPTED

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: RUNTIME ERROR

Test 30

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 31

Group: 3

Verdict: RUNTIME ERROR

Test 32

Group: 3

Verdict: RUNTIME ERROR

Test 33

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 34

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 35

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 36

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 37

Group: 4

Verdict: RUNTIME ERROR

Test 38

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 39

Group: 4

Verdict: RUNTIME ERROR

Test 40

Group: 4

Verdict: RUNTIME ERROR

Test 41

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 42

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 43

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 44

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 45

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 46

Group: 4

Verdict: RUNTIME ERROR

Test 47

Group: 4

Verdict: RUNTIME ERROR

Test 48

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 49

Group: 4

Verdict: RUNTIME ERROR

Test 50

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 51

Group: 4

Verdict: RUNTIME ERROR

Test 52

Group: 4

Verdict: TIME LIMIT EXCEEDED