输入为3个整数，分别对应N，K，W，中间用空格隔开
其中0 <= K <= N <= 10000，1 <= W <= 10000
输出为概率值，保留5位小数
21 17 10
0.73278
class Solution: def __init__(self): self.res = 0 self.count = 0 def solution(self, n, k, w): def helper(i, u): if i >= k: self.count += u if i <= n: self.res += u return for j in range(1, w+1): helper(i+j, u/w) helper(0, 10000) ans = self.res/self.count print(round(ans, 5)) if __name__ == "__main__": n, k, w = map(int, input().split()) case = Solution() case.solution(n,k,w)
if __name__=='__main__': n, k, w = list(map(int, input().split())) dp = [0]*(k+w) for i in range(k, n+1): dp[i] = 1 s = min(w, n-k+1) for i in range(k-1, -1, -1): dp[i] = s / w s += dp[i] - dp[i+w] print(round(dp[0],5))
扫描二维码，关注牛客网
下载牛客APP，随时随地刷题
扫一扫，把题目装进口袋
扫描二维码，进入QQ群
扫描二维码，关注牛客公众号
京公网安备
11010502036488号
