結果

問題 No.160 最短経路のうち辞書順最小
ユーザー chaemonchaemon
提出日時 2019-08-18 19:36:40
言語 Nim
(2.0.2)
結果
AC  
実行時間 13 ms / 5,000 ms
コード長 2,714 bytes
コンパイル時間 3,264 ms
コンパイル使用メモリ 70,584 KB
実行使用メモリ 7,932 KB
最終ジャッジ日時 2023-09-15 12:35:42
合計ジャッジ時間 4,525 ms
ジャッジサーバーID
(参考情報)
judge11 / judge15
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,376 KB
testcase_01 AC 1 ms
4,376 KB
testcase_02 AC 2 ms
4,376 KB
testcase_03 AC 1 ms
4,376 KB
testcase_04 AC 5 ms
4,376 KB
testcase_05 AC 9 ms
6,224 KB
testcase_06 AC 11 ms
6,548 KB
testcase_07 AC 3 ms
4,380 KB
testcase_08 AC 3 ms
4,380 KB
testcase_09 AC 3 ms
4,380 KB
testcase_10 AC 3 ms
4,380 KB
testcase_11 AC 4 ms
4,376 KB
testcase_12 AC 3 ms
4,376 KB
testcase_13 AC 3 ms
4,380 KB
testcase_14 AC 3 ms
4,380 KB
testcase_15 AC 3 ms
4,380 KB
testcase_16 AC 3 ms
4,376 KB
testcase_17 AC 3 ms
4,380 KB
testcase_18 AC 3 ms
4,380 KB
testcase_19 AC 3 ms
4,380 KB
testcase_20 AC 4 ms
4,380 KB
testcase_21 AC 3 ms
4,380 KB
testcase_22 AC 3 ms
4,376 KB
testcase_23 AC 3 ms
4,380 KB
testcase_24 AC 3 ms
4,380 KB
testcase_25 AC 3 ms
4,376 KB
testcase_26 AC 3 ms
4,380 KB
testcase_27 AC 2 ms
4,376 KB
testcase_28 AC 13 ms
7,932 KB
testcase_29 AC 2 ms
4,380 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
/home/judge/data/code/Main.nim(2, 45) Warning: imported and not used: 'math' [UnusedImport]

ソースコード

diff #

#{{{ header
import algorithm, sequtils, tables, macros, math, sets, strutils
when defined(MYDEBUG):
  import header

proc scanf(formatstr: cstring){.header: "<stdio.h>", varargs.}
proc getchar(): char {.header: "<stdio.h>", varargs.}
proc nextInt(): int = scanf("%lld",addr result)
proc nextFloat(): float = scanf("%lf",addr result)
proc nextString(): string =
  var get = false
  result = ""
  while true:
    var c = getchar()
    if int(c) > int(' '):
      get = true
      result.add(c)
    else:
      if get: break
      get = false
template `max=`*(x,y:typed):void = x = max(x,y)
template `min=`*(x,y:typed):void = x = min(x,y)
template infty(T): untyped = ((T(1) shl T(sizeof(T)*8-2)) - 1)
#}}}

#{{{ Graph
type Edge[T] = object
  src,dst:int
  weight:T
  rev:int

proc `<`[T](l,r:Edge[T]):bool = l.weight < r.weight

proc newEdge[T](src,dst:int,weight:T,rev:int = -1):Edge[T] =
  var e:Edge[T]
  e.src = src
  e.dst = dst
  e.weight = weight
  e.rev = rev
  return e

type Graph[T] = seq[seq[Edge[T]]]

proc newGraph[T](n:int):Graph[T]=
  var g:Graph[T]
  g = newSeqWith(n,newSeq[Edge[T]]())
  return g

proc addBiEdge[T](g:var Graph[T],src,dst:int,weight:T=1):void =
  g[src].add(newEdge(src,dst,weight,g[dst].len))
  g[dst].add(newEdge(dst,src,weight,g[src].len-1))

proc addEdge[T](g:var Graph[T],src,dst:int,weight:T=1):void =
  g[src].add(newEdge(src,dst,weight,-1))
#}}}

import heapqueue

proc cmp2(l,r:Edge[int]):int = 
  if l.dst < r.dst : -1 else: 1

proc shortestPath[T](g:Graph[T], s:int): (seq[T],seq[int]) = 
  var
    n = g.len
    dist = newSeqWith(n,int.infty)
    prev = newSeqWith(n,-1)
    Q = initHeapQueue[Edge[T]]()
  dist[s] = 0
  Q.push(newEdge[int](-2,s,0))
  while Q.len > 0:
    var e = Q.pop()
    if prev[e.dst] != -1: continue
    prev[e.dst] = e.src;
    for f in g[e.dst]:
      var w = e.weight + f.weight;
      if dist[f.dst] > w:
        dist[f.dst] = w;
        Q.push(newEdge(f.src, f.dst, w));
    discard
  return (dist,prev)
proc path(t: int, prev: seq[int]): seq[int] = 
  var
    u = t
  while u >= 0:
    result.add(u)
    u = prev[u]
  result.reverse()


proc main():void =
  var
    N = nextInt()
    M = nextInt()
    S = nextInt()
    G = nextInt()
    a = newSeq[int](M)
    b = newSeq[int](M)
    c = newSeq[int](M)
    g = newGraph[int](N)
  for i in 0..M-1:
    a[i] = nextInt();b[i] = nextInt();c[i] = nextInt()
    g.addBiEdge(a[i],b[i],c[i])
  for u in 0..N-1:
    g[u].sort(cmp2)
  var (dist,_) = shortest_path(g,G)
  var
    cur = S
    ans = newSeq[int]()
  while cur != G:
    ans.add(cur)
    for e in g[cur]:
      if dist[cur] == dist[e.dst] + e.weight:
        cur = e.dst
        break
  ans.add(cur)
  echo ans.join(" ")
  discard

main()
0