CSES - Datatähti 2022 alku - Results
Submission details
Task:Spiraali (Spiral)
Sender:parasdata
Submission time:2021-10-05 18:10:10
Language:CPython3
Status:READY
Result:35
Feedback
groupverdictscore
#1ACCEPTED15
#2ACCEPTED20
#30
Test results
testverdicttimegroup
#1ACCEPTED0.02 s1details
#2ACCEPTED0.67 s2details
#3--3details

Code

def get_input():
    n, t = input().split()
    return int(n), int(t)


def ulko_kerrokset(n, x, y):
    if n-x >= n/2:
        kerrokset1 = x - 1
    else:
        kerrokset1 = n - x
    if n-y >= n/2:
        kerrokset2 = y - 1
    else:
        kerrokset2 = n - y
    return min(kerrokset1, kerrokset2)


n, t = get_input()
vastaus = ""

while t > 0:
    x, y = get_input()

    kerrokset = ulko_kerrokset(n, x, y)
    luku = n**2 - ((n - kerrokset*2) ** 2)  # suurin luku edellisessä kerroksessa
    kerros = []
    i = (n - kerrokset*2 - 1) * 4
    while i > 0:
        luku += 1
        kerros.append(luku)
        i -= 1
    if y == kerrokset + 1:
        luku = kerros[(x-kerrokset-1)]
    elif x == n - kerrokset:
        luku = kerros[((n-kerrokset*2) + (y-kerrokset) - 2)]
    elif y == n - kerrokset:
        luku = kerros[((n-kerrokset*2)*2-1 + (n-kerrokset-x)-1)]
    else:
        luku = kerros[((n-kerrokset*2)*3-2 + (n-kerrokset-y)-1)]

    vastaus += str(luku) + "\n"
    t -= 1

print(vastaus)

Test details

Test 1

Group: 1

Verdict: ACCEPTED

input
10 100
1 1
1 2
1 3
1 4
...

correct output
1
36
35
34
33
...

user output
1
36
35
34
33
...

Test 2

Group: 2

Verdict: ACCEPTED

input
1000 1000
371 263
915 322
946 880
53 738
...

correct output
773533
312166
206053
200080
593922
...

user output
773533
312166
206053
200080
593922
...

Test 3

Group: 3

Verdict:

input
1000000000 1000
177757853 827347032
409613589 419171337
739269360 256524697
328695530 896842209
...

correct output
571375684522141210
967321186816598569
762879105851175000
370065046779516790
936897883750373771
...

user output
(empty)