英文原文
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, and - Serve
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
中文题目
有 A 和 B 两种类型的汤。一开始每种类型的汤有 N
毫升。有四种分配操作:
- 提供 100ml 的汤A 和 0ml 的汤B。
- 提供 75ml 的汤A 和 25ml 的汤B。
- 提供 50ml 的汤A 和 50ml 的汤B。
- 提供 25ml 的汤A 和 75ml 的汤B。
当我们把汤分配给某人之后,汤就没有了。每个回合,我们将从四种概率同为0.25的操作中进行分配选择。如果汤的剩余量不足以完成某次操作,我们将尽可能分配。当两种类型的汤都分配完时,停止操作。
注意不存在先分配100 ml汤B的操作。
需要返回的值: 汤A先分配完的概率 + 汤A和汤B同时分配完的概率 / 2。
示例: 输入: N = 50 输出: 0.625 解释: 如果我们选择前两个操作,A将首先变为空。对于第三个操作,A和B会同时变为空。对于第四个操作,B将首先变为空。 所以A变为空的总概率加上A和B同时变为空的概率的一半是 0.25 *(1 + 1 + 0.5 + 0)= 0.625。
注释:
0 <= N <= 10^9
。-
返回值在
10^-6
的范围将被认为是正确的。
通过代码
官方题解
动态规划:
首先,由于四种分配操作都是 25
的倍数,因此我们可以将 N
除以 25
(如果有余数,则补 1
),并将分配操作变为 (4, 0)
,(3, 1)
,(2, 2)
和 (1, 3)
。
当 N
较小时,我们可以用动态规划来解决这个问题。设 dp(i, j)
表示汤 A
和汤 B
分别剩下 i
和 j
份时,所求的概率值。状态转移方程为:
dp(i, j) = 1/4 * (dp(i - 4, y) + dp(i - 3, y - 1) + dp(i - 2, y - 2) + dp(i - 1, y - 3))
边界条件为:
dp(i, j) = 0.5 if i <= 0 and j <= 0
dp(i, j) = 1.0 if i <= 0 and j > 0
dp(i, j) = 0.0 if i > 0 and j <= 0
即如果同时分配完(边界条件中的第一行),概率值为 1.0
的一半即为 0.5
;如果汤 A
先分配完,概率值为 1.0
;如果汤 B
先分配完,概率值为 0.0
。
这个动态规划的时间复杂度是 $O(N^2)$,即使将 N
除以 25
之后,仍然没法在短时间内得到答案,因此我们需要尝试一些别的思路。可以发现,分配操作有 (4, 0)
,(3, 1)
,(2, 2)
和 (1, 3)
四种,那么在一次分配中,汤 A
平均会少 (4 + 3 + 2 + 1) / 4 = 2.5
份,汤 B
平均会少 (0 + 1 + 2 + 3) / 4 = 1.5
份。因此在 N
非常大的时候,A
会有很大的概率比 B
先分配完,所有概率应该非常接近 1
。事实上,当 N >= 500 * 25
时,所求概率已经大于 0.999999
了(可以通过上面的动态规划方法求出),它和 1
的误差(无论是绝对误差还是相对误差)都小于 10^-6
。因此在 N >= 500 * 25
时,我们只需要输出 1
作为答案即可。在其它的情况下,我们使用动态规划计算出答案。
class Solution {
public double soupServings(int N) {
N = N/25 + (N%25 > 0 ? 1 : 0);
if (N >= 500) return 1.0;
double[][] memo = new double[N+1][N+1];
for (int s = 0; s <= 2*N; ++s) {
for (int i = 0; i <= N; ++i) {
int j = s-i;
if (j < 0 || j > N) continue;
double ans = 0.0;
if (i == 0) ans = 1.0;
if (i == 0 && j == 0) ans = 0.5;
if (i > 0 && j > 0) {
ans = 0.25 * (memo[M(i-4)][j] + memo[M(i-3)][M(j-1)] +
memo[M(i-2)][M(j-2)] + memo[M(i-1)][M(j-3)]);
}
memo[i][j] = ans;
}
}
return memo[N][N];
}
public int M(int x) { return Math.max(0, x); }
}
class Solution(object):
def soupServings(self, N):
Q, R = divmod(N, 25)
N = Q + (R > 0)
if N >= 500: return 1
memo = {}
def dp(x, y):
if (x, y) not in memo:
if x <= 0 or y <= 0:
ans = 0.5 if x<=0 and y<=0 else 1.0 if x<=0 else 0.0
else:
ans = 0.25 * (dp(x-4,y)+dp(x-3,y-1)+dp(x-2,y-2)+dp(x-1,y-3))
memo[x, y] = ans
return memo[x, y]
return dp(N, N)
复杂度分析
时间复杂度:$O(1)$,因为存在常数 $C$,使得当 $N > C$ 时,所求的概率和
1
的误差在10^-6
以内。空间复杂度:$O(1)$,原因同上。
统计信息
通过次数 | 提交次数 | AC比率 |
---|---|---|
4587 | 9509 | 48.2% |
提交历史
提交时间 | 提交结果 | 执行时间 | 内存消耗 | 语言 |
---|