結果

問題 No.929 よくあるボールを移動するやつ
ユーザー FromBooskaFromBooska
提出日時 2023-03-15 20:49:39
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 77 ms / 2,000 ms
コード長 1,350 bytes
コンパイル時間 1,096 ms
コンパイル使用メモリ 81,504 KB
実行使用メモリ 85,112 KB
最終ジャッジ日時 2023-10-18 12:32:06
合計ジャッジ時間 3,011 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 37 ms
53,348 KB
testcase_01 AC 37 ms
53,348 KB
testcase_02 AC 37 ms
53,348 KB
testcase_03 AC 37 ms
53,348 KB
testcase_04 AC 36 ms
53,348 KB
testcase_05 AC 37 ms
53,348 KB
testcase_06 AC 38 ms
53,348 KB
testcase_07 AC 69 ms
73,812 KB
testcase_08 AC 77 ms
85,108 KB
testcase_09 AC 77 ms
85,112 KB
testcase_10 AC 57 ms
78,660 KB
testcase_11 AC 58 ms
78,684 KB
testcase_12 AC 57 ms
78,680 KB
testcase_13 AC 55 ms
78,636 KB
testcase_14 AC 56 ms
78,636 KB
testcase_15 AC 56 ms
78,644 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