結果
問題 | No.1601 With Animals into Institute |
ユーザー | SI1000-github |
提出日時 | 2022-03-24 07:57:28 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 2,955 ms / 3,000 ms |
コード長 | 5,092 bytes |
コンパイル時間 | 399 ms |
コンパイル使用メモリ | 82,224 KB |
実行使用メモリ | 218,496 KB |
最終ジャッジ日時 | 2024-10-12 13:17:17 |
合計ジャッジ時間 | 40,861 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 45 ms
53,632 KB |
testcase_01 | AC | 44 ms
53,760 KB |
testcase_02 | AC | 42 ms
53,888 KB |
testcase_03 | AC | 159 ms
79,940 KB |
testcase_04 | AC | 152 ms
80,476 KB |
testcase_05 | AC | 146 ms
80,000 KB |
testcase_06 | AC | 2,662 ms
216,500 KB |
testcase_07 | AC | 2,615 ms
216,448 KB |
testcase_08 | AC | 2,642 ms
216,492 KB |
testcase_09 | AC | 2,591 ms
215,132 KB |
testcase_10 | AC | 2,809 ms
217,984 KB |
testcase_11 | AC | 2,718 ms
215,808 KB |
testcase_12 | AC | 2,944 ms
214,912 KB |
testcase_13 | AC | 2,955 ms
218,496 KB |
testcase_14 | AC | 2,717 ms
215,040 KB |
testcase_15 | AC | 2,907 ms
216,448 KB |
testcase_16 | AC | 2,889 ms
215,808 KB |
testcase_17 | AC | 2,804 ms
215,424 KB |
testcase_18 | AC | 172 ms
80,384 KB |
testcase_19 | AC | 175 ms
80,384 KB |
testcase_20 | AC | 173 ms
80,768 KB |
testcase_21 | AC | 165 ms
80,128 KB |
testcase_22 | AC | 167 ms
80,384 KB |
testcase_23 | AC | 164 ms
80,128 KB |
testcase_24 | AC | 176 ms
80,768 KB |
testcase_25 | AC | 177 ms
80,384 KB |
testcase_26 | AC | 169 ms
80,384 KB |
testcase_27 | AC | 44 ms
53,248 KB |
testcase_28 | AC | 44 ms
52,992 KB |
testcase_29 | AC | 45 ms
53,248 KB |
testcase_30 | AC | 44 ms
53,120 KB |
testcase_31 | AC | 44 ms
53,248 KB |
testcase_32 | AC | 44 ms
53,376 KB |
testcase_33 | AC | 44 ms
53,376 KB |
testcase_34 | AC | 45 ms
53,248 KB |
testcase_35 | AC | 44 ms
53,504 KB |
testcase_36 | AC | 44 ms
53,248 KB |
testcase_37 | AC | 44 ms
53,632 KB |
testcase_38 | AC | 44 ms
53,376 KB |
ソースコード
#!/usr/bin/python # -*- coding: utf-8 -*- # import from sys import stdin, setrecursionlimit setrecursionlimit(10**8) # def def int_map(): return map(int,input().split()) def int_list(): return ['Null']+list(map(int,input().split())) # 整数のリストのリスト:N+1行のリスト(各行のリストは同じ長さでなくてよい.) def int_mtx(N): x = [['Null']+list(map(int, stdin.readline().split())) for _ in range(N)] x.insert(0,['Null']) return x # 文字のリストのリスト:N+1行のリスト(各行のリストは同じ長さでなくてよい.) # AAAAA # BBBBB # CCCCC # のような標準入力を # '0AAAAA' # '0BBBBB' # '0CCCCC' # と受け取る def str_mtx(N): x = ['0'+stdin.readline()[:-1] for _ in range(N)] x.insert(0,'0') return x # 文字のリストのリスト:N+1行のリスト(各行のリストは同じ長さでなくてよい.) # AAAAA XXXXX # BBBBB YYYYY # CCCCC ZZZZZ # のような標準入力を # [[['0'],[AAAAA],[XXXXX]], # [['0'],[BBBBB],[YYYYY]], # [['0'],[CCCCC],[ZZZZZ]] ] # と受け取る def str_list(N): x = [['0']+list(map(str, stdin.readline().split())) for _ in range(N)] x.insert(0,['0']) return x # 高さH+1, 幅W+1, のゼロ行列の作成 def zero_mtx(H,W): x = [[0]*(W+1) for i in range(H+1)] return x # リストをスペースで分割する(先頭を省く) def print_space(l): return print(" ".join([str(x) for x in l[1:]])) # ゼロインデックスの場合 # 整数のリストのリスト:N行のリスト(各行のリストは同じ長さでなくてよい.) def int_mtx_0(N): x = [list(map(int, stdin.readline().split())) for _ in range(N)] return x # ゼロインデックスの場合 # 文字のリストのリスト:N+1行のリスト(各行のリストは同じ長さでなくてよい.) def str_mtx_0(N): x = [list(stdin.readline()[:-1]) for _ in range(N)] return x # ゼロインデックスの場合 # リストをスペースで分割する(先頭を省かない) def print_space_0(l): return print(" ".join([str(x) for x in l])) # ゼロインデックスの場合 # 高さH, 幅W, のゼロ行列の作成 def zero_mtx_0(H,W): x = [[0]*W for i in range(H)] return x def int_list_0(): return list(map(int,input().split())) ## インポート # from collections import deque # 順列に使う # import itertools # 最大公約数などに使う # import math # リストの要素の数をdict形式で # import collections # 二次元配列のコピーを作りたいとき # a_copy = deepcopy(a) # from copy import deepcopy # main code N,M = int_map() ABCX = int_mtx_0(M) """ ダイクストラ法(ヒープによる優先度付きキューを用いて) 計算量 O( |E|*log(|V|)) E:辺の数 V:頂点数 ベルマンフォードより早いが,負のコストがあると使えない. """ import heapq def dijkstra(edges, start_id): num_node = len(edges) c = [[float('inf')] * 2 for _ in range(num_node)] #スタート地点以外の値は∞で初期化 c[start_id][0] = 0 #スタートは0で初期化 visit = [0]*num_node node_name = [] heapq.heappush(node_name, [0,0, start_id]) while len(node_name) > 0: #ヒープから取り出し _, ani, min_point = heapq.heappop(node_name) if visit[min_point] == 0: #経路の要素を各変数に格納することで,視覚的に見やすくする for factor in edges[min_point]: goal = factor[0] #終点 cost = factor[1] #コスト x = factor[2] # 動物がいるかどうか if ani == 1: if c[min_point][1] + cost < c[goal][1]: c[goal][1] = c[min_point][1] + cost heapq.heappush(node_name, [c[goal][1], 1, goal]) if ani == 0: if x == 0: if c[min_point][0] + cost < c[goal][0]: c[goal][0] = c[min_point][0] + cost if c[goal][0] < c[goal][1]: heapq.heappush(node_name, [c[goal][0], 0, goal]) if x == 1: if c[min_point][0] + cost < c[goal][1]: c[goal][1] = c[min_point][0] + cost heapq.heappush(node_name, [c[goal][1], 1, goal]) if ani == 1: visit[min_point] = 1 return c start_id = N-1 # UVC = [ [始点1,終点1,コスト1], ..., [始点M,終点M,コストM] ] UVC = [] for i in range(M): UVC.append([ABCX[i][0]-1, ABCX[i][1]-1, ABCX[i][2], ABCX[i][3]]) Di = False Edges = [[] for _ in range(N)] if Di: for ui, vi, ci in UVC: Edges[ui].append([vi,ci]) else: for ui, vi, ci, xi in UVC: Edges[ui].append([vi,ci,xi]) Edges[vi].append([ui,ci,xi]) opt = dijkstra(Edges, start_id) for i in range(N-1): print(opt[i][1])