結果

問題 No.108 トリプルカードコンプ
ユーザー ayaoniayaoni
提出日時 2020-12-05 13:36:46
言語 PyPy3
(7.3.13)
結果
AC  
実行時間 428 ms / 5,000 ms
コード長 921 bytes
コンパイル時間 375 ms
コンパイル使用メモリ 87,276 KB
実行使用メモリ 154,612 KB
最終ジャッジ日時 2023-10-13 21:30:14
合計ジャッジ時間 5,565 ms
ジャッジサーバーID
(参考情報)
judge12 / judge15
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 102 ms
72,444 KB
testcase_01 AC 102 ms
72,344 KB
testcase_02 AC 103 ms
72,572 KB
testcase_03 AC 98 ms
72,528 KB
testcase_04 AC 101 ms
72,168 KB
testcase_05 AC 105 ms
72,512 KB
testcase_06 AC 99 ms
72,456 KB
testcase_07 AC 428 ms
154,612 KB
testcase_08 AC 143 ms
83,992 KB
testcase_09 AC 106 ms
76,432 KB
testcase_10 AC 100 ms
72,360 KB
testcase_11 AC 98 ms
72,300 KB
testcase_12 AC 100 ms
72,576 KB
testcase_13 AC 168 ms
85,180 KB
testcase_14 AC 143 ms
81,528 KB
testcase_15 AC 134 ms
81,848 KB
testcase_16 AC 177 ms
84,520 KB
testcase_17 AC 120 ms
77,588 KB
testcase_18 AC 358 ms
125,628 KB
testcase_19 AC 223 ms
97,220 KB
testcase_20 AC 180 ms
85,332 KB
testcase_21 AC 278 ms
108,852 KB
testcase_22 AC 141 ms
81,144 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