AtCoder / Python エラーを引き起こす例外が見当たらない

Question

#問題

ABC121 C問題より

#方針
(A,B)の組をソートされた辞書で管理します。
残りM本買わないといけないので、安い店の方から順々に店を回っていき、①その店のストックよりもさらに多くのドリンクを買わないといけない場合は、その店のストックを全部掻っ攫う ②その店のストックの分で、目標のM本を揃えることが出来る場合、必要な分だけ買って終了 の条件分岐でいけると考えました。

#コード

N, M = map(int, input().split())

dic = {}
for _ in range(N):
	A, B = map(int, input().split())
	dic[A] = B

dic = dict(sorted(dic.items()))

price = list(dic.keys())
stock = list(dic.values())

print(price)
print(stock)

bottles = M #あと何本必要か
payed = 0

for i in range(len(dic)):

	if stock[i] >= bottles:
		payed += price[i] * bottles
		bottles -= bottles

	elif stock[i] < bottles:
		payed += price[i] * stock[i]
		bottles -= stock[i]

	if bottles == 0:
		print(payed)
		exit()

#質問
入力例では全て所望の数値が出力されるのですが、テストケースの途中でWAが出力されます。比較的シンプルな問題にも思うのですが、どのような取りこぼしや例外処理の怠りが原因となっていると推測されますでしょうか。雑な形の質問とはなってしまいましたが、ご教授のほど何卒よろしくお願い申し上げます。

Answer 1

dic = {}
略
    dic[A] = B

辞書に、単価をキーとして、在庫数を格納しています。

しかし、すべての店について単価が異なるとは限りません。

現状だと、同じ単価が複数入力された場合、先行して入力した単価に対応する在庫数は、後に入力した方で上書きされてしまうと思います。

Comments

[kay_ventris4]

肝心なところを見落としておりました！
ご丁寧にありがとうとざいました。