Problem
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:
Serve
100
ml of soup A and0
ml of soup B,Serve
75
ml of soup A and25
ml of soup B,Serve
50
ml of soup A and50
ml of soup B, andServe
25
ml of soup A and75
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 an equal probability 0.25
. If the remaining volume of soup is not enough to complete the operation, we will serve as much as possible. We stop once we no longer have some quantity of both types of soup.
Note that we do not have an 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**. Answers within 10-5
of the actual answer will be accepted.
Example 1:
Input: n = 50
Output: 0.62500
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.
Example 2:
Input: n = 100
Output: 0.71875
Constraints:
0 <= n <= 109
Solution
/**
* @param {number} n
* @return {number}
*/
var soupServings = function(n) {
var m = Math.ceil(n / 25);
var dp = {};
for (var i = 1; i <= m; i++) {
if (solve(i, i, dp) > (1 - Math.pow(10, -5))) {
return 1;
}
}
return solve(m, m, dp);
};
var solve = function(a, b, dp) {
if (a <= 0 && b <= 0) return 0.5;
if (a <= 0) return 1;
if (b <= 0) return 0;
var key = `${a}-${b}`;
if (dp[key] === undefined) {
dp[key] = 0.25 * solve(a - 4, b, dp)
+ 0.25 * solve(a - 3, b - 1, dp)
+ 0.25 * solve(a - 2, b - 2, dp)
+ 0.25 * solve(a - 1, b - 3, dp);
}
return dp[key];
};
Explain:
Top-down, DP.
Complexity:
- Time complexity : O(n).
- Space complexity : O(n).