結果
問題 |
No.21 平均の差
|
ユーザー |
|
提出日時 | 2021-01-11 23:09:59 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
TLE
|
実行時間 | - |
コード長 | 759 bytes |
コンパイル時間 | 101 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 21,248 KB |
最終ジャッジ日時 | 2024-11-21 09:25:23 |
合計ジャッジ時間 | 16,885 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 8 TLE * 2 |
ソースコード
import sys from itertools import islice, product def _solve(N, K, values): groups = list(range(K)) ans = 0 for g in product(groups, repeat=N): a = [[] for _ in range(K)] for k, v in zip(g, values): a[k].append(v) b = [] try: for v in a: b.append(sum(v) / len(v)) except ZeroDivisionError: continue average_d = -int(-(max(b) - min(b))) ans = max(ans, average_d) return ans def solve(in_): N = int(next(in_)) K = int(next(in_)) values = [int(line) for line in islice(in_, N)] return _solve(N, K, values) def main(): answer = solve(sys.stdin.buffer) print(answer) if __name__ == '__main__': main()