結果

問題 No.1935 Water Simulation
コンテスト
ユーザー lloyz
提出日時 2022-05-13 22:04:49
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 31 ms / 2,000 ms
コード長 1,128 bytes
コンパイル時間 205 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 11,008 KB
最終ジャッジ日時 2024-07-22 04:34:50
合計ジャッジ時間 2,227 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

if len(sys.argv) == 2:
    sys.stdin = open(sys.argv[1])

v1, v2, v3, v4, n = map(int, input().split())

cnt = 0
ANS = [v1, 0, 0, 0]
L = []
flag = False
while n > 0:
    if cnt % 4 == 0:
        if ANS[0] + ANS[1] <= v2:
            ANS[1] += ANS[0]
            ANS[0] = 0
        else:
            ANS[0] -= v2 - ANS[1]
            ANS[1] = v2
    elif cnt % 4 == 1:
        if ANS[1] + ANS[2] <= v3:
            ANS[2] += ANS[1]
            ANS[1] = 0
        else:
            ANS[1] -= v3 - ANS[2]
            ANS[2] = v3
    elif cnt % 4 == 2:
        if ANS[2] + ANS[3] <= v4:
            ANS[3] += ANS[2]
            ANS[2] = 0
        else:
            ANS[2] -= v4 - ANS[3]
            ANS[3] = v4
    else:
        if ANS[3] + ANS[0] <= v1:
            ANS[0] += ANS[3]
            ANS[3] = 0
        else:
            ANS[3] -= v1 - ANS[0]
            ANS[0] = v1
    cnt += 1
    n -= 1
    if len(L) < 4:
        L.append(ANS[:])
    else:
        if L[0] == ANS:
            flag = True
            break
        L = L[1:]
        L.append(ANS[:])
if flag:
    print(*L[n % 4])
else:
    print(*L[-1])
0