結果
問題 | No.108 トリプルカードコンプ |
ユーザー |
![]() |
提出日時 | 2015-08-08 23:24:20 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 441 ms / 5,000 ms |
コード長 | 831 bytes |
コンパイル時間 | 85 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 42,636 KB |
最終ジャッジ日時 | 2024-07-18 06:06:17 |
合計ジャッジ時間 | 2,646 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 20 |
ソースコード
from collections import defaultdict def solve(): N = int(input()) A = list(map(int, input().split())) n3 = n2 = n1 = 0 for a in A: if a == 0: n3 += 1 elif a == 1: n2 += 1 elif a == 2: n1 += 1 memo = defaultdict(int) def rec(i, j, k): if i < 0 or j < 0 or k < 0: return 0 ijk = (i + j + k) if ijk == 0: return 0 if (i, j, k) in memo: return memo[i, j, k] ans = ( N / ijk + rec(i, j, k - 1) * k / ijk + rec(i, j - 1, k + 1) * j / ijk + rec(i - 1, j + 1, k) * i / ijk ) memo[i, j, k] = ans return ans print(rec(n3, n2, n1)) def main(): solve() if __name__ == '__main__': main()