Submission #6320438

Source Code Expand

import sys
input = sys.stdin.readline
import numpy as np
from scipy.sparse.csgraph import csgraph_from_dense, dijkstra
from scipy.spatial.distance import cdist

xs,ys,xt,yt = map(int,input().split())
N = int(input())
XYR = np.empty((N+2,3),dtype=np.int64)
XYR[:2,] = [[xs,ys,0], [xt,yt,0]]

XYR[2:] = np.array([input().split() for _ in range(N)], dtype=np.int64)

XY = XYR[:,:2]
dist = cdist(XY, XY)
R = XYR[:,2]
dist -= R[:,None] + R[None,:]
np.maximum(dist, 0, out = dist)
dist[np.arange(N+2), np.arange(N+2)] = np.inf

graph = csgraph_from_dense(dist, null_value = np.inf)

answer = dijkstra(graph, indices = 0)[1]

print(answer)

Submission Info

Submission Time
Task E - Cosmic Rays
User maspy
Language Python (3.4.3)
Score 600
Code Size 658 Byte
Status AC
Exec Time 245 ms
Memory 52456 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 600 / 600
Status
AC × 3
AC × 49
Set Name Test Cases
Sample 0_00.txt, 0_01.txt, 0_02.txt
All 0_00.txt, 0_01.txt, 0_02.txt, 1_00.txt, 1_01.txt, 1_02.txt, 1_03.txt, 1_04.txt, 1_05.txt, 1_06.txt, 1_07.txt, 1_08.txt, 1_09.txt, 1_10.txt, 1_11.txt, 1_12.txt, 1_13.txt, 1_14.txt, 1_15.txt, 1_16.txt, 1_17.txt, 1_18.txt, 1_19.txt, 1_20.txt, 1_21.txt, 1_22.txt, 1_23.txt, 1_24.txt, 1_25.txt, 1_26.txt, 1_27.txt, 1_28.txt, 1_29.txt, 1_30.txt, 1_31.txt, 1_32.txt, 1_33.txt, 1_34.txt, 1_35.txt, 1_36.txt, 1_37.txt, 1_38.txt, 1_39.txt, 1_40.txt, 1_41.txt, 1_42.txt, 1_43.txt, 1_44.txt, 1_45.txt
Case Name Status Exec Time Memory
0_00.txt AC 207 ms 16072 KiB
0_01.txt AC 182 ms 14812 KiB
0_02.txt AC 180 ms 14824 KiB
1_00.txt AC 180 ms 14816 KiB
1_01.txt AC 181 ms 14824 KiB
1_02.txt AC 238 ms 51048 KiB
1_03.txt AC 240 ms 51696 KiB
1_04.txt AC 239 ms 51580 KiB
1_05.txt AC 238 ms 51904 KiB
1_06.txt AC 240 ms 51164 KiB
1_07.txt AC 239 ms 51376 KiB
1_08.txt AC 240 ms 51688 KiB
1_09.txt AC 240 ms 52004 KiB
1_10.txt AC 242 ms 52312 KiB
1_11.txt AC 242 ms 51380 KiB
1_12.txt AC 242 ms 51268 KiB
1_13.txt AC 245 ms 52216 KiB
1_14.txt AC 241 ms 52444 KiB
1_15.txt AC 242 ms 52444 KiB
1_16.txt AC 241 ms 52324 KiB
1_17.txt AC 240 ms 51900 KiB
1_18.txt AC 240 ms 51604 KiB
1_19.txt AC 239 ms 52440 KiB
1_20.txt AC 241 ms 52016 KiB
1_21.txt AC 241 ms 51056 KiB
1_22.txt AC 240 ms 51896 KiB
1_23.txt AC 240 ms 51892 KiB
1_24.txt AC 242 ms 52456 KiB
1_25.txt AC 240 ms 51692 KiB
1_26.txt AC 240 ms 51804 KiB
1_27.txt AC 239 ms 50960 KiB
1_28.txt AC 240 ms 52116 KiB
1_29.txt AC 240 ms 52196 KiB
1_30.txt AC 239 ms 51692 KiB
1_31.txt AC 239 ms 52012 KiB
1_32.txt AC 239 ms 52324 KiB
1_33.txt AC 240 ms 51284 KiB
1_34.txt AC 236 ms 51036 KiB
1_35.txt AC 240 ms 52408 KiB
1_36.txt AC 238 ms 52184 KiB
1_37.txt AC 239 ms 51676 KiB
1_38.txt AC 242 ms 52396 KiB
1_39.txt AC 242 ms 52284 KiB
1_40.txt AC 240 ms 51548 KiB
1_41.txt AC 242 ms 52280 KiB
1_42.txt AC 237 ms 50992 KiB
1_43.txt AC 240 ms 52372 KiB
1_44.txt AC 240 ms 52252 KiB
1_45.txt AC 239 ms 52060 KiB