結果

問題 No.1935 Water Simulation
ユーザー 炭酸水
提出日時 2022-05-13 22:28:02
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,392 bytes
コンパイル時間 103 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 11,136 KB
最終ジャッジ日時 2024-07-22 02:33:07
合計ジャッジ時間 2,112 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25 WA * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

a, b, c, d, n = map(int, input().split())
s = [a, b, c, d]

dp = [a, 0, 0, 0]
s_0 = []
s_1 = set()

j = 0
sw = True
while sw:
    s_0.append(dp.copy())
    s_1.add(tuple(dp))
    
    if dp[j % 4] + dp[(j + 1) % 4] <= s[(j + 1) % 4]:
        dp[j % 4], dp[(j + 1) % 4] = 0, dp[j % 4] + dp[(j + 1) % 4]
    else:
        dp[j % 4], dp[(j + 1) % 4] = dp[j % 4] + dp[(j + 1) % 4] - s[(j + 1) % 4], s[(j + 1) % 4]    
    j += 1
    if tuple(dp) in s_1:
        for i in range(len(s_0)):
            if dp == s_0[i]:
                loop_dp = dp
                loop = 17
                loop_T = j - i
        sw = False

if n <= loop:

    dp = [a, 0, 0, 0]    
    j = 0
    while j < n:
        if dp[j % 4] + dp[(j + 1) % 4] <= s[(j + 1) % 4]:
            dp[j % 4], dp[(j + 1) % 4] = 0, dp[j % 4] + dp[(j + 1) % 4]
        else:
            dp[j % 4], dp[(j + 1) % 4] = dp[j % 4] + dp[(j + 1) % 4] - s[(j + 1) % 4], s[(j + 1) % 4]    
        j += 1    
    
    print(*dp)
    
    
else:
    m = (n - loop) // loop_T
    k = n - loop - loop_T * m
    
    dp = loop_dp
    
    j = 0
    while j <= k:
        if dp[j % 4] + dp[(j + 1) % 4] <= s[(j + 1) % 4]:
            dp[j % 4], dp[(j + 1) % 4] = 0, dp[j % 4] + dp[(j + 1) % 4]
        else:
            dp[j % 4], dp[(j + 1) % 4] = dp[j % 4] + dp[(j + 1) % 4] - s[(j + 1) % 4], s[(j + 1) % 4]    
        j += 1
    
    print(*dp)
0