結果

問題 No.108 トリプルカードコンプ
コンテスト
ユーザー ntuda
提出日時 2025-01-02 15:44:08
言語 PyPy3
(8.0.0 + ACL)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 307 ms / 5,000 ms
+ 447µs
コード長 822 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 64 ms
コンパイル使用メモリ 82,568 KB
実行使用メモリ 99,904 KB
最終ジャッジ日時 2026-10-10 07:15:55
合計ジャッジ時間 3,278 ms
ジャッジサーバーID
（参考情報） 		judge1_1 / judge3_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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