結果

問題 No.108 トリプルカードコンプ
ユーザー terasaterasa
提出日時 2022-11-22 22:24:05
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 376 ms / 5,000 ms
コード長 831 bytes
コンパイル時間 639 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 135,036 KB
最終ジャッジ日時 2024-09-23 06:01:55
合計ジャッジ時間 3,781 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 69 ms
67,968 KB
testcase_01 AC 67 ms
67,840 KB
testcase_02 AC 65 ms
67,968 KB
testcase_03 AC 64 ms
67,968 KB
testcase_04 AC 70 ms
68,096 KB
testcase_05 AC 64 ms
67,840 KB
testcase_06 AC 69 ms
67,840 KB
testcase_07 AC 376 ms
135,036 KB
testcase_08 AC 100 ms
78,768 KB
testcase_09 AC 66 ms
68,096 KB
testcase_10 AC 65 ms
67,968 KB
testcase_11 AC 65 ms
68,224 KB
testcase_12 AC 65 ms
68,096 KB
testcase_13 AC 129 ms
81,688 KB
testcase_14 AC 105 ms
79,044 KB
testcase_15 AC 98 ms
78,492 KB
testcase_16 AC 120 ms
80,588 KB
testcase_17 AC 76 ms
73,708 KB
testcase_18 AC 287 ms
116,756 KB
testcase_19 AC 167 ms
91,776 KB
testcase_20 AC 122 ms
83,456 KB
testcase_21 AC 224 ms
105,068 KB
testcase_22 AC 105 ms
78,464 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