結果

問題 No.929 よくあるボールを移動するやつ
ユーザー FromBooskaFromBooska
提出日時 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
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 33 ms
53,208 KB
testcase_01 AC 33 ms
52,540 KB
testcase_02 AC 34 ms
52,864 KB
testcase_03 AC 40 ms
52,824 KB
testcase_04 AC 36 ms
52,472 KB
testcase_05 AC 36 ms
52,732 KB
testcase_06 AC 35 ms
53,660 KB
testcase_07 AC 63 ms
74,508 KB
testcase_08 AC 68 ms
86,320 KB
testcase_09 AC 71 ms
85,432 KB
testcase_10 AC 54 ms
78,916 KB
testcase_11 AC 54 ms
80,736 KB
testcase_12 AC 56 ms
79,052 KB
testcase_13 AC 58 ms
78,996 KB
testcase_14 AC 56 ms
79,272 KB
testcase_15 AC 56 ms
78,496 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

# たとえば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 = 0
        elif dist2 == 0:
            move = dist1*(dist1+1)//2
        else:
            move = dist1*(dist1+1)//2 - (dist2-1)*dist2//2
    elif i <= j and i <= k:
        dist1 = abs(k-i)
        dist2 = abs(j-i)
        if dist1 == 0 and dist2 == 0:
            move = 0
        elif dist2 == 0:
            move = dist1*(dist1+1)//2
        else:
            move = dist1*(dist1+1)//2 - (dist2-1)*dist2//2
    else:
        # つまり j < i and i < k
        dist1 = abs(i-j)
        dist2 = abs(k-i)
        move = dist1*(dist1+1)//2 + dist2*(dist2+1)//2
    return move

# test
# move_to_A(3, 2, 4)

N = int(input())
B = list(map(int, input().split()))

left = 0
right = 0
ans = 0
for i in range(N):
    if B[i] > 0:
        right = left + B[i] - 1
        calc = move_to_A(i, left, right)
        ans += calc
        left = right +1
        right = left
    #print(i, B[i], 'calc', calc, 'ans', ans, left, right)
print(ans)
0