結果

問題 No.54 Happy Hallowe'en
ユーザー rpy3cpp
提出日時 2015-08-02 18:02:23
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,719 bytes
コンパイル時間 186 ms
コンパイル使用メモリ 13,056 KB
実行使用メモリ 15,744 KB
最終ジャッジ日時 2024-07-18 00:48:49
合計ジャッジ時間 1,691 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 2 WA * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
sys.setrecursionlimit(1000000)

def read_data():
    N = int(input())
    VT = []
    for n in range(N):
        v, t = map(int, input().split())
        VT.append((v + t, v, t))
    return N, VT

def solve(N, VT):
    VT.sort(reverse=True)
    upper = min(VT[0][0], sum(v for vt, v, t in VT) + 1)
    lower = 1
    mid = upper - 1
    while lower + 1 < upper:
        if is_valid(mid, VT):
            lower = mid
        else:
            upper = mid
        mid = (upper + lower) // 2
    return lower

def is_valid(val, VT):
    head = 0
    nexts = list(range(1, len(VT) + 1))
    prevs = list(range(-1, len(VT) - 1))
    lower = val
    upper = float('inf')
    return dfs(lower, upper, VT, head, nexts, prevs)

def dfs(lower, upper, VT, head, nexts, prevs):
    '''VTのうち、used でないものを使って、lower 以上、upper 未満の選び方をできるかを返す。
    '''
    print(head)
    if lower <= 0:
        return True
    i = head
    end = len(VT)
    while i != end:
        val, v, t = VT[i]
        if v >= upper:
            i = nexts[i]
            continue
        if val < lower:
            return False
        if i == head:
            nhead = nexts[i]
        else:
            nhead = head
            nexts[prevs[i]] = nexts[i]
        if nexts[i] != end:
            prevs[nexts[i]] = prevs[i]
        result = dfs(lower - v, min(upper, t), VT, nhead, nexts, prevs)
        if i != head:
            nexts[prevs[i]] = i
        if nexts[i] != end:
            prevs[nexts[i]] = i
        if result:
            return True
        i = nexts[i]
    return False


N, VT = read_data()
#N = 4040
#VT = [(1 + i, 1, i) for i in range(1, N + 1)]

print(solve(N, VT))
0