結果
問題 |
No.2918 Divide Applicants Fairly
|
ユーザー |
|
提出日時 | 2024-09-19 00:32:35 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 436 ms / 2,000 ms |
コード長 | 1,265 bytes |
コンパイル時間 | 159 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 18,304 KB |
最終ジャッジ日時 | 2024-10-07 21:05:51 |
合計ジャッジ時間 | 6,276 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 61 |
ソースコード
from bisect import bisect_left from collections import defaultdict N = int(input()) if N > 22: print(0) exit() rates = list(map(int, input().split())) rate_sums = defaultdict(lambda: []) def frontexh(i, dif, rate_sum): if i == N // 2: rate_sums[dif].append(rate_sum) else: frontexh(i + 1, dif, rate_sum) frontexh(i + 1, dif + 1, rate_sum + rates[i]) frontexh(i + 1, dif - 1, rate_sum - rates[i]) frontexh(0, 0, 0) if rate_sums[0]: rate_sums[0].pop(0) for dif in range(-N, N + 1): rate_sums[dif].sort() ans = 2 ** 30 has_anyone = False def backexh(i, dif, rate_sum): global ans, has_anyone if i == N - N // 2: if dif == 0: if has_anyone: ans = min(ans, abs(rate_sum)) else: has_anyone = True it = bisect_left(rate_sums[-dif], -rate_sum) if it < len(rate_sums[dif]): ans = min(ans, abs(rate_sum + rate_sums[-dif][it])) if it > 0: ans = min(ans, abs(rate_sum + rate_sums[-dif][it - 1])) else: backexh(i + 1, dif, rate_sum) backexh(i + 1, dif + 1, rate_sum + rates[N // 2 + i]) backexh(i + 1, dif - 1, rate_sum - rates[N // 2 + i]) backexh(0, 0, 0) print(ans)