結果
問題 | No.155 生放送とBGM |
ユーザー | d01phn |
提出日時 | 2021-02-02 12:58:21 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,188 bytes |
コンパイル時間 | 84 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 62,316 KB |
最終ジャッジ日時 | 2024-06-29 23:39:56 |
合計ジャッジ時間 | 14,599 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | TLE | - |
testcase_01 | -- | - |
testcase_02 | -- | - |
testcase_03 | -- | - |
testcase_04 | -- | - |
testcase_05 | -- | - |
testcase_06 | -- | - |
testcase_07 | -- | - |
testcase_08 | -- | - |
testcase_09 | -- | - |
testcase_10 | -- | - |
testcase_11 | -- | - |
testcase_12 | -- | - |
testcase_13 | -- | - |
testcase_14 | -- | - |
ソースコード
from math import comb import numpy as np def str2int(str): mm = int(str[:2]) ss = int(str[3:]) return 60 * mm + ss n, l = map(int, input().split()) l *= 60 s = [0] * n str_s = list(input().split()) for i in range(n): s[i] = str2int(str_s[i]) ssm = 0 for si in s: ssm += si if ssm <= l: print(n) else: dp = np.zeros((n + 1, n + 1, l + 1)) for k in range(l + 1): dp[0][0][k] = 1 for i in range(n): for j in range(n + 1): for k in range(l + 1): dp[i + 1][j][k] += dp[i][j][k] if j - 1 >= 0 and k - s[i] >= 0: dp[i + 1][j][k] += dp[i][j - 1][k - s[i]] for i in range(n): for j in range(n + 1): for k in range(l + 1): dp[i][j][k] = dp[n][j][k] if j - 1 >= 0 and k - s[i] >= 0: dp[i][j][k] -= dp[i][j - 1][k - s[i]] ans = 0.0 for i in range(n): for j in range(n): if l - s[i] - 1 < 0: ans += dp[i][j][l - 1] * (j + 1) / n / comb(n - 1, j) else: ans += (dp[i][j][l - 1] - dp[i][j][l - s[i] - 1]) * (j + 1) / n / comb(n - 1, j) print(ans)