結果

問題 No.1065 電柱 / Pole (Easy)
ユーザー uni_pythonuni_python
提出日時 2020-10-05 22:37:55
言語 PyPy3
(7.3.2)
結果
AC  
実行時間 1,034 ms / 2,000 ms
コード長 3,902 Byte
コンパイル時間 204 ms
使用メモリ 145,688 KB
最終ジャッジ日時 2020-10-05 22:38:27
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
testcase_00 AC 47 ms
66,348 KB
testcase_01 AC 47 ms
66,232 KB
testcase_02 AC 600 ms
111,644 KB
testcase_03 AC 777 ms
144,144 KB
testcase_04 AC 834 ms
144,144 KB
testcase_05 AC 503 ms
144,776 KB
testcase_06 AC 503 ms
144,652 KB
testcase_07 AC 167 ms
92,704 KB
testcase_08 AC 411 ms
139,272 KB
testcase_09 AC 120 ms
82,776 KB
testcase_10 AC 232 ms
105,976 KB
testcase_11 AC 185 ms
97,156 KB
testcase_12 AC 179 ms
95,052 KB
testcase_13 AC 761 ms
126,580 KB
testcase_14 AC 818 ms
131,036 KB
testcase_15 AC 953 ms
139,280 KB
testcase_16 AC 530 ms
112,628 KB
testcase_17 AC 1,034 ms
145,688 KB
testcase_18 AC 421 ms
104,264 KB
testcase_19 AC 918 ms
138,976 KB
testcase_20 AC 369 ms
97,036 KB
testcase_21 AC 483 ms
110,912 KB
testcase_22 AC 901 ms
134,848 KB
testcase_23 AC 82 ms
79,428 KB
testcase_24 AC 92 ms
79,984 KB
testcase_25 AC 173 ms
94,784 KB
testcase_26 AC 529 ms
113,048 KB
testcase_27 AC 602 ms
114,668 KB
testcase_28 AC 936 ms
138,848 KB
testcase_29 AC 241 ms
89,208 KB
testcase_30 AC 943 ms
138,480 KB
testcase_31 AC 617 ms
121,312 KB
testcase_32 AC 500 ms
106,672 KB
testcase_33 AC 1,021 ms
138,108 KB
testcase_34 AC 448 ms
104,340 KB
testcase_35 AC 979 ms
142,192 KB
testcase_36 AC 98 ms
79,920 KB
testcase_37 AC 144 ms
80,936 KB
testcase_38 AC 102 ms
80,548 KB
testcase_39 AC 133 ms
80,812 KB
testcase_40 AC 68 ms
74,096 KB
testcase_41 AC 1,012 ms
142,008 KB
testcase_42 AC 375 ms
99,288 KB
testcase_43 AC 581 ms
113,052 KB
testcase_44 AC 284 ms
92,436 KB
testcase_45 AC 553 ms
112,516 KB
testcase_46 AC 49 ms
66,252 KB
testcase_47 AC 48 ms
66,400 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
input=sys.stdin.readline
def I(): return int(input())
def MI(): return map(int, input().split())
def LI(): return list(map(int, input().split()))

def main():
    mod=10**9+7

    ##########################################
    import heapq
    class Dijkstra():
        """
        ・有向 / 無向は問わない(無向の場合は,逆向きの辺もたす)
        ・負のコストがない場合のみ
        ・計算量はO(E log|V|) 
        ・heapを使うことで頂点を走査する必要がなくなる(代わりに,距離更新したものは確定でなくともqueに入れておく)
        ・復元なし
        """

        #最短のpathをたす


        class Edge():
            #重み付き有向辺

            def __init__(self, _to, _cost):
                self.to =_to
                self.cost = _cost

        def __init__(self, V):
            #引数Vは頂点数
            self.inf=10**20
            self.G = [[] for _ in range(V)] #隣接リストG[u][i]が頂点uのi番目の辺
            self. _E = 0 #辺の数
            self._V = V #頂点数

        #proparty - 辺の数
        def E(self):
            return self._E

        #proparty - 頂点数
        def V(self):
            return self._V

        def add(self, _from, _to, _cost):
            #2頂点と辺のコストを追加
            self.G[_from].append(self.Edge(_to,_cost))
            self._E +=1

        def add2(self, _from, _to, _cost):
            #2頂点と辺のコスト(無向)を追加
            self.G[_from].append(self.Edge(_to, _cost))
            self.G[_to].append(self.Edge(_from, _cost))
            self._E +=2

        def shortest_path(self,s):#,g):
            #始点sから頂点iまでの最短経路長のリストを返す
            que = [] #priority queue
            d = [self.inf] * self.V()

            #prev = [None]*self.V() #prev[j]は,sからjへ最短経路で行くときのjの一つ前の場所
            #復元で使う

            d[s] = 0
            heapq.heappush(que,(0,s)) #始点の距離と頂点番号をヒープに追加

            while len(que)!=0:
                #キューに格納されてある中で一番コストが小さい頂点を取り出す
                cost,v = heapq.heappop(que)

                #キューに格納された最短経路長候補がdの距離よりも大きい場合に処理をスキップ
                if d[v] < cost:
                    continue

                #頂点vに隣接する各頂点iに対して,vを経由した場合の距離を計算して,これがd[i]よりも小さい場合に更新
                for i in range(len(self.G[v])):
                    e = self.G[v][i] #vのi個目の隣接辺
                    if d[e.to] > d[v] + e.cost:
                        d[e.to] = d[v] + e.cost #更新

                        #prev[e.to] = v
                        #復元で使う

                        heapq.heappush(que,(d[e.to],e.to)) #queに新たな最短経路長候補を追加

            """#sからgまでの最短経路
            path = []
            pos = g #今いる場所,ゴールで初期化
            for _ in range(self.V()+1):
                path.append(pos)
                if pos == s:
                    break
                #print("pos:",format(pos))
                pos = prev[pos]
            path.reverse()
            #print(path)"""

            return d#,path
    ########################
    N,M=MI()
    djk=Dijkstra(N)
    
    X,Y=MI()
    X-=1
    Y-=1
    p=[0]*N
    q=[0]*N
    for i in range(N):
        p[i],q[i]=MI()
        
    for i in range(M):
        P,Q=MI()
        P-=1
        Q-=1
        d=(p[P]-p[Q])**2 + (q[P]-q[Q])**2
        d=d**0.5
        
        djk.add2(P,Q,d)
        
    d=djk.shortest_path(X)
    ans=d[Y]
    
    print(ans)


main()
0