CSES - Datatähti 2019 alku - Results
Submission details
Task:Taulukko
Sender:N00B.exe
Submission time:2018-10-14 22:16:34 +0300
Language:C++
Status:READY
Result:65
Feedback
groupverdictscore
#1ACCEPTED12
#2ACCEPTED35
#3ACCEPTED18
#40
Test results
testverdicttimegroup
#1ACCEPTED0.01 s1details
#2ACCEPTED0.02 s1details
#3ACCEPTED0.02 s1details
#4ACCEPTED0.02 s1details
#5ACCEPTED0.02 s1details
#6ACCEPTED0.02 s1details
#7ACCEPTED0.01 s1details
#8ACCEPTED0.03 s1details
#9ACCEPTED0.03 s1details
#10ACCEPTED0.02 s1details
#11ACCEPTED0.02 s1details
#12ACCEPTED0.02 s1details
#13ACCEPTED0.01 s1details
#14ACCEPTED0.02 s1details
#15ACCEPTED0.02 s2details
#16ACCEPTED0.01 s2details
#17ACCEPTED0.02 s2details
#18ACCEPTED0.01 s2details
#19ACCEPTED0.01 s2details
#20ACCEPTED0.01 s2details
#21ACCEPTED0.02 s2details
#22ACCEPTED0.02 s2details
#23ACCEPTED0.02 s2details
#24ACCEPTED0.03 s2details
#25ACCEPTED0.02 s2details
#26ACCEPTED0.02 s2details
#27ACCEPTED0.02 s2details
#28ACCEPTED0.02 s2details
#29ACCEPTED0.07 s3details
#30ACCEPTED0.93 s3details
#31ACCEPTED0.08 s3details
#32ACCEPTED0.08 s3details
#33ACCEPTED0.08 s3details
#34ACCEPTED0.08 s3details
#35ACCEPTED0.94 s3details
#36ACCEPTED0.10 s3details
#37ACCEPTED0.07 s4details
#38ACCEPTED0.94 s4details
#39ACCEPTED0.07 s4details
#40ACCEPTED0.06 s4details
#41ACCEPTED0.09 s4details
#42ACCEPTED0.94 s4details
#43ACCEPTED0.15 s4details
#44ACCEPTED0.22 s4details
#45ACCEPTED0.34 s4details
#46ACCEPTED0.07 s4details
#47ACCEPTED0.14 s4details
#48--4details
#49ACCEPTED0.07 s4details
#50ACCEPTED0.31 s4details
#51ACCEPTED0.06 s4details
#52ACCEPTED0.30 s4details

Compiler report

input/code.cpp: In function 'int main()':
input/code.cpp:42: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(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;

	//ll waste = 0;
	bool wasteFlag = false;
	int endPrev;

	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)
	{
		//LBRK;
		//PRT("remove number "); PRTI(*(nums + start));
		//LBRK;
		m[*startP] -= 1;
		if (m[*startP] == 0)
		{
			dNumbers--;
			wasteFlag = false;
		}

		start++;
		if (start >= n - k)
		{
			break;
		}
		startP++;

		if(!wasteFlag)
		{
			endPrev = end;
			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;
			}


			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;
			}
		}

		if(!wasteFlag && endPrev == end)
		{ wasteFlag = true; }

		ans += (ll)(end - start);

		//SITUATION;
	}

	ll lk = k;
	ans += lk * ((lk + 1) / 2.0);
	printf("%lld", ans);

	free(nums);
	/*
	LBRK;
	PRT("Wasted move "); printf("%lld", waste);
	PAUSE;
	*/

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

input
500000 1
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
125000250000

user output
125000250000

Test 30

Group: 3

Verdict: ACCEPTED

input
500000 1
28107328 613212742 298033960 7...

correct output
500000

user output
500000

Test 31

Group: 3

Verdict: ACCEPTED

input
500000 2
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
125000250000

user output
125000250000

Test 32

Group: 3

Verdict: ACCEPTED

input
500000 2
1 1 2 1 1 2 2 1 2 1 2 1 1 2 1 ...

correct output
125000250000

user output
125000250000

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

input
500000 1
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
125000250000

user output
125000250000

Test 38

Group: 4

Verdict: ACCEPTED

input
500000 1
709486749 350496125 796065873 ...

correct output
500000

user output
500000

Test 39

Group: 4

Verdict: ACCEPTED

input
500000 2
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
125000250000

user output
125000250000

Test 40

Group: 4

Verdict: ACCEPTED

input
500000 2
1 2 2 2 2 2 1 1 2 2 2 2 2 2 1 ...

correct output
125000250000

user output
125000250000

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

input
500000 10000
4956 8463 8582 980 9278 1747 2...

correct output
41655235436

user output
41655235436

Test 46

Group: 4

Verdict: ACCEPTED

input
500000 100000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
125000250000

user output
125000250000

Test 47

Group: 4

Verdict: ACCEPTED

input
500000 100000
40198 78364 3724 51802 378 130...

correct output
125000250000

user output
125000250000

Test 48

Group: 4

Verdict:

input
500000 100000
320513086 811766509 339605137 ...

correct output
45001856854

user output
(empty)

Test 49

Group: 4

Verdict: ACCEPTED

input
500000 499999
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
125000250000

user output
125000250000

Test 50

Group: 4

Verdict: ACCEPTED

input
500000 499999
995132060 587162982 59723733 8...

correct output
125000250000

user output
125000250000

Test 51

Group: 4

Verdict: ACCEPTED

input
500000 500000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
125000250000

user output
125000250000

Test 52

Group: 4

Verdict: ACCEPTED

input
500000 500000
72083718 753463162 730560260 6...

correct output
125000250000

user output
125000250000