結果

問題 No.2627 Unnatural Pitch
ユーザー AngrySadEight
提出日時 2024-02-08 13:38:53
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 378 ms / 4,000 ms
コード長 1,326 bytes
コンパイル時間 210 ms
コンパイル使用メモリ 82,052 KB
実行使用メモリ 261,696 KB
最終ジャッジ日時 2024-09-28 12:44:46
合計ジャッジ時間 7,914 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict

N, K, L, U = map(int, input().split())
A = list(map(int, input().split()))

assert 1 <= N <= 200000
assert 1 <= K <= 200000
assert 0 <= L <= 10 ** 12
assert 0 <= U <= 10 ** 12
assert U - L >= K
for i in range(N):
    assert 0 <= A[i] <= 10 ** 12

diffs = U - L
ok = 0
ng = 1000000000000002

while ng - ok > 1:
    mid = (ok + ng) // 2
    lower = 0
    upper = 0
    for i in range(N):
        if A[i] < mid:
            lower += 1
        elif A[i] > mid + diffs:
            upper += 1
    if lower <= upper:
        ok = mid
    else:
        ng = mid

d = defaultdict(int)
for i in range(N):
    d[A[i]] += 1

ans = 10 ** 18 + 2

now_score = 0
now_L = ok - K
now_U = now_L + diffs

left_rem = [0 for _ in range(K)]
right_rem = [0 for _ in range(K)]

for i in range(N):
    if A[i] < now_L:
        now_score += (now_L - A[i] + K - 1) // K
        left_rem[A[i] % K] += 1
    if A[i] > now_U:
        now_score += (A[i] - now_U + K - 1) // K
        right_rem[A[i] % K] += 1

ans = min(ans, now_score)

for i in range(2 * K):
    left_rem[(now_L % K + K) % K] += d[now_L]
    now_score += left_rem[(now_L % K + K) % K]
    now_L += 1
    now_U += 1
    now_score -= right_rem[(now_U % K + K) % K]
    right_rem[(now_U % K + K) % K] -= d[now_U]
    ans = min(ans, now_score)

print(ans)
0