結果

問題 No.158 奇妙なお使い
ユーザー rpy3cpp
提出日時 2015-06-02 17:31:57
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
RE  
実行時間 -
コード長 1,618 bytes
コンパイル時間 90 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 32,000 KB
最終ジャッジ日時 2024-07-06 13:44:05
合計ジャッジ時間 17,175 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 20 WA * 2 RE * 4 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

def read_data():
    a3, a2, a1 = map(int, input().split())
    db = int(input())
    b3, b2, b1 = map(int, input().split())
    dc = int(input())
    c3, c2, c1 = map(int, input().split())
    return a3, a2, a1, b3, b2, b1, c3, c2, c1, db, dc


def solve(a3, a2, a1, b3, b2, b1, c3, c2, c1, db, dc):
    memo = dict()
    dif_b = db - b3*1000 - b2*100 - b1
    dif_c = dc - c3*1000 - c2*100 - c1
    dif = min(dif_b, dif_c)
    record = 0

    def dfs(a3, a2, a1, depth):
        nonlocal record
        nonlocal memo
        if (a3, a2, a1) in memo:
            return memo[a3, a2, a1]
        if (a3 * 1000 + a2 * 100 + a1) // dif <= record - depth:
            return 0
        tmp = 0
        for d3, d2, d1 in generate_payment_patterns(a3, a2, a1, db):
            t = dfs(a3-d3+b3, a2-d2+b2, a1-d1+b1, depth+1)
            if t > tmp:
                tmp = t
        for d3, d2, d1 in generate_payment_patterns(a3, a2, a1, dc):
            t = dfs(a3-d3+c3, a2-d2+c2, a1-d1+c1, depth+1)
            if t > tmp:
                tmp = t
        memo[a3, a2, a1] = tmp
        if depth + tmp > record:
            record = depth + tmp
        return tmp

    dfs(a3, a2, a1, 0)
    return record


def generate_payment_patterns(a3, a2, a1, d):
    for d3 in range(0, d+1, 1000):
        if d3//1000 > a3:
            break
        for d2 in range(0, d - d3 + 1, 100):
            if d2//100 > a2:
                break
            d1 = d - d3 - d2
            if d1 > a1:
                continue
            yield d3//1000, d2//100, d1


if __name__ == '__main__':
    param = read_data()
    print(solve(*param))
0