結果
問題 | No.108 トリプルカードコンプ |
ユーザー | mkawa2 |
提出日時 | 2020-01-20 17:22:50 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 674 ms / 5,000 ms |
コード長 | 1,604 bytes |
コンパイル時間 | 83 ms |
コンパイル使用メモリ | 11,204 KB |
実行使用メモリ | 49,616 KB |
最終ジャッジ日時 | 2023-09-15 21:18:10 |
合計ジャッジ時間 | 2,777 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 21 ms
8,796 KB |
testcase_01 | AC | 20 ms
8,624 KB |
testcase_02 | AC | 20 ms
8,772 KB |
testcase_03 | AC | 20 ms
8,652 KB |
testcase_04 | AC | 20 ms
8,704 KB |
testcase_05 | AC | 20 ms
8,664 KB |
testcase_06 | AC | 20 ms
8,724 KB |
testcase_07 | AC | 674 ms
49,616 KB |
testcase_08 | AC | 25 ms
8,992 KB |
testcase_09 | AC | 19 ms
8,700 KB |
testcase_10 | AC | 19 ms
8,724 KB |
testcase_11 | AC | 19 ms
8,868 KB |
testcase_12 | AC | 20 ms
8,704 KB |
testcase_13 | AC | 55 ms
10,812 KB |
testcase_14 | AC | 27 ms
9,076 KB |
testcase_15 | AC | 21 ms
8,820 KB |
testcase_16 | AC | 31 ms
9,568 KB |
testcase_17 | AC | 21 ms
8,848 KB |
testcase_18 | AC | 355 ms
30,148 KB |
testcase_19 | AC | 72 ms
11,804 KB |
testcase_20 | AC | 36 ms
9,652 KB |
testcase_21 | AC | 145 ms
16,556 KB |
testcase_22 | AC | 24 ms
9,056 KB |
ソースコード
import sys sys.setrecursionlimit(10 ** 6) from bisect import * from collections import * from heapq import * int1 = lambda x: int(x) - 1 p2D = lambda x: print(*x, sep="\n") def II(): return int(sys.stdin.readline()) def MI(): return map(int, sys.stdin.readline().split()) def MI1(): return map(int1, sys.stdin.readline().split()) def MF(): return map(float, sys.stdin.readline().split()) def LI(): return list(map(int, sys.stdin.readline().split())) def LI1(): return list(map(int1, sys.stdin.readline().split())) def LF(): return list(map(float, sys.stdin.readline().split())) def LLI(rows_number): return [LI() for _ in range(rows_number)] dij = [(0, 1), (1, 0), (0, -1), (-1, 0)] def main(): n = II() aa = LI() c0 = c1 = c2 = 0 for a in aa: if a == 0: c0 += 1 if a == 1: c1 += 1 if a == 2: c2 += 1 # dp[i][j][k]…0枚なのがi種類、1枚なのがj種類、2枚なのがk種類のとき、あと何回引くかの期待値 dp = [[[0] * (c0 + c1 + c2 + 1) for _ in range(c0 + c1 + 1)] for _ in range(c0 + 1)] for i in range(c0 + 1): for j in range(c0 + c1 + 1): for k in range(c0 + c1 + c2 + 1): if (i, j, k) == (0, 0, 0): continue v = n if i - 1 >= 0 and j + 1 <= c0 + c1: v += dp[i - 1][j + 1][k] * i if j - 1 >= 0 and k + 1 <= c0 + c1 + c2: v += dp[i][j - 1][k + 1] * j if k - 1 >= 0: v += dp[i][j][k - 1] * k dp[i][j][k] = v / (i + j + k) print(dp[c0][c1][c2]) main() # dp[7][5][3] dp[6][6][3] dp[7][4][4] dp[7][5][2]