結果

問題 No.108 トリプルカードコンプ
コンテスト
ユーザー ayaoni
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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))
0