人工知能と競プロやってくブログ

深層学習・機械学習・AI・atcoder・競技プログラミングについて調べてやってみたことをまとめるブログです

AtCoder Beginner Contest 051 [ D - Candidates of No Shortest Paths ] をScipyのdijkstraで最短経路を求めて解く

scipyのダイクストラの使い方を覚えたので、ダイクストラで解けるAtCoderの別問題に挑戦してみました。

atcoder.jp

書いたPythonコード

  • return_predecessors = True で経路が返ってくる (デフォルトはFalse)
    とのことで、これをつけてdijkstraを呼び出し。
  • indices = range(N) を指定。街0 〜 N-1(1〜N)までの経路を取得。

ここまで出来れば、ほぼ重要な情報は取得済み。
経路に含まれていない辺を総当たりで調べて、カウントする。

参考

kawap23.hatenablog.com