CSES - Datatähti 2019 alku - Results
Submission details
Task:Taulukko
Sender:N00B.exe
Submission time:2018-10-12 19:01:39 +0300
Language:C++
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
#40
Test results
testverdicttimegroup
#1ACCEPTED0.03 s1details
#20.02 s1details
#30.01 s1details
#4ACCEPTED0.02 s1details
#5ACCEPTED0.01 s1details
#60.01 s1details
#70.03 s1details
#8ACCEPTED0.01 s1details
#9ACCEPTED0.01 s1details
#100.01 s1details
#110.01 s1details
#12ACCEPTED0.01 s1details
#13ACCEPTED0.01 s1details
#140.02 s1details
#150.40 s2details
#160.53 s2details
#17--2details
#18ACCEPTED0.01 s2details
#190.57 s2details
#200.61 s2details
#21--2details
#22--2details
#23--2details
#24--2details
#25ACCEPTED0.03 s2details
#26ACCEPTED0.03 s2details
#270.03 s2details
#280.02 s2details
#29--3details
#30--3details
#310.07 s3details
#32--3details
#33--3details
#34--3details
#35--3details
#36--3details
#37--4details
#38--4details
#390.07 s4details
#40--4details
#41--4details
#42--4details
#43--4details
#44--4details
#45--4details
#460.06 s4details
#470.32 s4details
#48--4details
#490.05 s4details
#500.55 s4details
#510.05 s4details
#520.09 s4details

Compiler report

input/code.cpp: In function 'int main()':
input/code.cpp:30:35: warning: format '%d' expects argument of type 'int', but argument 2 has type 'double' [-Wformat=]
   printf("%d", (n*((n + 1.0) / 2)));
                ~~~~~~~~~~~~~~~~~~~^
input/code.cpp:12:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d %d", &n, &k);
  ~~~~~^~~~~~~~~~~~~~~~~

Code

#include <iostream>
#include <map>

using namespace std;

