Atcoder 172 C Python どこが間違っているか

Question

前提・実現したいこと

Atcoder 172 CをPythonで解いたのですがどこが間違っているかわかりません。
教えてください。
問題文は以下の通りです。

![]

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

エラーメッセージ

該当のソースコード

N,M,K=map(int,input().split())
A=list(map(int,input().split()))
B=list(map(int,input().split()))
count=A[0]+sum(B)
a=A[0]
p=0
for i in range(N):
    while count>K:
        count-=B[M-1]
        M-=1
    if i!=N-1:
        count+=A[i+1]
        a+=A[i+1]
    p=max(p,i+M+1)
    if a>K:
        break
print(p)

試したこと

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

b09.txt,b19.txtだけWAです。

Comments

[8524ba23]

インデントが無くコードが読めないのでコードブロックで囲んでください。
https://teratail.storage.googleapis.com/uploads/contributed_images/56957fe805d9d7befa7dba6a98676d2b.gif
また、そのコードがどのような動きをするのか簡単な説明を記載すると回答得られやすくなると思います。

[Lhankor_Mhy]

ご提示のコードを試してみました。

3 3 4
1 1 1
1 1 1

の入力に対し、

6

が返りましたが、これは想定通りですか？

[chronostasis]

想定通りでないです。

[Lhankor_Mhy]

コードの意図を読み取れていないので何とも言えませんが、全体のロジックから見直した方が早いのではありませんか？

Answer 1

いくつかマズイ点があります。

• 初期状態でA[0]を加算して開始していますが、A側が一つも使われないパターンではNGではないですか？
• ループ中で、Ai側の加算を無条件に行っていますが本当に問題がないですか？

Aiを加算してKを超えたらその解は回答として不適切ではないですか？

大まかな方針は間違っていない（初期状態でBを全加算し、Bを後ろから減らしつつAを前から加算する）ので、図解しながらもう一度コードを作り直してみることをお勧めします。図を書いてみたら考慮漏れが理解できるかもしれませんよ。

Comments

[hope_mucci]

なお、WAとなっているケースは境界値ケースだと思われるので、A側やB側が1つも使われないケースや合計加算値がKぴったりになるようなケースを重点的に考慮してみてください。