結果
問題 | No.929 よくあるボールを移動するやつ |
ユーザー |
![]() |
提出日時 | 2023-03-15 20:49:39 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 71 ms / 2,000 ms |
コード長 | 1,350 bytes |
コンパイル時間 | 327 ms |
コンパイル使用メモリ | 81,908 KB |
実行使用メモリ | 86,320 KB |
最終ジャッジ日時 | 2024-09-18 08:52:37 |
合計ジャッジ時間 | 2,035 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 12 |
ソースコード
# たとえば00140# 左から貪欲法で動かせばいい# Bを見ていって、数字があったら、Aの左から埋めていく# ABC004Dと近いかもしれない# BiをAjからAkに配るときのムーブ数を出力するdefがあるといいdef move_to_A(i, j, k):if j <= i and k <= i:dist1 = abs(i-j)dist2 = abs(i-k)if dist1 == 0 and dist2 == 0:move = 0elif dist2 == 0:move = dist1*(dist1+1)//2else:move = dist1*(dist1+1)//2 - (dist2-1)*dist2//2elif i <= j and i <= k:dist1 = abs(k-i)dist2 = abs(j-i)if dist1 == 0 and dist2 == 0:move = 0elif dist2 == 0:move = dist1*(dist1+1)//2else:move = dist1*(dist1+1)//2 - (dist2-1)*dist2//2else:# つまり j < i and i < kdist1 = abs(i-j)dist2 = abs(k-i)move = dist1*(dist1+1)//2 + dist2*(dist2+1)//2return move# test# move_to_A(3, 2, 4)N = int(input())B = list(map(int, input().split()))left = 0right = 0ans = 0for i in range(N):if B[i] > 0:right = left + B[i] - 1calc = move_to_A(i, left, right)ans += calcleft = right +1right = left#print(i, B[i], 'calc', calc, 'ans', ans, left, right)print(ans)