AtCoderのABC229のC問題「Cheese」を解いていました。
提出すると、いくつかのケースのみ「RE」（実行時エラー）となってしまいました（19個がAC、3個がRE）。
なぜREになってしまうのか、教えていただけると幸いです。
自分でも考えてみましたが、わかりませんでした。
問題のリンク
n, w = map(int, input().split())
cheese = dict()
for i in range(n):
a, b = map(int, input().split())
if a in cheese:
cheese[a] += b
else:
cheese[a] = b
c = sorted(cheese)
c.reverse()
rd = 0 #最終的に出力します
i = 0
while w > 0 and i < n:
a = c[i]
if w - cheese[a] == 0:
rd += a * cheese[a]
w -= cheese[a]
print(rd)
exit()
elif w - cheese[a] > 0:
rd += a * cheese[a]
w -= cheese[a]
else: # w - cheese[a] < 0
rd += a * w
print(rd)
exit()
i += 1
print(rd)
このコードよりも効率的な書き方があるのは承知していますが、このコードのエラーの原因を知りたいのでよろしくお願いします。
テストファイルが公開されているので、ご自身で実行してみるといいですよ。
https://www.dropbox.com/sh/arnpe0ef5wds8cv/AAAk_SECQ2Nc6SVGii3rHX6Fa?dl=0
順番に試してみたところ test_04.txt でエラー発生しました。
$ python3 abc229c.py < test_04.txt
Traceback (most recent call last):
File "abc229c.py", line 15, in <module>
a = c[i]
IndexError: list index out of range
@i_rei
Questionerこちらの入力例でデバッグしてみてください。
2 3
1 1
1 1
@i_rei
Questioner美味しさが同じ値のケースだと、チーズの種類Nよりユニークなcの数が減るからですね。
while w > 0 and i < len(c):
にすれば直るんじゃないでしょうか
@i_rei
Questioner