結果

問題 No.108 トリプルカードコンプ
ユーザー ayaoniayaoni
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 47 ms
55,040 KB
testcase_01 AC 46 ms
55,168 KB
testcase_02 AC 47 ms
55,040 KB
testcase_03 AC 47 ms
55,040 KB
testcase_04 AC 47 ms
55,040 KB
testcase_05 AC 46 ms
54,912 KB
testcase_06 AC 47 ms
54,528 KB
testcase_07 AC 412 ms
140,984 KB
testcase_08 AC 102 ms
79,616 KB
testcase_09 AC 55 ms
60,672 KB
testcase_10 AC 49 ms
55,168 KB
testcase_11 AC 50 ms
55,040 KB
testcase_12 AC 50 ms
54,784 KB
testcase_13 AC 133 ms
80,896 KB
testcase_14 AC 119 ms
78,336 KB
testcase_15 AC 92 ms
77,184 KB
testcase_16 AC 131 ms
80,000 KB
testcase_17 AC 80 ms
71,936 KB
testcase_18 AC 332 ms
117,992 KB
testcase_19 AC 193 ms
91,520 KB
testcase_20 AC 125 ms
82,688 KB
testcase_21 AC 247 ms
105,236 KB
testcase_22 AC 101 ms
76,800 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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