結果

問題 No.3051 Make All Divisible
ユーザー titia
提出日時 2025-03-09 01:59:34
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 718 ms / 2,000 ms
コード長 1,298 bytes
コンパイル時間 679 ms
コンパイル使用メモリ 82,112 KB
実行使用メモリ 79,800 KB
最終ジャッジ日時 2025-03-09 01:59:41
合計ジャッジ時間 5,749 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

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

import sys
input = sys.stdin.readline
from heapq import heappop,heappush
T=int(input())
for tests in range(T):
    n,k=map(int,input().split())
    A=list(map(int,input().split()))
    SUM=sum(A)
    if SUM%k!=0:
        print(-1)
        continue
    flag=1
    for a in A:
        if a%k!=0:
            flag=0
            break
    if flag:
        print(0)
        continue
    A2=[]
    for a in A:
        if a!=0:
            A2.append(a)
    A=A2
    n=len(A)
    if n<k:
        print(-1)
        continue
    if n==k:
        B=[a%k for a in A]
        x=B[0]
        flag=1
        for b in B:
            if x!=b:
                flag=0
                break
        if flag:
            print(x)
        else:
            print(-1)
        continue
    B=[]
    C=[]
    SUM=0
    MAX=0
    for a in A:
        if a>=k:
            B.append((a%k,a//k))
        else:
            C.append(a)
        SUM+=a%k
        MAX=max(MAX,a%k)
    if MAX<=SUM//k:
        print(SUM//k)
        continue
    B.sort()
    ANS=-1
    while B:
        x,com=heappop(B)
        if com==0:
            continue
        MAX=max(MAX,x+k)
        SUM+=k
        heappush(B,(x+k,com-1))
        if MAX<=SUM//k:
            ANS=SUM//k
            break
    print(ANS)
        
    
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0