結果

問題 No.108 トリプルカードコンプ
コンテスト
ユーザー yakeshiba
提出日時 2021-03-12 08:06:56
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 152 ms / 5,000 ms
コード長 652 bytes
コンパイル時間 334 ms
コンパイル使用メモリ 82,280 KB
実行使用メモリ 92,928 KB
最終ジャッジ日時 2024-10-13 17:30:48
合計ジャッジ時間 3,215 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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