結果

問題 No.1935 Water Simulation
ユーザー vwxyzvwxyz
提出日時 2024-06-02 05:08:05
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,628 ms / 2,000 ms
コード長 1,205 bytes
コンパイル時間 94 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 102,500 KB
最終ジャッジ日時 2024-12-22 22:40:10
合計ジャッジ時間 13,107 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 48 ms
11,264 KB
testcase_01 AC 47 ms
11,392 KB
testcase_02 AC 31 ms
10,880 KB
testcase_03 AC 645 ms
52,232 KB
testcase_04 AC 197 ms
21,816 KB
testcase_05 AC 1,172 ms
87,992 KB
testcase_06 AC 78 ms
13,292 KB
testcase_07 AC 71 ms
13,548 KB
testcase_08 AC 607 ms
49,984 KB
testcase_09 AC 1,313 ms
91,240 KB
testcase_10 AC 1,083 ms
85,664 KB
testcase_11 AC 132 ms
17,300 KB
testcase_12 AC 38 ms
10,880 KB
testcase_13 AC 36 ms
11,008 KB
testcase_14 AC 205 ms
17,340 KB
testcase_15 AC 425 ms
34,848 KB
testcase_16 AC 593 ms
49,600 KB
testcase_17 AC 276 ms
29,144 KB
testcase_18 AC 35 ms
10,880 KB
testcase_19 AC 773 ms
57,024 KB
testcase_20 AC 58 ms
11,776 KB
testcase_21 AC 672 ms
52,472 KB
testcase_22 AC 300 ms
30,592 KB
testcase_23 AC 189 ms
17,064 KB
testcase_24 AC 1,628 ms
102,500 KB
testcase_25 AC 58 ms
12,416 KB
testcase_26 AC 41 ms
11,264 KB
testcase_27 AC 270 ms
29,076 KB
testcase_28 AC 170 ms
20,212 KB
testcase_29 AC 31 ms
11,008 KB
testcase_30 AC 34 ms
11,008 KB
testcase_31 AC 106 ms
15,568 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