結果
問題 |
No.108 トリプルカードコンプ
|
ユーザー |
|
提出日時 | 2020-12-04 20:55:54 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 305 ms / 5,000 ms |
コード長 | 598 bytes |
コンパイル時間 | 78 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 24,960 KB |
最終ジャッジ日時 | 2024-09-15 07:40:38 |
合計ジャッジ時間 | 2,658 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 20 |
ソースコード
N = int(input()) A = list(map(int,input().split())) B = [0,0,0,0] for a in A: if a >= 3: B[3] += 1 else: B[a] += 1 n0 = B[0]; n1 = B[1]; n2 = B[2] nS = n0+n1+n2+1 dp = [[[-1]*nS for _ in range(nS)] for _ in range(nS)] dp[0][0][0] = 0 def solve(n0,n1,n2): ns = n0 + n1 + n2 if ns > N or n0 < 0 or n1 < 0 or n2 < 0: return 0 elif dp[n0][n1][n2] != -1: return dp[n0][n1][n2] else: v = (N + n0*solve(n0-1,n1+1,n2)+n1*solve(n0,n1-1,n2+1)+n2*solve(n0,n1,n2-1))/ns dp[n0][n1][n2] = v return v print(solve(n0,n1,n2))