結果

問題 No.1935 Water Simulation
ユーザー vwxyzvwxyz
提出日時 2024-06-02 05:08:05
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 1,355 ms / 2,000 ms
コード長 1,205 bytes
コンパイル時間 100 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 102,424 KB
最終ジャッジ日時 2024-06-02 05:08:17
合計ジャッジ時間 10,902 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 42 ms
11,136 KB
testcase_01 AC 41 ms
11,136 KB
testcase_02 AC 26 ms
10,752 KB
testcase_03 AC 566 ms
52,040 KB
testcase_04 AC 168 ms
21,544 KB
testcase_05 AC 1,026 ms
87,748 KB
testcase_06 AC 67 ms
13,164 KB
testcase_07 AC 64 ms
13,296 KB
testcase_08 AC 535 ms
49,872 KB
testcase_09 AC 1,058 ms
91,300 KB
testcase_10 AC 891 ms
85,580 KB
testcase_11 AC 111 ms
17,424 KB
testcase_12 AC 33 ms
10,752 KB
testcase_13 AC 32 ms
10,752 KB
testcase_14 AC 182 ms
17,468 KB
testcase_15 AC 369 ms
34,724 KB
testcase_16 AC 504 ms
49,480 KB
testcase_17 AC 239 ms
29,160 KB
testcase_18 AC 31 ms
10,624 KB
testcase_19 AC 646 ms
56,636 KB
testcase_20 AC 48 ms
11,520 KB
testcase_21 AC 590 ms
52,276 KB
testcase_22 AC 252 ms
30,392 KB
testcase_23 AC 167 ms
16,936 KB
testcase_24 AC 1,355 ms
102,424 KB
testcase_25 AC 49 ms
12,288 KB
testcase_26 AC 34 ms
11,136 KB
testcase_27 AC 226 ms
28,744 KB
testcase_28 AC 173 ms
20,116 KB
testcase_29 AC 29 ms
10,752 KB
testcase_30 AC 28 ms
10,752 KB
testcase_31 AC 90 ms
15,304 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

def Permutation_Product(x,n,f):
    lst=[]
    dct={}
    idx=0
    cur=x
    while True:
        if cur in dct.keys():
            loop_begin=dct[cur]
            loop_end=idx
            break
        else:
            lst.append(cur)
            dct[cur]=idx
            cur=f(cur)
            idx+=1
    if loop_end-1>=n:
        return lst[n]
    n-=loop_begin
    loop_length=loop_end-loop_begin
    lst=lst[loop_begin:]
    return lst[n%(loop_length)]

A,B,C,D,N=map(int,input().split())
V=A
ABCDE=[]
for a in range(A+1):
    for b in range(B+1):
        for c in range(C+1):
            d=V-a-b-c
            if 0<=d<=D:
                for e in range(4):
                    ABCDE.append((a,b,c,d,e))
idx={ABCDE[i]:i for i in range(len(ABCDE))}
le=len(ABCDE)
perm=[None]*le
for i,(a,b,c,d,e) in enumerate(ABCDE):
    if e==0:
        x=min(B-b,a)
        a-=x
        b+=x
    elif e==1:
        x=min(C-c,b)
        b-=x
        c+=x
    elif e==2:
        x=min(D-d,c)
        c-=x
        d+=x
    elif e==3:
        x=min(A-a,d)
        d-=x
        a+=x
    e+=1
    e%=4
    perm[i]=idx[(a,b,c,d,e)]
A,B,C,D,E=ABCDE[Permutation_Product(idx[(A,0,0,0,0)],N,f=lambda i:perm[i])]
print(A,B,C,D)
0