結果

問題 No.158 奇妙なお使い
ユーザー titia
提出日時 2022-03-28 04:42:20
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 2,587 ms / 5,000 ms
コード長 1,570 bytes
コンパイル時間 95 ms
コンパイル使用メモリ 13,056 KB
実行使用メモリ 86,800 KB
最終ジャッジ日時 2024-11-06 20:56:52
合計ジャッジ時間 5,441 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

import heapq

A=list(map(int,input().split()))
DB=int(input())
B=list(map(int,input().split()))
DC=int(input())
C=list(map(int,input().split()))

DP=dict()

DP[tuple(A)]=0

def SUM(x,y,z):
    return x*1000+y*100+z

Q=[]
Q.append((-SUM(A[0],A[1],A[2]),A[0],A[1],A[2]))

def calc_b(x,y,z):
    money=DB
    
    k=money//1000
    MIN=min(k,x)
    x-=MIN
    money-=MIN*1000

    k=money//100
    MIN=min(k,y)
    y-=MIN
    money-=MIN*100

    z-=money

    if x>=0 and y>=0 and z>=0:
        return x+B[0],y+B[1],z+B[2]

    return -1,-1,-1

def calc_c(x,y,z):
    money=DC
    
    k=money//1000
    MIN=min(k,x)
    x-=MIN
    money-=MIN*1000

    k=money//100
    MIN=min(k,y)
    y-=MIN
    money-=MIN*100

    z-=money

    if x>=0 and y>=0 and z>=0:
        return x+C[0],y+C[1],z+C[2]

    return -1,-1,-1

while Q:
    S,a,b,c=heapq.heappop(Q)
    time=DP[a,b,c]

    a2,b2,c2=calc_b(a,b,c)

    if a2>=0 and b2>=0 and c2>=0:
        if (a2,b2,c2) in DP:
            if DP[(a2,b2,c2)]<time+1:
                DP[a2,b2,c2]=time+1
                heapq.heappush(Q,(-SUM(a2,b2,c2),a2,b2,c2))
                
        else:
            DP[a2,b2,c2]=time+1
            heapq.heappush(Q,(-SUM(a2,b2,c2),a2,b2,c2))

    a2,b2,c2=calc_c(a,b,c)

    if a2>=0 and b2>=0 and c2>=0:
        if (a2,b2,c2) in DP:
            if DP[(a2,b2,c2)]<time+1:
                DP[a2,b2,c2]=time+1
                heapq.heappush(Q,(-SUM(a2,b2,c2),a2,b2,c2))
        else:
            DP[a2,b2,c2]=time+1
            heapq.heappush(Q,(-SUM(a2,b2,c2),a2,b2,c2))

print(max(DP.values()))
0