結果

問題 No.21 平均の差
コンテスト
ユーザー kohei2019
提出日時 2021-02-10 10:58:20
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 592 bytes
コンパイル時間 189 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 78,048 KB
最終ジャッジ日時 2024-07-07 21:00:35
合計ジャッジ時間 1,302 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 8 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

import itertools
import sys
import copy
import math
N = int(input())
K = int(input())
lsn = [int(input()) for i in range(N)]
if N==K:
    print(max(lsn)-min(lsn))
    sys.exit()
group = [[lsn[i]] for i in range(K)]
lsk = lsn[K:]
lsp = itertools.product(range(K),repeat=N-K)
ans = 0
for p in lsp:
    lsg2 = copy.deepcopy(group)
    for i in range(N-K):
        lsg2[p[i]].append(lsk[i])
    maxa = 0
    mina = 10000
    for i in range(K):
        maxa = max(maxa,sum(lsg2[i])/len(lsg2[i]))
        mina = min(mina,sum(lsg2[i])/len(lsg2[i]))
    ans = max(ans,math.ceil(maxa-mina))
print(ans)
0