結果

問題 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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)
0