結果

問題 No.21 平均の差
ユーザー 学ぶマン
提出日時 2025-06-06 22:27:31
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 647 bytes
コンパイル時間 473 ms
コンパイル使用メモリ 82,348 KB
実行使用メモリ 84,852 KB
最終ジャッジ日時 2025-06-06 22:27:40
合計ジャッジ時間 7,534 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 4 TLE * 1 -- * 5
権限があれば一括ダウンロードができます

ソースコード

diff #

from itertools import product
N = int(input())
K = int(input())
nums = [int(input()) for _ in range(N)]
ansl = []
for pat in product([i for i in range(K)], repeat=N):
    sum = [0]*K
    cnt = [0]*K
    for i in range(N):
        # pat[i] には グループ番号が書かれている
        sum[pat[i]] += nums[i]
        cnt[pat[i]] += 1
    # Kグループ成立しているか?
    memo = []
    for i in range(K):
        if cnt[i] == 0:
            break
        memo.append(sum[i]/cnt[i])
    else:
        memo.sort()
        ansl.append(int(memo[-1] - memo[0]))
    # break したら終わり
    continue

ansl.sort()
print(ansl[-1])
0