#884587 (PyPy3) No.2354 Poor Sight in Winter

• 問題: No.2354 Poor Sight in Winter
• ユーザー: komkompi
• 提出日時: 2023-06-24 17:05:54
• 言語: PyPy3 (7.3.15)
• 結果: AC
• 実行時間: 529 ms / 2,000 ms
• コード長: 1,599 bytes
• コンパイル時間: 174 ms
• コンパイル使用メモリ: 81,920 KB
• 実行使用メモリ: 103,752 KB
• 最終ジャッジ日時: 2024-07-01 20:05:10
• 合計ジャッジ時間: 7,225 ms
• ジャッジサーバーID （参考情報）: judge1 / judge2

ファイルパターン	結果
sample	AC * 3
other	AC * 26

ソースコード

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
# coding: utf-8
# Your code here!
from collections import defaultdict
import heapq as hq
N,K=map(int,input().split())
MAX=10**9
edges=defaultdict(dict)
def get_distance(loc1,loc2):
    x1,y1=loc1
    x2,y2=loc2
    
    return abs(x2-x1)+abs(y2-y1)
sx,sy,gx,gy=map(int,input().split())
distance=get_distance((sx,sy),(gx,gy))
edges[(sx,sy)][(gx,gy)]=distance
edges[(gx,gy)][(sx,sy)]=distance
for _ in range(N):
    x,y=map(int,input().split())
    
    nodes=list(edges.keys())
    
    for nx,ny in nodes:
        distance=get_distance((x,y),(nx,ny))
        edges[(x,y)][(nx,ny)]=distance
        edges[(nx,ny)][(x,y)]=distance
def search_way(p,goal):
    dp=defaultdict(lambda:10**9)
    start=[[0,(sx,sy)]]
    hq.heapify(start)
    
    while start:
        cost,now=hq.heappop(start)
        if dp[now]<cost:
            continue
        else:
            dp[now]=cost
            if now==goal:
                break
        
        for (nx,ny),dist in edges[now].items():
            shortage=dist-p
            if p!=0:
                need_light=max(-(-shortage//p),0)
            else:
                need_light=shortage
            
            if cost+need_light<=K and dp[(nx,ny)]>cost+need_light:
                dp[(nx,ny)]=cost+need_light
                hq.heappush(start,[cost+need_light,(nx,ny)])
    if dp[goal]<=K:
        return True
    else:
        return False
            
high=2*10**5+10
low=0
while high-low>1:
    middle=(high+low)//2
    judge=search_way(middle,(gx,gy))
    
    if judge:
        high=middle
    else:
        low=middle
    
print(high)
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX