CSES - Datatähti 2021 alku - Results
Submission details
Task:Alitaulukot
Sender:Microwave Abuser
Submission time:2020-09-30 17:53:24 +0300
Language:CPython3
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
Test results
testverdicttimegroup
#1ACCEPTED0.02 s1, 2, 3details
#20.02 s1, 2, 3details
#3ACCEPTED0.02 s1, 2, 3details
#40.02 s1, 2, 3details
#5ACCEPTED0.02 s1, 2, 3details
#6ACCEPTED0.03 s2, 3details
#70.03 s2, 3details
#80.03 s2, 3details
#90.03 s2, 3details
#10ACCEPTED0.03 s2, 3details
#11ACCEPTED0.11 s3details
#120.20 s3details
#130.19 s3details
#140.12 s3details
#15ACCEPTED0.12 s3details
#160.22 s3details
#170.22 s3details

Code

def muodostaYlitaulukot(gayArray, n, k):
    ylitaulukkoLista = []
    tPointer = 0
    ePointer = 1
    lastEPointer = -1
    while ePointer < n: #niin kauan kuin taulukkoa ei olla käyty loppuun
        ylitaulukko = {}
        ylitaulukko["alku"] = tPointer
        ylitaulukko["loppu"] = tPointer
        ylitaulukko["maxVal"] = gayArray[tPointer]
        ylitaulukko["minVal"] = gayArray[tPointer]

        while (gayArray[ePointer] <= ylitaulukko["minVal"] + k) and (gayArray[ePointer] >= ylitaulukko["maxVal"] - k): #niin kauan kun tuleva lisäys on laillinen

            ylitaulukko["maxVal"] = max(ylitaulukko["maxVal"], gayArray[ePointer])  #update records
            ylitaulukko["minVal"] = min(ylitaulukko["minVal"], gayArray[ePointer])
            ePointer += 1
            if ePointer == n:
                break

        # LISÄYS EI OLE ENÄÄN LAILLINEN
        # päätepiste eri meinaa uutta taulukkoa:
        if ePointer != lastEPointer:
            ylitaulukko["loppu"] = ePointer - 1
            ylitaulukkoLista.append(ylitaulukko)
            lastEPointer = ePointer

        tPointer += 1
        ePointer = max(ePointer, tPointer + 1)

    if ylitaulukkoLista[len(ylitaulukkoLista) - 1]["loppu"] != n - 1:
        ylitaulukko = {}
        ylitaulukko["alku"] = tPointer
        ylitaulukko["loppu"] = tPointer
        ylitaulukkoLista.append(ylitaulukko)

    return ylitaulukkoLista



def laskeAlitaulukot(ylitaulukkoLista, n):  #[{alku: x, loppu: y}, {}]
    result = 0
    lastEnd = -1
    for ylitaulukko in ylitaulukkoLista:
        pituus = ylitaulukko["loppu"] - ylitaulukko["alku"] + 1
        result += ((pituus**2) + pituus)/2
        if lastEnd >= ylitaulukko["alku"]: #jos on toisen taulokon päällä niin vähennä overlap
            pituus = ylitaulukko["alku"] - lastEnd + 1
            result -= ((pituus**2) + pituus)/2
        lastEnd = ylitaulukko["loppu"]

    return result

def main():
    n, k = [int(i) for i in input().split(" ")]
    gayArray = [int(i) for i in input().split(" ")]
    print(int(laskeAlitaulukot(muodostaYlitaulukot(gayArray, n, k), n)))

main()

Test details

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

input
100 0
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, 2, 3

Verdict:

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

correct output
317

user output
4017

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

input
100 10
71 60 61 96 25 10 10 9 84 85 1...

correct output
119

user output
119

Test 4

Group: 1, 2, 3

Verdict:

input
100 990000000
111122929 961821360 578238211 ...

correct output
4006

user output
5475

Test 5

Group: 1, 2, 3

Verdict: ACCEPTED

input
100 1000000000
553190572 453407680 667300705 ...

correct output
5050

user output
5050

Test 6

Group: 2, 3

Verdict: ACCEPTED

input
2000 0
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
2001000

user output
2001000

Test 7

Group: 2, 3

Verdict:

input
2000 2
4 4 1 4 2 3 1 2 1 3 5 2 2 4 4 ...

correct output
6340

user output
1942376

Test 8

Group: 2, 3

Verdict:

input
2000 10
65 88 33 88 41 10 17 38 22 3 8...

correct output
2413

user output
2440

Test 9

Group: 2, 3

Verdict:

input
2000 999000000
746120950 772769620 721488968 ...

correct output
1287776

user output
2005377

Test 10

Group: 2, 3

Verdict: ACCEPTED

input
2000 1000000000
621947980 510355354 756705418 ...

correct output
2001000

user output
2001000

Test 11

Group: 3

Verdict: ACCEPTED

input
100000 0
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
5000050000

user output
5000050000

Test 12

Group: 3

Verdict:

input
100000 2
3 3 1 3 3 1 1 5 1 2 5 4 1 3 1 ...

correct output
317066

user output
4610360282

Test 13

Group: 3

Verdict:

input
100000 10
10 3 6 3 43 60 5 48 15 27 86 4...

correct output
123292

user output
124898

Test 14

Group: 3

Verdict:

input
100000 999990000
460235639 963048588 47270983 3...

correct output
4946886742

user output
5000347590

Test 15

Group: 3

Verdict: ACCEPTED

input
100000 1000000000
885457070 18257718 927615960 3...

correct output
5000050000

user output
5000050000

Test 16

Group: 3

Verdict:

input
100000 50000
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

correct output
3750075000

user output
11249775003

Test 17

Group: 3

Verdict:

input
100000 50000
100000 99999 99998 99997 99996...

correct output
3750075000

user output
11249775003