[LeetCode]Soup Servings

题目描述:

LeetCode 808. Soup Servings

There are two types of soup: type A and type B. Initially we have N ml of each type of soup. There are four kinds of operations:

  1. Serve 100 ml of soup A and 0 ml of soup B
  2. Serve 75 ml of soup A and 25 ml of soup B
  3. Serve 50 ml of soup A and 50 ml of soup B
  4. Serve 25 ml of soup A and 75 ml of soup B

When we serve some soup, we give it to someone and we no longer have it.  Each turn, we will choose from the four operations with equal probability 0.25. If the remaining volume of soup is not enough to complete the operation, we will serve as much as we can.  We stop once we no longer have some quantity of both types of soup.

Note that we do not have the operation where all 100 ml's of soup B are used first.  

Return the probability that soup A will be empty first, plus half the probability that A and B become empty at the same time.

Example:
Input: N = 50
Output: 0.625
Explanation: 
If we choose the first two operations, A will become empty first. For the third operation, A and B will become empty at the same time. For the fourth operation, B will become empty first. So the total probability of A becoming empty first plus half the probability that A and B become empty at the same time, is 0.25 * (1 + 1 + 0.5 + 0) = 0.625.

Notes:

  • 0 <= N <= 10^9
  • Answers within 10^-6 of the true value will be accepted as correct.

题目大意:

有A,B两种汤。初始每种汤各有N毫升,现有4种操作:

1. A倒出100ml,B倒出0ml
2. A倒出75ml, B倒出25ml
3. A倒出50ml, B倒出50ml
4. A倒出25ml, B倒出75ml

每种操作的概率均等为0.25。如果汤的剩余容量不足完成某次操作,则有多少倒多少。当每一种汤都倒完时停止操作。

求A先倒完的概率,加上A和B同时倒完的概率。

解题思路:

记忆化搜索 + 特判

if A <= 0 and B <= 0: dp[A][B] = 0.5 

elif A <= 0: dp[A][B] = 1 

elif B <= 0: dp[A][B] = 0

else: dp[A][B] = 0.25 * (dp[A - 100][B] + dp[A - 75][B - 25] + dp[A - 50][B - 50] + dp[A - 25][B - 75])

当N >= 14000时,近似等于1,直接返回1即可

Python代码:

class Solution(object):
    def soupServings(self, N):
        """
        :type N: int
        :rtype: float
        """
        self.memo = {}
        if N >= 14000: return 1.0
        return self.solve(N, N)
    def solve(self, A, B):
        if A <= 0 and B <= 0: return 0.5
        if A <= 0: return 1
        if B <= 0: return 0
        if (A, B) in self.memo:
            return self.memo[(A, B)]
        ans = 0.25 * (self.solve(A - 100, B) + self.solve(A - 75, B - 25) + 
                      self.solve(A - 50, B - 50) + self.solve(A - 25, B - 75))
        self.memo[(A, B)] = ans
        return ans

 

本文链接:http://bookshadow.com/weblog/2018/04/02/leetcode-soup-servings/
请尊重作者的劳动成果,转载请注明出处!书影博客保留对文章的所有权利。

如果您喜欢这篇博文,欢迎您捐赠书影博客: ,查看支付宝二维码

Pingbacks已关闭。

暂无评论

张贴您的评论