Taulukko

input/code.cpp: In function 'int main()':
input/code.cpp:40:22: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   while (uniq.size() <= x && end < n) {
          ~~~~~~~~~~~~^~~~
input/code.cpp:41:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    if (uniq.size() == x && freq[a[end]] == 0) {
        ~~~~~~~~~~~~^~~~

#include <iostream>
#include <vector>
#include <algorithm>
#include <set>
#include <map>

using namespace std;

int sub(int n) {
	return (n*n + n) / 2;
}

int secSub(int n) {
	return sub(n) - n;
}

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

	vector<int> a;
	for (int i = 1; i <= n; i++) {
		int b;
		cin >> b;
		a.push_back(b);
	}

	if (x == 1) {
		cout << sub(n) << endl;
		return 0;
	}


	int end = 0;
	long long ans = 0;

	set<int> uniq;
	map<int, int> freq;
	for (int start = 0; start < n; start++) {
		while (uniq.size() <= x && end < n) {
			if (uniq.size() == x && freq[a[end]] == 0) {
				break;
			}
			uniq.insert(a[end]);
			freq[a[end]]++;
			end++;
		}
		ans += end - start;
		freq[a[start]]--;
		if (freq[a[start]] == 0) {
			uniq.erase(a[start]);
		}
	}
	cout << ans;
	int q;
	cin >> q;
}

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: 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: 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: WRONG ANSWER

Test 30

Group: 3

Verdict: WRONG ANSWER

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: WRONG ANSWER

Test 38

Group: 4

Verdict: WRONG ANSWER

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

Test 45

Group: 4

Verdict: ACCEPTED

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

Test 51

Group: 4

Verdict: ACCEPTED

Test 52

Group: 4

Verdict: TIME LIMIT EXCEEDED