結果
問題 | No.108 トリプルカードコンプ |
ユーザー |
![]() |
提出日時 | 2025-01-02 15:43:38 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 851 bytes |
コンパイル時間 | 265 ms |
コンパイル使用メモリ | 82,324 KB |
実行使用メモリ | 96,008 KB |
最終ジャッジ日時 | 2025-01-02 15:43:45 |
合計ジャッジ時間 | 5,323 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 WA * 2 |
other | AC * 4 WA * 16 |
ソースコード
import syssys.setrecursionlimit(200050)def code(X):ret = 0for i in range(4):ret *= (N + 1)ret += X[i]return retdef decode(x: int):RET = [0] * 4for i in reversed(range(4)):RET[i] = x % (N + 1)x //= (N + 1)return RETN = int(input())A = list(map(int, input().split()))ST = [0] * 4for a in A:ST[min(a, 3)] += 1from functools import lru_cache@lru_cache(maxsize=20000)def dfs(x: int):if x == N:return 0a, 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)print((a, b, c, d), ret)return retprint(dfs(code(ST)))