結果
問題 |
No.108 トリプルカードコンプ
|
ユーザー |
|
提出日時 | 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 |
ソースコード
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))