| Task: | Taulukko | 
| Sender: | N00B.exe | 
| Submission time: | 2018-10-14 20:33:57 +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.02 s | 1 | details | 
| #2 | ACCEPTED | 0.02 s | 1 | details | 
| #3 | ACCEPTED | 0.02 s | 1 | details | 
| #4 | ACCEPTED | 0.02 s | 1 | details | 
| #5 | ACCEPTED | 0.03 s | 1 | details | 
| #6 | ACCEPTED | 0.01 s | 1 | details | 
| #7 | ACCEPTED | 0.02 s | 1 | details | 
| #8 | ACCEPTED | 0.02 s | 1 | details | 
| #9 | ACCEPTED | 0.02 s | 1 | details | 
| #10 | ACCEPTED | 0.02 s | 1 | details | 
| #11 | ACCEPTED | 0.02 s | 1 | details | 
| #12 | ACCEPTED | 0.01 s | 1 | details | 
| #13 | ACCEPTED | 0.02 s | 1 | details | 
| #14 | ACCEPTED | 0.02 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.01 s | 2 | details | 
| #19 | ACCEPTED | 0.01 s | 2 | details | 
| #20 | ACCEPTED | 0.01 s | 2 | details | 
| #21 | ACCEPTED | 0.03 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.01 s | 2 | details | 
| #26 | ACCEPTED | 0.03 s | 2 | details | 
| #27 | ACCEPTED | 0.02 s | 2 | details | 
| #28 | ACCEPTED | 0.02 s | 2 | details | 
| #29 | WRONG ANSWER | 0.06 s | 3 | details | 
| #30 | WRONG ANSWER | 0.30 s | 3 | details | 
| #31 | WRONG ANSWER | 0.07 s | 3 | details | 
| #32 | WRONG ANSWER | 0.06 s | 3 | details | 
| #33 | WRONG ANSWER | 0.06 s | 3 | details | 
| #34 | WRONG ANSWER | 0.06 s | 3 | details | 
| #35 | WRONG ANSWER | 0.30 s | 3 | details | 
| #36 | WRONG ANSWER | 0.07 s | 3 | details | 
| #37 | WRONG ANSWER | 0.06 s | 4 | details | 
| #38 | WRONG ANSWER | 0.31 s | 4 | details | 
| #39 | WRONG ANSWER | 0.06 s | 4 | details | 
| #40 | WRONG ANSWER | 0.06 s | 4 | details | 
| #41 | WRONG ANSWER | 0.07 s | 4 | details | 
| #42 | WRONG ANSWER | 0.31 s | 4 | details | 
| #43 | WRONG ANSWER | 0.07 s | 4 | details | 
| #44 | WRONG ANSWER | 0.07 s | 4 | details | 
| #45 | WRONG ANSWER | 0.08 s | 4 | details | 
| #46 | WRONG ANSWER | 0.06 s | 4 | details | 
| #47 | WRONG ANSWER | 0.13 s | 4 | details | 
| #48 | WRONG ANSWER | 0.30 s | 4 | details | 
| #49 | WRONG ANSWER | 0.06 s | 4 | details | 
| #50 | WRONG ANSWER | 0.30 s | 4 | details | 
| #51 | WRONG ANSWER | 0.05 s | 4 | details | 
| #52 | WRONG ANSWER | 0.31 s | 4 | details | 
Compiler report
input/code.cpp: In function 'int main()':
input/code.cpp:49:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  if(nn.size() <= k)
     ~~~~~~~~~~^~~~
input/code.cpp:28:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d %d", &n, &k);
  ~~~~~^~~~~~~~~~~~~~~~~
input/code.cpp:35:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d",(nums + i));
   ~~~~~^~~~~~~~~~~~~~~~~Code
