CSES - NOI 2024 - Results
Submission details
Task:Anime Shops
Sender:Tine Dahlberg
Submission time:2024-03-06 16:21:41 +0200
Language:Python3 (CPython3)
Status:READY
Result:23
Feedback
groupverdictscore
#1ACCEPTED23
#20
#30
Test results
testverdicttimegroup
#1ACCEPTED0.03 s1, 3details
#2ACCEPTED0.06 s1, 3details
#3ACCEPTED0.23 s1, 3details
#4ACCEPTED0.05 s1, 3details
#50.72 s3details
#60.78 s3details
#70.93 s3details
#8--3details
#90.38 s2, 3details
#100.44 s2, 3details
#11--2, 3details
#120.38 s3details
#13--3details
#14--3details
#15--3details
#160.38 s3details

Code

n,m,k = map(int,input().split())
shops = list(map(int,input().split()))
adj = [[] for i in range(n+1)]
for i in range(m):
  a,b = map(int,input().split())
  adj[a].append(b)
  adj[b].append(a)

if n<=1000:

  ans = [-1]*n
  
  for start in shops:
    kö = [start]
    dist = [-1]*n
    dist[start-1]=0
    for curr in kö:
      for granne in adj[curr]:
        if dist[granne-1]!=-1:
          continue
        dist[granne-1]=dist[curr-1]+1
        if ans[granne-1]==-1:
          ans[granne-1]=dist[granne-1]
        elif dist[granne-1]<ans[granne-1]:
          ans[granne-1]=dist[granne-1]
        if granne not in shops:
          kö.append(granne)
  
  print(*ans)

else:
  ans = [1000000]*n
  senaste = -1000000
  for i in range(1,n+1):
    if i-senaste<ans[i-1]:
      ans[i-1]=i-senaste
    if i in shops:
      senaste = i

  senaste = 10000000
  for i in reversed(range(1,n+1)):
    if senaste-i<ans[i-1]:
      ans[i-1]=senaste-i
    if i in shops:
      senaste = i

  for i in range(n):
    if ans[n]==1000000:
      ans[n]=-1

  print(*ans)

Test details

Test 1

Group: 1, 3

Verdict: ACCEPTED

input
1000 2000 1
816
1 868
976 995
377 536
...

correct output
4 3 3 4 6 4 -1 4 5 2 2 5 5 3 5...

user output
4 3 3 4 6 4 -1 4 5 2 2 5 5 3 5...
Truncated

Test 2

Group: 1, 3

Verdict: ACCEPTED

input
1000 2000 20
578 955 222 813 494 962 753 71...

correct output
5 6 4 3 4 2 -1 3 3 3 4 3 2 3 -...

user output
5 6 4 3 4 2 -1 3 3 3 4 3 2 3 -...
Truncated

Test 3

Group: 1, 3

Verdict: ACCEPTED

input
1000 2000 100
945 230 119 680 975 520 490 28...

correct output
2 2 3 -1 2 -1 3 2 2 1 2 -1 3 2...

user output
2 2 3 -1 2 -1 3 2 2 1 2 -1 3 2...
Truncated

Test 4

Group: 1, 3

Verdict: ACCEPTED

input
1000 2000 1000
150 443 960 545 218 487 896 38...

correct output
1 -1 1 1 -1 1 1 1 -1 1 1 1 -1 ...

user output
1 -1 1 1 -1 1 1 1 -1 1 1 1 -1 ...
Truncated

Test 5

Group: 3

Verdict:

input
100000 200000 1
77222
59470 61811
43092 48939
14395 19964
...

correct output
8 10 8 8 8 8 8 8 9 7 7 8 8 8 6...

user output
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 48, in <module>
    i...

Test 6

Group: 3

Verdict:

input
100000 200000 20
70440 82302 64483 96767 51485 ...

correct output
-1 8 6 5 5 7 6 7 6 6 8 5 6 4 5...

user output
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 48, in <module>
    i...

Test 7

Group: 3

Verdict:

input
100000 200000 100
68738 37820 55519 77758 46482 ...

correct output
4 5 5 5 5 4 6 -1 5 4 5 6 4 5 5...

user output
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 48, in <module>
    i...

Test 8

Group: 3

Verdict:

input
100000 200000 100000
47137 80137 73347 78145 9205 6...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
(empty)

Test 9

Group: 2, 3

Verdict:

input
100000 99999 1
44158
1 2
2 3
3 4
...

correct output
44157 44156 44155 44154 44153 ...

user output
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 48, in <module>
    i...

Test 10

Group: 2, 3

Verdict:

input
100000 99999 20
44158 25720 84658 90057 99607 ...

correct output
361 360 359 358 357 356 355 35...

user output
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 48, in <module>
    i...

Test 11

Group: 2, 3

Verdict:

input
100000 99999 100000
44158 25720 84658 90057 99607 ...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
(empty)

Test 12

Group: 3

Verdict:

input
100000 99999 3
99998 99999 100000
1 2
1 3
1 4
...

correct output
33333 33332 33332 33332 33331 ...

user output
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 48, in <module>
    i...

Test 13

Group: 3

Verdict:

input
100000 99999 300
99701 99702 99703 99704 99705 ...

correct output
333 333 333 333 333 333 333 33...

user output
(empty)

Test 14

Group: 3

Verdict:

input
100000 99999 30000
70001 70002 70003 70004 70005 ...

correct output
3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 ...

user output
(empty)

Test 15

Group: 3

Verdict:

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

correct output
-1 -1 -1 -1 -1 -1 -1 -1 -1 -1 ...

user output
(empty)

Test 16

Group: 3

Verdict:

input
100000 100000 2
1 50001
1 2
2 3
3 4
...

correct output
50000 1 2 3 4 5 6 7 8 9 10 11 ...

user output
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 48, in <module>
    i...