結果
問題 |
No.108 トリプルカードコンプ
|
ユーザー |
![]() |
提出日時 | 2020-12-05 13:36:46 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 412 ms / 5,000 ms |
コード長 | 921 bytes |
コンパイル時間 | 499 ms |
コンパイル使用メモリ | 81,920 KB |
実行使用メモリ | 140,984 KB |
最終ジャッジ日時 | 2024-09-15 17:05:54 |
合計ジャッジ時間 | 4,196 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge6 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 20 |
ソースコード
import sys from functools import lru_cache sys.setrecursionlimit(10**7) def I(): return int(sys.stdin.readline().rstrip()) def MI(): return map(int,sys.stdin.readline().rstrip().split()) def LI(): return list(map(int,sys.stdin.readline().rstrip().split())) def LI2(): return list(map(int,sys.stdin.readline().rstrip())) def S(): return sys.stdin.readline().rstrip() def LS(): return list(sys.stdin.readline().rstrip().split()) def LS2(): return list(sys.stdin.readline().rstrip()) N = I() A = LI() x,y,z = 0,0,0 for a in A: if a == 0: x += 1 elif a == 1: y += 1 elif a == 2: z += 1 @lru_cache(maxsize=None) def dp(i,j,k): # 0,1,2枚集まっているカードがi,j,k種類ある状態での期待値 if i < 0 or j < 0 or k < 0: return 0 if i == j == k == 0: return 0 return (i*dp(i-1,j+1,k)+j*dp(i,j-1,k+1)+k*dp(i,j,k-1)+N)/(i+j+k) print(dp(x,y,z))