#include <iostream>
#include <stdio.h>
#include <unordered_set>
#include <map>
#include <string>
#define USETI unordered_set<int>
#define CNT(c, val) c.count(val)
#define NSRT(c, val) c.insert(val)
#define PRTI(i) printf("%d", i)
#define PRT(s) printf(s)
#define LBRK printf("\n")
#define PAUSE getchar();getchar()
#define FORS(num, val, n) for(int num = val; num < n; num++)
#define FOR(num, n) for(int num = 0; num < n; num++)
#define SITUATION LBRK; PRT("dNumbers = "); PRTI(dNumbers); LBRK; PRT("Start and end = "); PRTI(start); PRT(" "); PRTI(end); LBRK; PRT("Ans = "); PRTI(ans); LBRK; PAUSE; PAUSE
typedef long long ll;
using namespace std;
int main() {
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	int n, k;
	ll ans = 0;
	scanf("%d %d", &n, &k);
	int* nums = (int*)malloc(4*n);
	USETI nn;
	FOR(i, n)
	{ 
		scanf("%d",(nums + i));
		NSRT(nn, *(nums + i));
		
		//if(nn.size() <= k)
		//{ ans++; }
	}
	if(n > 499900)
	{
		PRT("x");
		return 0;
	}
	if(nn.size() <= k)
	{
		ll f = n;
		ans = f * ((f + 1) / 2.0);
		printf("%lld", ans);
		free(nums);
		return 0;
	}
	nn.clear();
	map<int, int> m;
	int start = 0;
	int end = -1;
	int* endP = nums + end;
	int* startP = nums + start;
	int dNumbers = 0;
	while (dNumbers <= k)
	{
		end++;
		endP++;
		if (m[*endP] == 0)
		{ dNumbers++; }
		//LBRK;
		//PRT("add number "); PRTI(*(nums + end));
		//LBRK;
		m[*endP] += 1;
	}
	ans += (ll)(end - start);
	//SITUATION;
	while (start < n - k)
	{
		end++;
		endP++;
		if(end >= n)
		{
			end = n;
			endP = nums + n;
		}
		else
		{
			//LBRK;
			//PRT("add number "); PRTI(*(nums + end));
			//LBRK;
			if (m[*endP] == 0)
			{ dNumbers++; }
			m[*endP] += 1;
		}
		//LBRK;
		//PRT("remove number "); PRTI(*(nums + start));
		//LBRK;
		m[*startP] -= 1;
		if (m[*startP] == 0)
		{
			dNumbers--;
		}
		start++;
		if(start >= n - k)
		{ break; }
		startP++;
		
		if(dNumbers > k)
		{
			while (end - start >= k + 1)
			{
				//LBRK;
				//PRT("remove number "); PRTI(*(nums + end));
				//LBRK;
				m[*endP] -= 1;
				if (m[*endP] == 0)
				{
					if (dNumbers == k + 1)
					{
						m[*endP] += 1;
						break;
					}
					dNumbers--;
				}
				end--;
				endP--;
				/*
				if (dNumbers <= k)
				{
					end++;
					endP++;
					if (m[*endP] == 0)
					{
						dNumbers++;
					}
					else
					{
						PRT("x"); return 0;
					}
					//LBRK;
					//PRT("add number "); PRTI(*(nums + end));
					//LBRK;
					m[*endP] += 1;
					break;
				}
				*/
			}
		}
		
		while (dNumbers <= k)
		{
			end++;
			endP++;
			if (end >= n)
			{
				end = n;
				break;
			}
			if (m[*endP] == 0)
			{ dNumbers++; }
			m[*endP] += 1;
		}
		ans += (ll)(end - start);
		//SITUATION;
	}
	ll lk = k;
	ans += lk * ((lk + 1) / 2.0);
	printf("%lld", ans);
	free(nums);
	//PAUSE;
	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 | 
|---|
| x | 
Test 30
Group: 3
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 1 28107328 613212742 298033960 7... | 
| correct output | 
|---|
| 500000 | 
| user output | 
|---|
| x | 
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 | 
|---|
| x | 
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 | 
|---|
| x | 
Test 33
Group: 3
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 2 3 2 2 3 2 2 2 1 1 2 2 3 2 3 2 ... | 
| correct output | 
|---|
| 2245355 | 
| user output | 
|---|
| x | 
Test 34
Group: 3
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 2 1 1 1 2 2 2 1 1 2 1 1 2 1 2 1 ... | 
| correct output | 
|---|
| 1570486882 | 
| user output | 
|---|
| x | 
Test 35
Group: 3
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 2 318961563 84011941 882177798 1... | 
| correct output | 
|---|
| 999999 | 
| user output | 
|---|
| x | 
Test 36
Group: 3
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 2 1 1 2 1 2 1 2 1 2 2 2 2 1 2 1 ... | 
| correct output | 
|---|
| 483793169 | 
| user output | 
|---|
| x | 
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 | 
|---|
| x | 
Test 38
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 1 709486749 350496125 796065873 ... | 
| correct output | 
|---|
| 500000 | 
| user output | 
|---|
| x | 
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 | 
|---|
| x | 
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 | 
|---|
| x | 
Test 41
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 2 3 3 3 3 2 1 3 2 1 3 1 2 1 1 1 ... | 
| correct output | 
|---|
| 2255047 | 
| user output | 
|---|
| x | 
Test 42
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 2 414942284 362802746 108881396 ... | 
| correct output | 
|---|
| 999999 | 
| user output | 
|---|
| x | 
Test 43
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 100 59 14 100 74 49 43 91 84 31 16... | 
| correct output | 
|---|
| 260458849 | 
| user output | 
|---|
| x | 
Test 44
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 1000 505 511 86 321 780 495 106 330... | 
| correct output | 
|---|
| 3694803834 | 
| user output | 
|---|
| x | 
Test 45
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 10000 4956 8463 8582 980 9278 1747 2... | 
| correct output | 
|---|
| 41655235436 | 
| user output | 
|---|
| x | 
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 | 
|---|
| x | 
Test 47
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 100000 40198 78364 3724 51802 378 130... | 
| correct output | 
|---|
| 125000250000 | 
| user output | 
|---|
| x | 
Test 48
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 100000 320513086 811766509 339605137 ... | 
| correct output | 
|---|
| 45001856854 | 
| user output | 
|---|
| x | 
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 | 
|---|
| x | 
Test 50
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 499999 995132060 587162982 59723733 8... | 
| correct output | 
|---|
| 125000250000 | 
| user output | 
|---|
| x | 
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 | 
|---|
| x | 
Test 52
Group: 4
Verdict: WRONG ANSWER
| input | 
|---|
| 500000 500000 72083718 753463162 730560260 6... | 
| correct output | 
|---|
| 125000250000 | 
| user output | 
|---|
| x | 
