結果

問題 No.158 奇妙なお使い
コンテスト
ユーザー titia
提出日時 2022-03-28 04:43:50
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,370 ms / 5,000 ms
コード長 1,766 bytes
コンパイル時間 505 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 230,292 KB
最終ジャッジ日時 2024-11-06 20:57:38
合計ジャッジ時間 4,777 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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