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

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

ただいまの
回答率

89.55%

AOJ ALDS1_2_Dについて(Python)

解決済

回答 1

投稿 編集

  • 評価
  • クリップ 0
  • VIEW 831

cherrypi42

score 5

 前提・実現したいこと

PythonでAOJのALDS1_2_D(シェルソート)の問題にて下のコードを提出したところ、Runtimeエラーと表示されました。また、実行結果を見たところ34件のテストケースの内32件目までは正常に実行されており、33件目でRuntimeエラーを起こしていました。
エラーは入力データの増加によるものなのかと考えましたがはっきりせず、対処法も思いつきません。どなたかご教授いただけないでしょうか。

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

Judge Input #33でのRuntime Error

 該当のソースコード

n, gap = int(input()), []
data = list(map(int, [input() for i in range(n)]))
cnt, g = 0, 0
while g <= n//9:
    g = 3 * g + 1
    gap.append(g)
m = len(gap)
gap = gap[::-1]
for i in gap:
    for idx in range(i, n):
        j = idx - i
        while j >= 0 and data[j] > data[j+i]:
            data[j+i], data[j] = data[j], data[j+i]
            j -= i
            cnt += 1
print(m)
print(" ".join(map(str, gap)))
print(cnt)
for i in range(n):
    print(data[i])

 試したこと

数回提出しましたが、同じケースでRuntime Errorとなってしまいます

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

当該問題のURLです
http://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=ALDS1_2_D

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

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

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

    クリップを取り消します

  • 良い質問の評価を上げる

    以下のような質問は評価を上げましょう

    • 質問内容が明確
    • 自分も答えを知りたい
    • 質問者以外のユーザにも役立つ

    評価が高い質問は、TOPページの「注目」タブのフィードに表示されやすくなります。

    質問の評価を上げたことを取り消します

  • 評価を下げられる数の上限に達しました

    評価を下げることができません

    • 1日5回まで評価を下げられます
    • 1日に1ユーザに対して2回まで評価を下げられます

    質問の評価を下げる

    teratailでは下記のような質問を「具体的に困っていることがない質問」、「サイトポリシーに違反する質問」と定義し、推奨していません。

    • プログラミングに関係のない質問
    • やってほしいことだけを記載した丸投げの質問
    • 問題・課題が含まれていない質問
    • 意図的に内容が抹消された質問
    • 過去に投稿した質問と同じ内容の質問
    • 広告と受け取られるような投稿

    評価が下がると、TOPページの「アクティブ」「注目」タブのフィードに表示されにくくなります。

    質問の評価を下げたことを取り消します

    この機能は開放されていません

    評価を下げる条件を満たしてません

    評価を下げる理由を選択してください

    詳細な説明はこちら

    上記に当てはまらず、質問内容が明確になっていない質問には「情報の追加・修正依頼」機能からコメントをしてください。

    質問の評価を下げる機能の利用条件

    この機能を利用するためには、以下の事項を行う必要があります。

回答 1

checkベストアンサー

+1

RuntimeErrorなので時間が超過してエラーになっています。
もっとコードを高速化する必要があることを意味しています。


まずここですが、ループ内で2回代入します。

j = idx - i
while j >= 0 and data[j] > data[j+i]:
    data[j+i], data[j] = data[j], data[j+i]
    j -= i
    cnt += 1

一方で概念コードではループ内で1回しか代入しません。

insertionSort(A, n, g)
    for i = g to n-1
    v = A[i]
    j = i - g
    while j >= 0 && A[j] > v
        A[j+g] = A[j]
        j = j - g
        cnt++
    A[j+g] = v


最後のcnt回だけ代入が多いです。

これで5秒短縮。


次に、入力サイズが果てしない場合、Pythonは読み込みでボトルネックが来ます。

data = list(map(int, [input() for i in range(n)]))


よりも

import sys
data = [int(l) for l in sys.stdin.readlines()]


の方がずっと速いです。

これで4秒短縮。


そして最後に、
https://stackoverflow.com/questions/11241523/why-does-python-code-run-faster-in-a-function
関数のほうが実行速度が速いです。

つまり、

for g in gap[::-1]:
    c = 0
    for idx in range(g, n):
        v = data[idx]
        j = idx - g
        while j >= 0 and data[j] > v:
            data[j+g] = data[j]
            j -= g
            c += 1
        data[j+g] = v
    cnt += c


よりも

def f(data, g, n):
    c = 0
    for idx in range(g, n):
        v = data[idx]
        j = idx - g
        while j >= 0 and data[j] > v:
            data[j+g] = data[j]
            j -= g
            c += 1
        data[j+g] = v
    return c

for g in gap[::-1]:
    cnt += f(data, g, n)


の方がずっと速いです。

これで20秒短縮。


もともと43秒くらいだったのが16秒になります。
微妙にずれるのは四捨五入と測定が雑なせいです。

投稿

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

  • 2018/02/15 11:24

    回答ありがとうございました。クリアすることができました。
    ご指摘の通りにソースコードを見直したところ、シェルソートとは別物になっていたことにようやく気付きました…。また、関数にすることで実行時間を大きく短縮できることなども教えていただき、本当にありがとうございました。

    キャンセル

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

  • ただいまの回答率 89.55%
  • 質問をまとめることで、思考を整理して素早く解決
  • テンプレート機能で、簡単に質問をまとめられる

同じタグがついた質問を見る