結果

問題 No.108 トリプルカードコンプ
ユーザー yakeshibayakeshiba
提出日時 2021-03-12 08:06:56
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 150 ms / 5,000 ms
コード長 652 bytes
コンパイル時間 268 ms
コンパイル使用メモリ 81,940 KB
実行使用メモリ 92,416 KB
最終ジャッジ日時 2024-04-21 18:39:07
合計ジャッジ時間 2,874 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 48 ms
69,888 KB
testcase_01 AC 47 ms
69,504 KB
testcase_02 AC 47 ms
69,504 KB
testcase_03 AC 48 ms
69,632 KB
testcase_04 AC 47 ms
69,504 KB
testcase_05 AC 49 ms
69,632 KB
testcase_06 AC 48 ms
69,248 KB
testcase_07 AC 150 ms
92,416 KB
testcase_08 AC 72 ms
87,204 KB
testcase_09 AC 47 ms
69,632 KB
testcase_10 AC 47 ms
69,376 KB
testcase_11 AC 47 ms
69,504 KB
testcase_12 AC 47 ms
69,504 KB
testcase_13 AC 80 ms
87,468 KB
testcase_14 AC 80 ms
87,552 KB
testcase_15 AC 82 ms
87,932 KB
testcase_16 AC 80 ms
87,108 KB
testcase_17 AC 50 ms
70,912 KB
testcase_18 AC 131 ms
90,240 KB
testcase_19 AC 90 ms
88,300 KB
testcase_20 AC 84 ms
87,252 KB
testcase_21 AC 109 ms
88,832 KB
testcase_22 AC 80 ms
87,484 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
sys.setrecursionlimit(10**9)

n = int(input())
A = list(map(int,input().split()))

dp = [[[-1]*(110) for _ in range(110)] for _ in range(110)]

def rec(i,j,k):
    if dp[i][j][k] != -1:
        return dp[i][j][k]
    if i+j+k == 0:
        return 0.0
    
    ret = 0
    if i > 0:
        ret += i*rec(i-1,j+1,k)
    if j > 0:
        ret += j*rec(i,j-1,k+1)
    if k > 0:
        ret += k*rec(i,j,k-1)
    ret += n
    ret *= 1.0/(i+j+k)
    
    dp[i][j][k] = ret
    return dp[i][j][k]

I = 0
J = 0
K = 0
for a in A:
    if a == 0:
        I += 1
    elif a == 1:
        J += 1
    elif a == 2:
        K += 1
        
print(rec(I,J,K))
0