結果

問題 No.108 トリプルカードコンプ
ユーザー terasaterasa
提出日時 2022-11-22 22:24:05
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 381 ms / 5,000 ms
コード長 831 bytes
コンパイル時間 156 ms
コンパイル使用メモリ 81,692 KB
実行使用メモリ 135,296 KB
最終ジャッジ日時 2023-10-24 13:40:05
合計ジャッジ時間 4,245 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 62 ms
68,228 KB
testcase_01 AC 64 ms
68,228 KB
testcase_02 AC 62 ms
68,224 KB
testcase_03 AC 64 ms
68,228 KB
testcase_04 AC 62 ms
68,228 KB
testcase_05 AC 63 ms
68,224 KB
testcase_06 AC 63 ms
68,224 KB
testcase_07 AC 381 ms
135,296 KB
testcase_08 AC 98 ms
78,864 KB
testcase_09 AC 64 ms
68,348 KB
testcase_10 AC 63 ms
68,232 KB
testcase_11 AC 64 ms
68,232 KB
testcase_12 AC 65 ms
68,240 KB
testcase_13 AC 133 ms
81,896 KB
testcase_14 AC 106 ms
78,724 KB
testcase_15 AC 103 ms
78,364 KB
testcase_16 AC 123 ms
80,644 KB
testcase_17 AC 76 ms
73,464 KB
testcase_18 AC 310 ms
116,588 KB
testcase_19 AC 171 ms
91,528 KB
testcase_20 AC 126 ms
82,972 KB
testcase_21 AC 248 ms
104,888 KB
testcase_22 AC 107 ms
78,436 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from typing import List, Tuple, Callable, TypeVar
import sys
import itertools
import heapq
import bisect
import math
from collections import deque, defaultdict, Counter
from functools import lru_cache, cmp_to_key

input = sys.stdin.readline

if __file__ != 'prog.py':
    sys.setrecursionlimit(10 ** 6)


def readints(): return map(int, input().split())
def readlist(): return list(readints())
def readstr(): return input()[:-1]


N = int(input())
A = readlist()
cnt = defaultdict(int)
for a in A:
    cnt[a] += 1


@lru_cache(None)
def solve(a, b, c):
    if a + b + c == 0:
        return 0
    acc = 0
    if a:
        acc += a * solve(a - 1, b + 1, c)
    if b:
        acc += b * solve(a, b - 1, c + 1)
    if c:
        acc += c * solve(a, b, c - 1)
    return (acc + N) / (a + b + c)


print(solve(cnt[0], cnt[1], cnt[2]))
0