競技プログラミングをpythonで解いているがエラーが解決できない。

実現したいこと

AtCoderの東京海上日動プログラミングコンテスト2024(AtCoder Beginner Contest 355)のc問題をpythonで解きたい。https://atcoder.jp/contests/abc355/tasks/abc355_c 

発生している問題・分からないこと

AtCoderに提出するとACとなり正解だが、入力例1を入力した時エラーが発生する。
入力例1
3 5
5 1 8 9 7

エラーメッセージ

error

1Traceback (most recent call last): 2 File "c:\Users\mikim\python関係\hello.py", line 14, in <module> 3 heigt[h]+=1 4 ~~~~~^^^ 5IndexError: list index out of range

該当のソースコード

python

1N, T = map(int, input().split())2A = list(map(int, input().split()))3 4# 縦、横の判定リストを初期化5row_counts = [0] * N 6col_counts = [0] * N 7 8# 対角線の判定変数を初期化9diagonal_1_count, diagonal_2_count = 0, 010 11# ターンを表す変数を初期化12turn = 013 14# 各ターンの処理15for move in A:16 turn += 117 row = (move - 1) // N 18 col = (move - 1) % N 19 20 row_counts[row] += 121 col_counts[col] += 122 23 if row == col:24 diagonal_1_count += 125 if row + col == N - 1:26 diagonal_2_count += 127 28 # ビンゴかチェック29 if row_counts[row] == N or col_counts[col] == N or diagonal_1_count == N or diagonal_2_count == N:30 print(turn)31 exit()32 33print(-1)34

試したこと・調べたこと

上記の詳細・結果

解決することができなかった。

補足

特になし

コメントを投稿

0 コメント