Atcoder、Pythonにコードの間違い指摘への助言

Question

前提

Atcoder 東京海上日動プログラミングコンテスト2022（AtCoder Beginner Contest 256）にて、コードを書いたのですが、期待する出力がなされませんでした。コードを見ていただき、問題点を教えていただきたいです。

実現したいこと

以下の問題において、期待する出力を出すようなコードを書くこと。
https://atcoder.jp/contests/abc256/tasks/abc256_b

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

入力に対する出力が異なる。

該当のソースコード

n=int(input())
a = list(map(int,input().split()))

#コマが置いてある場所
piece_position=[]
#pの値
p=0

#Aiについて、処理する
for j in a:
  #最初に必ずマス 0 に駒を置く。
  piece_position.append(0)
  
  #配列piece_positionの要素全てについて、Aiを足す
  #添字をカウントする変数
  count=0
  for i in piece_position:
    i+=j
    #もし移動先のマスが存在しない駒であれば、それらを取り除いて P に取り除いた個数を加算する
    if i>3:
      p+=1
      piece_position.pop(count)
    count+=1
    
print(p)

試したこと

実際にコードを書きました。

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

なし

Comments

[y_waiwai]

提示のコードではどういう出力がなされるんでしょうか

[sakura.k]

例えば、
4
1 1 3 2

に対しては、
0
と出力され、

10
2 2 4 1 1 1 4 2 2 1
に対しては、
5
と出力されます。

Answer 1

for文中でリストの要素を増減させようとするとインデックスがずれるのでけっこう大変です。
例えば

xs=[0,1,2,3,4]

for i in range(len(xs)):
    if i == 1 or i ==3:
        xs.pop(i)
    
print(xs) # [0, 2, 3]

というわけでそのあたりを考慮する必要があります。
公式解説 のように「マス0~3にコマが置いてあるかどうか」を管理しましょう。

---

別解は、Aiの逆順の累積和をとって、3を超える要素がいくつあるかを調べればいいです。
例えば入力例1:

4
1 1 3 2

だと、
最初に置いたコマは 1+1+3+2マス動きますし
2番目は 1+3+2 マス動きますし
3番めは 3+2マス
4番目は 2マス 動きます

Comments

[sakura.k]

なるほど。インデックスがずれるせいなのですね。ありがとうございます。