結果

問題 No.3051 Make All Divisible
ユーザー 👑 rin204
提出日時 2025-02-03 23:05:14
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,284 ms / 2,000 ms
コード長 1,501 bytes
コンパイル時間 202 ms
コンパイル使用メモリ 82,416 KB
実行使用メモリ 77,036 KB
最終ジャッジ日時 2025-02-08 15:05:06
合計ジャッジ時間 9,467 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

def solve():
n, K = map(int, input().split())
A = list(map(int, input().split()))
A.sort()
C = [a % K for a in A]
if sum(C) % K != 0:
print(-1)
return
def ok(x: int) -> bool:
tot = 0
x_k = x % K
for a, c in zip(A, C):
if a < x:
tot += a
else:
if x < c:
return False
if c <= x_k:
tot += x - x_k + c
else:
tot += x - x_k + c - K
return x <= tot // K
rr = A[n - K] - 1 + K
inf = 1 << 62
max_c = inf
for d in range(K):
ma = (rr - d) // K * K + d
if not ok(ma):
continue
l = -1
r = ma // K
while r - l > 1:
mid = (l + r) >> 1
if ok(mid * K + d):
r = mid
else:
l = mid
if r * K + d < max_c:
max_c = r * K + d
x = max_c
tot = 0
max_cnt = 0
x_k = x % K
for a, c in zip(A, C):
if a < x:
tot += a
else:
if c <= x_k:
tot += x - x_k + c
if c == x_k:
max_cnt += 1
else:
tot += x - x_k + c - K
tot //= K
if max_c < K:
max_cnt = 0
ans = max(max_c, tot - max_cnt)
if ans == inf:
print(-1)
else:
print(ans)
for _ in range(int(input())):
solve()
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0