質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

新規登録して質問してみよう
ただいま回答率
85.46%
コードレビュー

コードレビューは、ソフトウェア開発の一工程で、 ソースコードの検査を行い、開発工程で見過ごされた誤りを検出する事で、 ソフトウェア品質を高めるためのものです。

Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

Q&A

解決済

1回答

653閲覧

atcoder ABC208_D pythonでTLEになってしまう

pawapoke

総合スコア3

コードレビュー

コードレビューは、ソフトウェア開発の一工程で、 ソースコードの検査を行い、開発工程で見過ごされた誤りを検出する事で、 ソフトウェア品質を高めるためのものです。

Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

0グッド

0クリップ

投稿2021/07/15 10:55

前提・実現したいこと

表題についてpythonで下記の問題のコードを提出したところ、TLEになってしまいました。
ABC208_D問題

原因は分かっており、コード内の下記の記載を変更したところACになりました。
(誤)float('inf') ⇒(正)INF ←INF=10**18 を設定
ここで質問なのですが、そもそも(誤)のパターンは何が問題だったのでしょうか?
(感覚的にはWAならまだ分かるのですが、、)

どなたかご教授頂けますと幸いです。

発生している問題・エラーメッセージ

10個のパターンでTLEになっています。

該当のソースコード

python

1N,M = map(int,input().split()) 2G = [[float('inf') for _ in range(N)] for _ in range(N)] 3 4for i in range(N): 5 G[i][i] = 0 6 7for i in range(M): 8 A,B,C = map(int,input().split()) 9 G[A-1][B-1] = C 10 11ans = 0 12for k in range(N): 13 for i in range(N): 14 for j in range(N): 15 G[i][j] = min(G[i][j],G[i][k]+G[k][j]) 16 if G[i][j] != float('inf'): 17 ans += G[i][j] 18print(ans)

試したこと

ここに問題に対して試したことを記載してください。

補足情報(FW/ツールのバージョンなど)

ここにより詳細な情報を記載してください

気になる質問をクリップする

クリップした質問は、後からいつでもMYページで確認できます。

またクリップした質問に回答があった際、通知やメールを受け取ることができます。

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

guest

回答1

0

ベストアンサー

遅くなる以下の部分です。

python

1G = [[float('inf') for _ in range(N)] for _ in range(N)]

float(inf)をN**2回実行します。以下の方が速いでしょう。

python

1G = [[float('inf')] * N] * N

次に、

python

1 G[i][j] = min(G[i][j],G[i][k]+G[k][j])

では、G[i][j] がintでG[i][k]+G[k][j]がinfの場合にintからfloatへの型変換が発生します。

python

1 if G[i][j] != float('inf'):

ここが一番遅くなる原因です。
まず、float(inf)がN**3回実行されます。さらにG[i][j] がintの場合にintからfloatへの型変換が発生します。

投稿2021/07/15 19:23

ppaul

総合スコア24666

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

pawapoke

2021/07/17 00:12

確かに至る所で型変換が必要になるパターンがありますね。。 (正)のコードで上手くいったのはint型で定義してたからなのですね。 型変換自体がそこまで計算量に影響を与えるという認識もありませんでした。 大変勉強になりました。 ppaulさん、ご回答ありがとうございます。
guest

あなたの回答

tips

太字

斜体

打ち消し線

見出し

引用テキストの挿入

コードの挿入

リンクの挿入

リストの挿入

番号リストの挿入

表の挿入

水平線の挿入

プレビュー

15分調べてもわからないことは
teratailで質問しよう!

ただいまの回答率
85.46%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

関連した質問