Taulukko

input/code.cpp: In function 'int main()':
input/code.cpp:61:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i = 0; i < ranges.size(); i++)
                 ~~^~~~~~~~~~~~~~~
input/code.cpp:64:24: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for(int j = 1; i + j < ranges.size(); j++)
                  ~~~~~~^~~~~~~~~~~~~~~

#include <iostream>
#include <map>
#include <unordered_set>
#include <vector>

using namespace std;

long long int tri(long long int n)
{
	return (n * (n + 1)) / 2;
}

int main()
{
	int n, k;
	cin >> n >> k;

	int *a = new int[n];
	for(int i = 0; i < n; i++)
		cin >> a[i];

	vector<pair<int, int>> ranges, overlaps;
	vector<int> ols;
	unordered_multiset<int> nums;

	for(int i = 0; i < n; i++)
	{
		unordered_multiset<int> nc = nums;
		if(i)
		{
			for(; i < n; i++)
			{
				nc.erase(a[i]);
				if(!nc.count(a[i]))
					break;
			}
		}
		
		if(i == n)
			break;
	
		int u = 0;
		nums.clear();

		int j = i;
		for(; j < n; j++)
		{
			if(!nums.count(a[j]))
			{
				u++;
				if(u > k)
					break;
			}
			nums.insert(a[j]);
		}

		if(ranges.empty() || ranges.back().second < j-1)
			ranges.push_back(make_pair(i, j-1));	
	}

	for(int i = 0; i < ranges.size(); i++)
	{
		//cout << "(" << ranges[i].first << ", " << ranges[i].second << "): ";
		for(int j = 1; i + j < ranges.size(); j++)
		{
			if(ranges[i].second < ranges[i+j].first)
				break;

			//cout << "(" << ranges[i+j].first << ", " << ranges[i+j].second << "), ";
			//overlaps.push_back(make_pair(ranges[i+j].first, ranges[i].second));
			ols.push_back(ranges[i].second - ranges[i+j].first + 1);
		}
		//cout << endl;
	}

	/*for(pair<int, int> p : ranges)
		cout << p.first << ", " << p.second << endl;

	cout << endl;
	for(int i = 0; i < ols.size(); i++)
		cout << overlaps[i].first << ", " << overlaps[i].second << ", " << ols[i] << endl;*/

	long long int ret = 0;

	for(pair<int, int> p : ranges)
		ret += tri(p.second - p.first + 1);

	for(int i : ols)
		ret -= tri(i);

	cout << ret << 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: 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: TIME LIMIT EXCEEDED

Test 16

Group: 2

Verdict: ACCEPTED

Test 17

Group: 2

Verdict: ACCEPTED

Test 18

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 19

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 20

Group: 2

Verdict: ACCEPTED

Test 21

Group: 2

Verdict: ACCEPTED

Test 22

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 23

Group: 2

Verdict: WRONG ANSWER

Test 24

Group: 2

Verdict: WRONG ANSWER

Test 25

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 26

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 27

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 28

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 29

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 30

Group: 3

Verdict: ACCEPTED

Test 31

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 32

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 33

Group: 3

Verdict: ACCEPTED

Test 34

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 35

Group: 3

Verdict: ACCEPTED

Test 36

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 37

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 38

Group: 4

Verdict: ACCEPTED

Test 39

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 40

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 41

Group: 4

Verdict: ACCEPTED

Test 42

Group: 4

Verdict: ACCEPTED

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: TIME LIMIT EXCEEDED

Test 47

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 48

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 49

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 50

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 51

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 52

Group: 4

Verdict: TIME LIMIT EXCEEDED