CSES - Datatähti 2019 alku - Results
Submission details
Task:Taulukko
Sender:Jaahas
Submission time:2018-10-13 17:10:10 +0300
Language:Java
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
#40
Test results
testverdicttimegroup
#10.14 s1details
#20.15 s1details
#30.14 s1details
#40.15 s1details
#50.15 s1details
#60.14 s1details
#70.15 s1details
#80.14 s1details
#90.14 s1details
#100.15 s1details
#110.14 s1details
#120.14 s1details
#130.15 s1details
#140.14 s1details
#150.17 s2details
#160.19 s2details
#17ACCEPTED0.19 s2details
#180.18 s2details
#190.18 s2details
#200.19 s2details
#210.20 s2details
#220.17 s2details
#230.18 s2details
#240.19 s2details
#250.18 s2details
#260.19 s2details
#270.18 s2details
#280.19 s2details
#290.64 s3details
#30ACCEPTED0.94 s3details
#310.62 s3details
#320.64 s3details
#330.72 s3details
#340.74 s3details
#350.94 s3details
#360.76 s3details
#370.65 s4details
#38ACCEPTED0.93 s4details
#390.63 s4details
#400.64 s4details
#410.72 s4details
#420.94 s4details
#430.70 s4details
#440.74 s4details
#450.83 s4details
#460.65 s4details
#470.75 s4details
#48--4details
#490.63 s4details
#500.94 s4details
#510.65 s4details
#520.92 s4details

Code

//package com.tuomas;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class Taulukko {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] s = br.readLine().split(" ");
        int n = Integer.parseInt(s[0]);
        int k = Integer.parseInt(s[1]);
        int[] taulukko = new int[n];

        s = br.readLine().split(" ");
        for(int i = 0; i < n; i++) {
            taulukko[i] = Integer.parseInt(s[i]);
        }

        int result = 0;

        /*if(k == 1) {
            int i = 0;
            int j = 0;
            while(i < n) {
                while(true) {
                    j++;
                    if(j == n || taulukko[j] != taulukko[i]) {
                        result += sum(1, j - i);
                        break;
                    }

                }
                i = j;
            }
        } else if(k == 2) {
            int i = 0;
            int j = 0;
            int l = 0;
            int prevl;
            while(i < n) {
                prevl = l;
                while(true) {
                    j++;
                    if(j >= n || taulukko[j] != taulukko[i]) {
                        l = j - i;
                        break;
                    }
                }
                result += l * prevl + sum(1, l);
                i = j;
            }
        } else */
            Map<Integer, Integer> map = new HashMap<>();
            int j = 0;
            int jprev;
            boolean addVal = true;
            for (int i = 0; i < n; i++) {
                jprev = j;
                while (j < n) {
                    if (addVal) {
                        if (map.get(taulukko[j]) == null) {
                            map.put(taulukko[j], 1);
                        } else {
                            map.put(taulukko[j], map.get(taulukko[j]) + 1);
                        }
                    }
                    if (map.size() > k) {
                        addVal = false;
                        break;
                    }
                    addVal = true;
                    j++;
                }
                if (map.get(taulukko[i]) != null) {
                    map.put(taulukko[i], map.get(taulukko[i]) - 1);
                    if (map.get(taulukko[i]) == 0) map.remove(taulukko[i]);
                }
                result += (jprev - i) * (j - jprev) + sum(1, j - jprev);
                if (j >= n) {
                    break;
                }
            }
        

        System.out.println(result);
    }

    static int sum(int a, int b) {
        return (a-b + 1) * (a + b) / 2;
    }
}

Test details

Test 1

Group: 1

Verdict:

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

correct output
5050

user output
-4949

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
7

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
90

Test 4

Group: 1

Verdict:

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

correct output
5050

user output
-4949

Test 5

Group: 1

Verdict:

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

correct output
5050

user output
-4949

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
88

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
180

Test 8

Group: 1

Verdict:

input
100 5
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
5050

user output
-4949

Test 9

Group: 1

Verdict:

input
100 5
4 4 2 2 4 3 3 2 3 4 3 5 3 1 5 ...

correct output
5050

user output
-4949

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
368

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
505

Test 12

Group: 1

Verdict:

input
100 10
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
5050

user output
-4949

Test 13

Group: 1

Verdict:

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

correct output
5050

user output
-4949

Test 14

Group: 1

Verdict:

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

correct output
994

user output
861

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
-12497499

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
-148

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:

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

correct output
12502500

user output
-12497499

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
-12497499

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
2600

Test 21

Group: 2

Verdict:

input
5000 2
132968 414421 468358 432744 43...

correct output
9999

user output
9996

Test 22

Group: 2

Verdict:

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

correct output
12502500

user output
-12497499

Test 23

Group: 2

Verdict:

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

correct output
2669737

user output
1602817

Test 24

Group: 2

Verdict:

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

correct output
498978

user output
488615

Test 25

Group: 2

Verdict:

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

correct output
12502500

user output
-12497499

Test 26

Group: 2

Verdict:

input
5000 4999
865706 910619 695192 183574 92...

correct output
12502500

user output
-12497499

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
-12497499

Test 28

Group: 2

Verdict:

input
5000 5000
774752 159472 183796 654476 69...

correct output
12502500

user output
-12497499

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
-445698415

Test 30

Group: 3

Verdict: ACCEPTED

input
500000 1
28107328 613212742 298033960 7...

correct output
500000

user output
500000

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
-445698415

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
-445698415

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
250815

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
-1567225289

Test 35

Group: 3

Verdict:

input
500000 2
318961563 84011941 882177798 1...

correct output
999999

user output
999996

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
-480727151

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
-445698415

Test 38

Group: 4

Verdict: ACCEPTED

input
500000 1
709486749 350496125 796065873 ...

correct output
500000

user output
500000

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
-445698415

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
-445698415

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
244227

Test 42

Group: 4

Verdict:

input
500000 2
414942284 362802746 108881396 ...

correct output
999999

user output
999996

Test 43

Group: 4

Verdict:

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

correct output
260458849

user output
161205996

Test 44

Group: 4

Verdict:

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

correct output
3694803834

user output
-1561564483

Test 45

Group: 4

Verdict:

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

correct output
41655235436

user output
-1783850812

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
-445698415

Test 47

Group: 4

Verdict:

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

correct output
125000250000

user output
-445698415

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
-445698415

Test 50

Group: 4

Verdict:

input
500000 499999
995132060 587162982 59723733 8...

correct output
125000250000

user output
-445698415

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
-445698415

Test 52

Group: 4

Verdict:

input
500000 500000
72083718 753463162 730560260 6...

correct output
125000250000

user output
-445698415