結果
問題 | No.108 トリプルカードコンプ |
ユーザー |
|
提出日時 | 2024-03-07 20:17:13 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 100 ms / 5,000 ms |
コード長 | 1,383 bytes |
コンパイル時間 | 223 ms |
コンパイル使用メモリ | 82,224 KB |
実行使用メモリ | 86,216 KB |
最終ジャッジ日時 | 2024-09-29 18:41:54 |
合計ジャッジ時間 | 2,669 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 20 |
ソースコード
import sysimport mathimport bisectfrom heapq import heapify, heappop, heappushfrom collections import deque, defaultdict, Counterfrom functools import lru_cachefrom itertools import accumulate, combinations, permutations, productsys.setrecursionlimit(1000000)MOD = 10 ** 9 + 7MOD99 = 998244353input = lambda: sys.stdin.readline().strip()NI = lambda: int(input())NMI = lambda: map(int, input().split())NLI = lambda: list(NMI())SI = lambda: input()SMI = lambda: input().split()SLI = lambda: list(SMI())EI = lambda m: [NLI() for _ in range(m)]def main():N = NI()A = NLI()INF = 10**20E = [[[INF]*(N+1) for _ in range(N+1)] for _ in range(N+1)]C = [0] * 3for a in A:if a >= 3:continueC[a] += 1E[0][0][0] = 0for n0 in range(N+1):for n1 in range(N+1-n0):for n2 in range(N+1-n0-n1):if n0 == n1 == n2 == 0:continuee = 1if n0 > 0:e += E[n0-1][n1+1][n2] * n0 / Nif n1 > 0:e += E[n0][n1-1][n2+1] * n1 / Nif n2 > 0:e += E[n0][n1][n2-1] * n2 / Na = (N-n0-n1-n2) / NE[n0][n1][n2] = e / (1-a)print(E[C[0]][C[1]][C[2]])if __name__ == "__main__":main()