結果
| 問題 |
No.2805 Go to School
|
| コンテスト | |
| ユーザー |
mattu34
|
| 提出日時 | 2024-07-12 21:55:00 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
TLE
|
| 実行時間 | - |
| コード長 | 3,152 bytes |
| コンパイル時間 | 478 ms |
| コンパイル使用メモリ | 82,860 KB |
| 実行使用メモリ | 168,284 KB |
| 最終ジャッジ日時 | 2024-07-12 21:55:29 |
| 合計ジャッジ時間 | 25,818 ms |
|
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 4 |
| other | AC * 32 TLE * 2 |
ソースコード
from collections import *
import sys
import heapq
from heapq import heapify, heappop, heappush
import bisect
import itertools
from functools import lru_cache
from types import GeneratorType
from fractions import Fraction
import math
import copy
import random
# import numpy as np
# sys.setrecursionlimit(int(1e7))
# @lru_cache(maxsize=None) # CPython特化
# @bootstrap # PyPy特化(こっちのほうが速い) yield dfs(), yield Noneを忘れずに
def bootstrap(f, stack=[]): # yield
def wrappedfunc(*args, **kwargs):
if stack:
return f(*args, **kwargs)
else:
to = f(*args, **kwargs)
while True:
if type(to) is GeneratorType:
stack.append(to)
to = next(to)
else:
stack.pop()
if not stack:
break
to = stack[-1].send(to)
return to
return wrappedfunc
dxdy1 = ((0, 1), (0, -1), (1, 0), (-1, 0)) # 上下右左
dxdy2 = (
(0, 1),
(0, -1),
(1, 0),
(-1, 0),
(1, 1),
(-1, -1),
(1, -1),
(-1, 1),
) # 8方向すべて
dxdy3 = ((0, 1), (1, 0)) # 右 or 下
dxdy4 = ((1, 1), (1, -1), (-1, 1), (-1, -1)) # 斜め
INF = float("inf")
_INF = 1 << 60
MOD = 998244353
mod = 998244353
MOD2 = 10**9 + 7
mod2 = 10**9 + 7
# memo : len([a,b,...,z])==26
# memo : 2^20 >= 10^6
# 小数の計算を避ける : x/y -> (x*big)//y ex:big=10**9
# @:小さい文字, ~:大きい文字,None: 空の文字列
# ユークリッドの互除法:gcd(x,y)=gcd(x,y-x)
# memo : d 桁以下の p 進表記を用いると p^d-1 以下のすべての
# 非負整数を表現することができる
# memo : (X,Y) -> (X+Y,X−Y) <=> 点を原点を中心に45度回転し、√2倍に拡大
# memo : (x,y)のx正から見た偏角をラジアンで(-πからπ]: math.atan2(y, x)
# memo : a < bのとき ⌊a⌋ ≦ ⌊b⌋
input = lambda: sys.stdin.readline().rstrip()
mi = lambda: map(int, input().split())
li = lambda: list(mi())
ii = lambda: int(input())
py = lambda: print("Yes")
pn = lambda: print("No")
pf = lambda: print("First")
ps = lambda: print("Second")
INF = 10**15
N, M, L, S, E = mi()
TT = S + E
adj = [[] for _ in range(N)]
memo = defaultdict(lambda: INF)
for _ in range(M):
a, b, t = mi()
if a == b:
continue
a -= 1
b -= 1
adj[a].append((t, b))
adj[b].append((t, a))
for i in range(N):
adj[i].sort()
dist = [[INF] * 2 for _ in range(N)]
T = li()
T = [t - 1 for t in T]
T = set(T)
pq = []
heappush(pq, (0, 0, 0))
while pq:
t, node, s = heappop(pq)
if dist[node][s] < t:
continue
dist[node][s] = t
if node in T and t < TT and s == 0:
wait = max(0, S - t)
dist[node][1] = t + wait + 1
heappush(pq, (dist[node][1], node, 1))
for w, nxt in adj[node]:
if dist[nxt][s] <= t + w:
continue
if s == 0 and t + w >= TT:
continue
dist[nxt][s] = t + w
heappush(pq, (dist[nxt][s], nxt, s))
if dist[-1][1] == INF:
print(-1)
else:
print(dist[-1][1])
mattu34