結果

問題 No.108 トリプルカードコンプ
ユーザー ntuda
提出日時 2025-01-02 15:44:08
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 477 ms / 5,000 ms
コード長 822 bytes
コンパイル時間 422 ms
コンパイル使用メモリ 82,608 KB
実行使用メモリ 94,488 KB
最終ジャッジ日時 2025-01-02 15:44:12
合計ジャッジ時間 4,044 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
sys.setrecursionlimit(200050)

def code(X):
    ret = 0
    for i in range(4):
        ret *= (N + 1)
        ret += X[i]
    return ret

def decode(x: int):
    RET = [0] * 4
    for i in reversed(range(4)):
        RET[i] = x % (N + 1)
        x //= (N + 1)
    return RET

N = int(input())
A = list(map(int, input().split()))
ST = [0] * 4
for a in A:
    ST[min(a, 3)] += 1

from functools import lru_cache
@lru_cache(maxsize=20000)
def dfs(x: int):
    if x == N:
        return 0
    a, b, c, d = map(int, decode(x))
    ret = 1 / (1 - d / N)
    if a > 0:
        ret += dfs(code([a - 1, b + 1, c, d])) * a / (N - d)
    if b > 0:
        ret += dfs(code([a, b - 1, c + 1, d])) * b / (N - d)
    if c > 0:
        ret += dfs(code([a, b, c - 1, d + 1])) * c / (N - d)
    return ret


print(dfs(code(ST)))
0