結果

問題 No.108 トリプルカードコンプ
ユーザー nohtaraynohtaray
提出日時 2019-06-14 17:04:58
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
RE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,018 bytes
コンパイル時間 77 ms
コンパイル使用メモリ 10,920 KB
実行使用メモリ 9,868 KB
最終ジャッジ日時 2023-08-07 20:15:33
合計ジャッジ時間 1,950 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge14
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 RE -
testcase_01 RE -
testcase_02 RE -
testcase_03 RE -
testcase_04 RE -
testcase_05 RE -
testcase_06 RE -
testcase_07 RE -
testcase_08 RE -
testcase_09 RE -
testcase_10 RE -
testcase_11 RE -
testcase_12 RE -
testcase_13 RE -
testcase_14 RE -
testcase_15 RE -
testcase_16 RE -
testcase_17 RE -
testcase_18 RE -
testcase_19 RE -
testcase_20 RE -
testcase_21 RE -
testcase_22 RE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import bisect
import os
from collections import Counter, deque
from fractions import gcd
from functools import lru_cache
from functools import reduce
import functools
import heapq
import itertools
import math
import numpy as np
import re
import sys

if os.getenv("LOCAL"):
    sys.stdin = open("_in.txt", "r")

sys.setrecursionlimit(2147483647)
INF = float("inf")

N = int(sys.stdin.readline())
A = np.array(list(map(int, sys.stdin.readline().split())))
reqs = np.maximum(0, 3 - A)


@lru_cache(maxsize=None)
def expected(i, j, k):
    # あと 1 個必要なカードが i 種類
    # あと 2 個必要なカードが j 種類
    # あと 3 個必要なカードが k 種類
    if i == j == k == 0:
        return 0
    s = 0
    if i > 0:
        s += expected(i - 1, j, k) * i
    if j > 0:
        s += expected(i + 1, j - 1, k) * j
    if k > 0:
        s += expected(i, j + 1, k - 1) * k
    return s / (i + j + k) + N / (i + j + k)


print(expected((reqs == 1).sum(), (reqs == 2).sum(), (reqs == 3).sum()))
0