int main() {

	ios_base::sync_with_stdio(0);
	cin.tie(0);

	int n, k, ans = 0;
	scanf("%d %d", &n, &k);

	map<int, int> m;


	int* numbers = (int*)malloc(4 * n);

	for(int i = 0; i < n; i++)
	{
		cin >> *(numbers + i);
		//scanf("%d", (numbers + i));
	}

	if (k == 0)
	{ printf("0"); return 0; }

	if(k >= n)
	{
		printf("%d", (n*((n + 1.0) / 2)));
		return 0;
	}

	int dAmount = 0;

	for(int i = 0; i < n; i++)
	{
		int index = *(numbers + i);

		if(m[index] == 0)
		{ m[index] = 1; dAmount++; }

		else
		{ m[index] += 1; }
	}

	if (dAmount <= k)
	{ ans = n * ((n + 1.0) / 2); }

	else
	{
		int start = 0;
		int end = n - 1;
		bool goRight = false;
		int growIndex, shrinkIndex;
		int a, b;
		int currentDNumbers = dAmount;
		int d = currentDNumbers - k;
		int min = 2147483647;

		if(d <= end - start - d)
		{
			for(int k = 0; k < d; k++)
			{
				m[*(numbers + start)] -= 1;
				
				if(m[*(numbers + start)] == 0)
				{ currentDNumbers--; }

				start++;
			}
		}

		else
		{
			start += d;
			m.clear();
			currentDNumbers = 0;

			for(int k = 0; k < (end-start + 1); k++)
			{
				if(m[*(numbers + start + k)] == 0)
				{ m[*(numbers + start + k)] = 1; currentDNumbers++; }

				else {
					m[*(numbers + start + k)] += 1;
				}
			}
		}

		while(end - start > (k - 1))
		{
			growIndex = (goRight) ? end + 1 : start - 1;

			if (growIndex > n - 1)
			{
				if (end - start == k)
				{ break; }

				d = min - k;

				if (d <= 1)
				{
					d = 1;
					m[*(numbers + start)] -= 1;

					if (m[*(numbers + start)] == 0)
					{ currentDNumbers--; }

					if (currentDNumbers <= k)
					{ ans++; }

					start++;
					goRight = !goRight;
				}

				else if (d <= end - start - d)
				{
					for (int k = 0; k < d; k++)
					{
						m[*(numbers + start)] -= 1;

						if (m[*(numbers + start)] == 0)
						{ currentDNumbers--; }

						start++;
					}
					goRight = !goRight;
				}

				else
				{
					start += d;
					m.clear();
					currentDNumbers = 0;

					for (int k = 0; k < (end - start + 1); k++)
					{
						if (m[*(numbers + start + k)] == 0)
						{ m[*(numbers + start + k)] = 1; currentDNumbers++; }

						else
						{ m[*(numbers + start + k)] += 1; }
					}

					goRight = !goRight;
				}

				if (currentDNumbers < min)
				{ min = currentDNumbers; }
			}

			else if(growIndex < 0)
			{
				if (end - start == k)
				{ break; }

				d = min - k;

				if (d <= 1)
				{
					d = 1;
					m[*(numbers + end)] -= 1;
					
					if (m[*(numbers + end)] == 0)
					{ currentDNumbers--; }
					
					if (currentDNumbers <= k)
					{ ans++; }

					end--;
					goRight = !goRight;
					}

					else if (d <= end - start - d)
					{
						for (int k = 0; k < d; k++)
						{
							m[*(numbers + end)] -= 1;

							if (m[*(numbers + end)] == 0)
							{ currentDNumbers--; }

							end--;
						}
						goRight = !goRight;
					}

					else
					{
						end -= d;
						m.clear();
						currentDNumbers = 0;

						for (int k = end - start; k > -1; k++)
						{
							if (m[*(numbers + end + k)] == 0)
							{ m[*(numbers + end + k)] = 1; currentDNumbers++; }

							else
							{ m[*(numbers + end + k)] += 1; }
						}

						goRight = !goRight;
					}

				}

			else
			{
				shrinkIndex = (goRight) ? start : end;

				a = m[*(numbers + shrinkIndex)];
				b = m[*(numbers + growIndex)];

				m[*(numbers + shrinkIndex)] -= 1;
				m[*(numbers + growIndex)] += 1;

				if (m[*(numbers + shrinkIndex)] == 0 && m[*(numbers + shrinkIndex)] != a)
				{ currentDNumbers--; }

				if (m[*(numbers + growIndex)] == 1 && m[*(numbers + growIndex)] != b)
				{ currentDNumbers++; }

				if (currentDNumbers <= k)
				{ ans++; }

				start += (goRight) ? 1 : -1;
				end += (goRight) ? 1 : -1;

				if(currentDNumbers < min)
				{ min = currentDNumbers; }
			}

		}
		ans += ((2 * n + 1)*k - k * k) / 2;
	}

	printf("%d", ans);
	m.clear();
	free(numbers);

	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:

input
100 1
1 1 1 1 2 2 1 1 2 2 2 2 2 1 1 ...

correct output
190

user output
5050

Test 3

Group: 1

Verdict:

input
100 1
5 9 9 6 9 8 1 4 7 7 8 9 5 5 6 ...

correct output
110

user output
5050

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:

input
100 2
2 3 1 1 2 2 3 1 2 1 1 1 3 3 1 ...

correct output
379

user output
5050

Test 7

Group: 1

Verdict:

input
100 2
4 6 10 8 6 8 10 8 4 7 8 9 6 2 ...

correct output
245

user output
5050

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:

input
100 5
5 3 4 1 1 1 1 4 5 5 4 6 6 3 3 ...

correct output
1488

user output
5050

Test 11

Group: 1

Verdict:

input
100 5
10 1 1 9 1 6 9 4 3 10 9 2 4 2 ...

correct output
743

user output
5050

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:

input
100 10
80 59 58 87 28 83 83 93 96 24 ...

correct output
994

user output
5050

Test 15

Group: 2

Verdict:

input
5000 1
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
12502500

user output
6459525

Test 16

Group: 2

Verdict:

input
5000 1
2 2 2 1 2 1 1 1 2 2 1 2 1 1 1 ...

correct output
10148

user output
2097958

Test 17

Group: 2

Verdict:

input
5000 1
434568 634119 102509 107238 72...

correct output
5000

user output
(empty)

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:

input
5000 2
2 1 1 1 2 2 2 2 1 1 1 1 2 1 2 ...

correct output
12502500

user output
6461570

Test 20

Group: 2

Verdict:

input
5000 2
1 3 2 2 3 3 1 2 1 2 1 2 3 1 2 ...

correct output
22451

user output
2109595

Test 21

Group: 2

Verdict:

input
5000 2
132968 414421 468358 432744 43...

correct output
9999

user output
(empty)

Test 22

Group: 2

Verdict:

input
5000 100
93 76 87 71 93 74 69 35 92 96 ...

correct output
12502500

user output
(empty)

Test 23

Group: 2

Verdict:

input
5000 100
2 57 51 4 35 76 5 40 51 55 20 ...

correct output
2669737

user output
(empty)

Test 24

Group: 2

Verdict:

input
5000 100
657 823 1773 1801 2785 4107 25...

correct output
498978

user output
(empty)

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:

input
5000 5000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
12502500

user output
5000

Test 28

Group: 2

Verdict:

input
5000 5000
774752 159472 183796 654476 69...

correct output
12502500

user output
5000

Test 29

Group: 3

Verdict:

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

correct output
125000250000

user output
(empty)

Test 30

Group: 3

Verdict:

input
500000 1
28107328 613212742 298033960 7...

correct output
500000

user output
(empty)

Test 31

Group: 3

Verdict:

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

correct output
125000250000

user output
-2147483648

Test 32

Group: 3

Verdict:

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

correct output
125000250000

user output
(empty)

Test 33

Group: 3

Verdict:

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

correct output
2245355

user output
(empty)

Test 34

Group: 3

Verdict:

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

correct output
1570486882

user output
(empty)

Test 35

Group: 3

Verdict:

input
500000 2
318961563 84011941 882177798 1...

correct output
999999

user output
(empty)

Test 36

Group: 3

Verdict:

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

correct output
483793169

user output
(empty)

Test 37

Group: 4

Verdict:

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

correct output
125000250000

user output
(empty)

Test 38

Group: 4

Verdict:

input
500000 1
709486749 350496125 796065873 ...

correct output
500000

user output
(empty)

Test 39

Group: 4

Verdict:

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

correct output
125000250000

user output
-2147483648

Test 40

Group: 4

Verdict:

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

correct output
125000250000

user output
(empty)

Test 41

Group: 4

Verdict:

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

correct output
2255047

user output
(empty)

Test 42

Group: 4

Verdict:

input
500000 2
414942284 362802746 108881396 ...

correct output
999999

user output
(empty)

Test 43

Group: 4

Verdict:

input
500000 100
59 14 100 74 49 43 91 84 31 16...

correct output
260458849

user output
(empty)

Test 44

Group: 4

Verdict:

input
500000 1000
505 511 86 321 780 495 106 330...

correct output
3694803834

user output
(empty)

Test 45

Group: 4

Verdict:

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

correct output
41655235436

user output
(empty)

Test 46

Group: 4

Verdict:

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

correct output
125000250000

user output
-2147483648

Test 47

Group: 4

Verdict:

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

correct output
125000250000

user output
-2147483648

Test 48

Group: 4

Verdict:

input
500000 100000
320513086 811766509 339605137 ...

correct output
45001856854

user output
(empty)

Test 49

Group: 4

Verdict:

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

correct output
125000250000

user output
-2147483648

Test 50

Group: 4

Verdict:

input
500000 499999
995132060 587162982 59723733 8...

correct output
125000250000

user output
-2147483648

Test 51

Group: 4

Verdict:

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

correct output
125000250000

user output
500000

Test 52

Group: 4

Verdict:

input
500000 500000
72083718 753463162 730560260 6...

correct output
125000250000

user output
500000