【行によって長さが違う二次元配列の表示】すべての行の長さと値が不定な 2 次元配列の出力

Question

前提

図のような、【行によって長さが違う二次元配列の表示】すべての行の長さと値が不定な 2 次元配列の出力を実装する問題を解いています。

実現したいこと

以下のエラーが出てしまいます。

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

Traceback (most recent call last):
  File "Main.py", line 15, in <module>
    sum += int(B[counter])
IndexError: list index out of range

該当のソースコード

init = input().split(" ")
N = int(init[0])
M = int(init[1])
A = input().split(" ")
B = input().split(" ")
counter = 0
sum = int(B[counter]) - 1
for i in range(N):
    if i == sum:
        print(A[i])
        counter += 1
        sum += int(B[counter])✨ここでエラー
    else:
        print(A[i], end=" ")

試したこと

このエラーは、Bのインデックスが3までなのにそれ以上の範囲を指定してしまっていることが原因と認識しました。そこで、counterの値を再度チェックしました。しかし、ここでBのインデックスが3を超えることはないように思われます。

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

今はpaizaの環境でテストしています。この環境の違いが原因でしょうか？
よろしくお願いいたします。

Comments

[TakaiY]

エラーの出力は、最後の行だけでなく、Traceback...などのある始めのところから全て貼ってください。 問題の発生箇所の情報などが載っています。(自分でも読むことをお勧めします)

[meg_]

> init = input().split(" ")
> A = input().split(" ")
> B = input().split(" ")
上記で入力した値は何でしょうか？

[nitori-kun]

エラーの画像変更しました。

meg_様、
画像のA, Bと同じになります。

[meg_]

> 画像のA, Bと同じになります。
init についてはどうなっていますか？
出来れば画像ではなくテキストで表示していただきますとコピー出来るので良いです。（ソースコードにコメントの形で入れていただけると分かりやすいかと思います）

Answer 1

counterの値をインクリメントした後にBの長さを超えていないかのチェックを追加すると良いかと思います。

if counter < len(B):
    sum += int(B[counter])

Answer 2

counterの値は超えますね。 「counter += 1」の次の行でcounterの値をprintしてみればわかると思います。

Comments

[nitori-kun]

解答ありがとうございます。
sum += int(B[counter])の直後に、print(counter)を入れた出力結果が以下になります。

1 2
1
3 4 5 6 7 8
2
9
3
10

これをみるとcounterは3を超えていないのですが、、

[TakaiY]

こちらでやるとこうなります。
1 2
1
3 4 5 6 7 8
2
9
3
10
4
この後、例外発生です。

というか、10を出力したあと、必ずcounterをインクリメントしてsumを更新しているので、かならずそうなると思うのですがね。

[kazuma-s]

sum += int(B[counter])の「直後に」ではなく、「直前に」ですね。

[TakaiY]

ああ、それですね。 「counter += 1 の次の行」です。 例外が発生する場所の後では意味がありません。

[nitori-kun]

「10を出力したあと、必ずcounterをインクリメントしてsumを更新しているので、かならずそうなると思うのですがね。」
->問題の核心はこれでした！色々アドバイスありがとうございました！