CSES - Datatähti 2022 alku - Results
Submission details
Task:Spiraali
Sender:Kaapipo
Submission time:2021-10-04 13:15:35 +0300
Language:CPython3
Status:READY
Result:100
Feedback
groupverdictscore
#1ACCEPTED15
#2ACCEPTED20
#3ACCEPTED65
Test results
testverdicttimegroup
#1ACCEPTED0.02 s1details
#2ACCEPTED0.03 s2details
#3ACCEPTED0.03 s3details

Code

def a(k, n):
	return n ** 2 - (n - 2 * k) ** 2 + 1



def distance_from_edge(x, y, n):
	return min(
		x,
		y,
		n - x - 1,
		n - y - 1
	)

def number_at(x, y, n):
	d = distance_from_edge(x, y, n)
	nn = a(d, n)
	
	if x == y == d:
		return nn
	
	bottom_left = nn + (n - 2 * d) - 1
	if x == d and nn < bottom_left:
		return nn + (y - d)
		
	bottom_right = bottom_left + (n - 2 * d) - 1
	if n - y - 1 == d and nn < bottom_right:
		return bottom_left + (x - d)
		
	top_right = bottom_right + (n - 2 * d) - 1
	if n - x - 1 == d and nn < top_right:
		return bottom_right + (n - d) - y - 1

	top_left = top_right + (n - 2 * d) - 2
	if y == d and nn < top_left:
		return top_right + (n - d) - x - 1
	

n, t = map(int, input().split())

for i in range(t):
	y, x = map(int, input().split())
	print(number_at(x - 1, y - 1, n))

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

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

correct output
571375684522141210
967321186816598569
762879105851175000
370065046779516790
936897883750373771
...

user output
571375684522141210
967321186816598569
762879105851175000
370065046779516790
936897883750373771
...