| Task: | Taulukko | 
| Sender: | paavoee | 
| Submission time: | 2018-10-06 14:32:05 +0300 | 
| Language: | C++ | 
| Status: | READY | 
| Result: | 47 | 
| group | verdict | score | 
|---|---|---|
| #1 | ACCEPTED | 12 | 
| #2 | ACCEPTED | 35 | 
| #3 | WRONG ANSWER | 0 | 
| #4 | WRONG ANSWER | 0 | 
| test | verdict | time | group | |
|---|---|---|---|---|
| #1 | ACCEPTED | 0.03 s | 1 | details | 
| #2 | ACCEPTED | 0.02 s | 1 | details | 
| #3 | ACCEPTED | 0.01 s | 1 | details | 
| #4 | ACCEPTED | 0.01 s | 1 | details | 
| #5 | ACCEPTED | 0.02 s | 1 | details | 
| #6 | ACCEPTED | 0.02 s | 1 | details | 
| #7 | ACCEPTED | 0.02 s | 1 | details | 
| #8 | ACCEPTED | 0.01 s | 1 | details | 
| #9 | ACCEPTED | 0.03 s | 1 | details | 
| #10 | ACCEPTED | 0.01 s | 1 | details | 
| #11 | ACCEPTED | 0.01 s | 1 | details | 
| #12 | ACCEPTED | 0.03 s | 1 | details | 
| #13 | ACCEPTED | 0.02 s | 1 | details | 
| #14 | ACCEPTED | 0.01 s | 1 | details | 
| #15 | ACCEPTED | 0.03 s | 2 | details | 
| #16 | ACCEPTED | 0.01 s | 2 | details | 
| #17 | ACCEPTED | 0.03 s | 2 | details | 
| #18 | ACCEPTED | 0.02 s | 2 | details | 
| #19 | ACCEPTED | 0.02 s | 2 | details | 
| #20 | ACCEPTED | 0.02 s | 2 | details | 
| #21 | ACCEPTED | 0.02 s | 2 | details | 
| #22 | ACCEPTED | 0.02 s | 2 | details | 
| #23 | ACCEPTED | 0.02 s | 2 | details | 
| #24 | ACCEPTED | 0.02 s | 2 | details | 
| #25 | ACCEPTED | 0.02 s | 2 | details | 
| #26 | ACCEPTED | 0.02 s | 2 | details | 
| #27 | ACCEPTED | 0.03 s | 2 | details | 
| #28 | ACCEPTED | 0.03 s | 2 | details | 
| #29 | WRONG ANSWER | 0.06 s | 3 | details | 
| #30 | ACCEPTED | 0.17 s | 3 | details | 
| #31 | WRONG ANSWER | 0.06 s | 3 | details | 
| #32 | WRONG ANSWER | 0.07 s | 3 | details | 
| #33 | ACCEPTED | 0.11 s | 3 | details | 
| #34 | ACCEPTED | 0.08 s | 3 | details | 
| #35 | ACCEPTED | 0.18 s | 3 | details | 
| #36 | ACCEPTED | 0.07 s | 3 | details | 
| #37 | WRONG ANSWER | 0.06 s | 4 | details | 
| #38 | ACCEPTED | 0.18 s | 4 | details | 
| #39 | WRONG ANSWER | 0.06 s | 4 | details | 
| #40 | WRONG ANSWER | 0.07 s | 4 | details | 
| #41 | ACCEPTED | 0.11 s | 4 | details | 
| #42 | ACCEPTED | 0.18 s | 4 | details | 
| #43 | ACCEPTED | 0.08 s | 4 | details | 
| #44 | ACCEPTED | 0.09 s | 4 | details | 
| #45 | WRONG ANSWER | 0.09 s | 4 | details | 
| #46 | WRONG ANSWER | 0.07 s | 4 | details | 
| #47 | WRONG ANSWER | 0.11 s | 4 | details | 
| #48 | WRONG ANSWER | 0.29 s | 4 | details | 
| #49 | WRONG ANSWER | 0.06 s | 4 | details | 
| #50 | WRONG ANSWER | 0.32 s | 4 | details | 
| #51 | WRONG ANSWER | 0.06 s | 4 | details | 
| #52 | WRONG ANSWER | 0.29 s | 4 | details | 
Code
#include <iostream>
#include <unordered_map>
#include <vector>
using namespace std;
static inline void remove(unordered_map<unsigned long, unsigned int>& seen, unsigned long l) {
	if (seen[l] > 1) seen[l]--;
	else seen.erase(l);
}
unsigned int mene(unsigned int loppu, unsigned int n, unsigned int k,
		const vector<unsigned long>& x,
		unordered_map<unsigned long, unsigned int>& seen) {
	while (loppu < n) {
		seen[x[loppu]]++;
		if (seen.size() > k) {
			remove(seen, x[loppu]);
			break;
		}
		loppu++;
	}
	return loppu;
}
inline constexpr unsigned int sum(unsigned int alku, unsigned int loppu) {
	return (loppu - alku) * (loppu - alku + 1) / 2;
}
int main() {
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	unsigned int n, k;
	cin >> n >> k;
	vector<unsigned long> x(n);
	for (unsigned int i = 0; i < n; i++)
		cin >> x[i];
	unsigned long r = 0;
	unordered_map<unsigned long, unsigned int> seen;
	unsigned int alku = 0, loppu = 0;
	loppu = mene(loppu, n, k, x, seen);
	r += sum(alku, loppu);
	while (loppu < n) {
		while (seen.size() == k)
			remove(seen, x[alku++]);
		const unsigned int vali = loppu;
		loppu = mene(loppu, n, k, x, seen);
		r += sum(alku, loppu) - sum(alku, vali);
	}
	cout << r << '\n';
	return 0;
}
Test details
Test 1
Group: 1
Verdict: ACCEPTED
| input | 
|---|
| 100 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... | 
| correct output | 
|---|
| 5050 | 
| user output | 
|---|
| 5050 | 
Test 2
Group: 1
Verdict: ACCEPTED
| input | 
|---|
| 100 1 1 1 1 1 2 2 1 1 2 2 2 2 2 1 1 ... | 
| correct output | 
|---|
| 190 | 
| user output | 
|---|
| 190 | 
Test 3
Group: 1
Verdict: ACCEPTED
| input | 
|---|
| 100 1 5 9 9 6 9 8 1 4 7 7 8 9 5 5 6 ... | 
| correct output | 
|---|
| 110 | 
| user output | 
|---|
| 110 | 
Test 4
Group: 1
Verdict: ACCEPTED
| input | 
|---|
| 100 2 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... | 
| correct output | 
|---|
| 5050 | 
| user output | 
|---|
| 5050 | 
Test 5
Group: 1
Verdict: ACCEPTED
| input | 
|---|
| 100 2 2 1 2 1 2 2 1 1 2 1 1 2 1 2 1 ... | 
| correct output | 
|---|
| 5050 | 
| user output | 
|---|
| 5050 | 
Test 6
Group: 1
Verdict: ACCEPTED
| input | 
|---|
| 100 2 2 3 1 1 2 2 3 1 2 1 1 1 3 3 1 ... | 
| correct output | 
|---|
| 379 | 
| user output | 
|---|
| 379 | 
Test 7
Group: 1
Verdict: ACCEPTED
| input | 
|---|
| 100 2 4 6 10 8 6 8 10 8 4 7 8 9 6 2 ... | 
| correct output | 
|---|
| 245 | 
| user output | 
|---|
| 245 | 
Test 8
Group: 1
Verdict: ACCEPTED
| input | 
|---|
| 100 5 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... | 
| correct output | 
|---|
| 5050 | 
| user output | 
|---|
| 5050 | 
Test 9
Group: 1
Verdict: ACCEPTED
| input | 
|---|
| 100 5 4 4 2 2 4 3 3 2 3 4 3 5 3 1 5 ... | 
| correct output | 
|---|
| 5050 | 
| user output | 
|---|
| 5050 | 
Test 10
Group: 1
Verdict: ACCEPTED
| input | 
|---|
| 100 5 5 3 4 1 1 1 1 4 5 5 4 6 6 3 3 ... | 
| correct output | 
|---|
| 1488 | 
| user output | 
|---|
| 1488 | 
Test 11
Group: 1
Verdict: ACCEPTED
| input | 
|---|
| 100 5 10 1 1 9 1 6 9 4 3 10 9 2 4 2 ... | 
| correct output | 
|---|
| 743 | 
| user output | 
|---|
| 743 | 
Test 12
Group: 1
Verdict: ACCEPTED
| input | 
|---|
| 100 10 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... | 
| correct output | 
|---|
| 5050 | 
| user output | 
|---|
| 5050 | 
Test 13
Group: 1
Verdict: ACCEPTED
| input | 
|---|
| 100 10 9 4 9 1 9 1 5 2 10 10 9 3 9 6 ... | 
| correct output | 
|---|
| 5050 | 
| user output | 
|---|
| 5050 | 
Test 14
Group: 1
Verdict: ACCEPTED
| input | 
|---|
| 100 10 80 59 58 87 28 83 83 93 96 24 ... | 
| correct output | 
|---|
| 994 | 
| user output | 
|---|
| 994 | 
Test 15
Group: 2
Verdict: ACCEPTED
| input | 
|---|
| 5000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... | 
| correct output | 
|---|
| 12502500 | 
| user output | 
|---|
| 12502500 | 
Test 16
Group: 2
Verdict: ACCEPTED
| input | 
|---|
| 5000 1 2 2 2 1 2 1 1 1 2 2 1 2 1 1 1 ... | 
| correct output | 
|---|
| 10148 | 
| user output | 
|---|
| 10148 | 
Test 17
Group: 2
Verdict: ACCEPTED
| input | 
|---|
| 5000 1 434568 634119 102509 107238 72... | 
| correct output | 
|---|
| 5000 | 
| user output | 
|---|
| 5000 | 
Test 18
Group: 2
Verdict: ACCEPTED
| input | 
|---|
| 5000 2 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... | 
| correct output | 
|---|
| 12502500 | 
| user output | 
|---|
| 12502500 | 
Test 19
Group: 2
Verdict: ACCEPTED
| input | 
|---|
| 5000 2 2 1 1 1 2 2 2 2 1 1 1 1 2 1 2 ... | 
| correct output | 
|---|
| 12502500 | 
| user output | 
|---|
| 12502500 | 
Test 20
Group: 2
Verdict: ACCEPTED
| input | 
|---|
| 5000 2 1 3 2 2 3 3 1 2 1 2 1 2 3 1 2 ... | 
| correct output | 
|---|
| 22451 | 
| user output | 
|---|
| 22451 | 
Test 21
Group: 2
Verdict: ACCEPTED
| input | 
|---|
| 5000 2 132968 414421 468358 432744 43... | 
| correct output | 
|---|
| 9999 | 
| user output | 
|---|
| 9999 | 
Test 22
Group: 2
Verdict: ACCEPTED
| input | 
|---|
| 5000 100 93 76 87 71 93 74 69 35 92 96 ... | 
| correct output | 
|---|
| 12502500 | 
| user output | 
|---|
| 12502500 | 
Test 23
Group: 2
Verdict: ACCEPTED
| input | 
|---|
| 5000 100 2 57 51 4 35 76 5 40 51 55 20 ... | 
| correct output | 
|---|
| 2669737 | 
| user output | 
|---|
| 2669737 | 
Test 24
Group: 2
Verdict: ACCEPTED
| input | 
|---|
| 5000 100 657 823 1773 1801 2785 4107 25... | 
| correct output | 
|---|
| 498978 | 
| user output | 
|---|
| 498978 | 
Test 25
Group: 2
Verdict: ACCEPTED
| input | 
|---|
| 5000 4999 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... | 
| correct output | 
|---|
| 12502500 | 
| user output | 
|---|
| 12502500 | 
Test 26
Group: 2
Verdict: ACCEPTED
| input | 
|---|
| 5000 4999 865706 910619 695192 183574 92... | 
| correct output | 
|---|
| 12502500 | 
| user output | 
|---|
| 12502500 | 
Test 27
Group: 2
Verdict: ACCEPTED
| input | 
|---|
| 5000 5000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... | 
| correct output | 
|---|
| 12502500 | 
| user output | 
|---|
| 12502500 | 
Test 28
Group: 2
Verdict: ACCEPTED
| input | 
|---|
| 5000 5000 774752 159472 183796 654476 69... | 
| correct output | 
|---|
| 12502500 | 
| user output | 
|---|
| 12502500 | 
Test 29
Group: 3
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... | 
| correct output | 
|---|
| 125000250000 | 
| user output | 
|---|
| 446198416 | 
Test 30
Group: 3
Verdict: ACCEPTED
| input | 
|---|
| 500000 1 28107328 613212742 298033960 7... | 
| correct output | 
|---|
| 500000 | 
| user output | 
|---|
| 500000 | 
Test 31
Group: 3
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 2 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... | 
| correct output | 
|---|
| 125000250000 | 
| user output | 
|---|
| 446198416 | 
Test 32
Group: 3
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 2 1 1 2 1 1 2 2 1 2 1 2 1 1 2 1 ... | 
| correct output | 
|---|
| 125000250000 | 
| user output | 
|---|
| 446198416 | 
Test 33
Group: 3
Verdict: ACCEPTED
| input | 
|---|
| 500000 2 3 2 2 3 2 2 2 1 1 2 2 3 2 3 2 ... | 
| correct output | 
|---|
| 2245355 | 
| user output | 
|---|
| 2245355 | 
Test 34
Group: 3
Verdict: ACCEPTED
| input | 
|---|
| 500000 2 1 1 1 2 2 2 1 1 2 1 1 2 1 2 1 ... | 
| correct output | 
|---|
| 1570486882 | 
| user output | 
|---|
| 1570486882 | 
Test 35
Group: 3
Verdict: ACCEPTED
| input | 
|---|
| 500000 2 318961563 84011941 882177798 1... | 
| correct output | 
|---|
| 999999 | 
| user output | 
|---|
| 999999 | 
Test 36
Group: 3
Verdict: ACCEPTED
| input | 
|---|
| 500000 2 1 1 2 1 2 1 2 1 2 2 2 2 1 2 1 ... | 
| correct output | 
|---|
| 483793169 | 
| user output | 
|---|
| 483793169 | 
Test 37
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... | 
| correct output | 
|---|
| 125000250000 | 
| user output | 
|---|
| 446198416 | 
Test 38
Group: 4
Verdict: ACCEPTED
| input | 
|---|
| 500000 1 709486749 350496125 796065873 ... | 
| correct output | 
|---|
| 500000 | 
| user output | 
|---|
| 500000 | 
Test 39
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 2 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... | 
| correct output | 
|---|
| 125000250000 | 
| user output | 
|---|
| 446198416 | 
Test 40
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 2 1 2 2 2 2 2 1 1 2 2 2 2 2 2 1 ... | 
| correct output | 
|---|
| 125000250000 | 
| user output | 
|---|
| 446198416 | 
Test 41
Group: 4
Verdict: ACCEPTED
| input | 
|---|
| 500000 2 3 3 3 3 2 1 3 2 1 3 1 2 1 1 1 ... | 
| correct output | 
|---|
| 2255047 | 
| user output | 
|---|
| 2255047 | 
Test 42
Group: 4
Verdict: ACCEPTED
| input | 
|---|
| 500000 2 414942284 362802746 108881396 ... | 
| correct output | 
|---|
| 999999 | 
| user output | 
|---|
| 999999 | 
Test 43
Group: 4
Verdict: ACCEPTED
| input | 
|---|
| 500000 100 59 14 100 74 49 43 91 84 31 16... | 
| correct output | 
|---|
| 260458849 | 
| user output | 
|---|
| 260458849 | 
Test 44
Group: 4
Verdict: ACCEPTED
| input | 
|---|
| 500000 1000 505 511 86 321 780 495 106 330... | 
| correct output | 
|---|
| 3694803834 | 
| user output | 
|---|
| 3694803834 | 
Test 45
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 10000 4956 8463 8582 980 9278 1747 2... | 
| correct output | 
|---|
| 41655235436 | 
| user output | 
|---|
| 58835104620 | 
Test 46
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 100000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... | 
| correct output | 
|---|
| 125000250000 | 
| user output | 
|---|
| 446198416 | 
Test 47
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 100000 40198 78364 3724 51802 378 130... | 
| correct output | 
|---|
| 125000250000 | 
| user output | 
|---|
| 446198416 | 
Test 48
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 100000 320513086 811766509 339605137 ... | 
| correct output | 
|---|
| 45001856854 | 
| user output | 
|---|
| 40706889558 | 
Test 49
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 499999 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... | 
| correct output | 
|---|
| 125000250000 | 
| user output | 
|---|
| 446198416 | 
Test 50
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 499999 995132060 587162982 59723733 8... | 
| correct output | 
|---|
| 125000250000 | 
| user output | 
|---|
| 446198416 | 
Test 51
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 500000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... | 
| correct output | 
|---|
| 125000250000 | 
| user output | 
|---|
| 446198416 | 
Test 52
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 500000 72083718 753463162 730560260 6... | 
| correct output | 
|---|
| 125000250000 | 
| user output | 
|---|
| 446198416 | 
