結果

問題 No.972 選び方のスコア
コンテスト
ユーザー gew1fw
提出日時 2025-06-12 21:03:38
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 889 bytes
コンパイル時間 151 ms
コンパイル使用メモリ 82,400 KB
実行使用メモリ 122,996 KB
最終ジャッジ日時 2025-06-12 21:05:11
合計ジャッジ時間 5,244 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25 WA * 7
権限があれば一括ダウンロードができます

ソースコード

diff # 

def main():
    import sys
    input = sys.stdin.read().split()
    N = int(input[0])
    a = list(map(int, input[1:N+1]))
    a.sort()
    prefix = [0] * (N + 1)
    for i in range(1, N + 1):
        prefix[i] = prefix[i - 1] + a[i - 1]
    
    max_S = -float('inf')
    
    for k in range(1, N + 1):
        if k % 2 == 1:
            m = (k + 1) // 2
            take = k - m
        else:
            m = k // 2
            take = k - m
        
        sum_sub = prefix[m]
        if take > 0:
            sum_sub += prefix[N] - prefix[N - take]
        
        if k % 2 == 1:
            median = a[m - 1]
            S = sum_sub - k * median
        else:
            idx = N - take
            median = (a[m - 1] + a[idx]) // 2
            S = sum_sub - k * median
        
        if S > max_S:
            max_S = S
    
    print(max_S)

if __name__ == '__main__':
    main()
0