Leetcode 808. 分汤 C++

Leetcode 808. 分汤

题目

有 A 和 B 两种类型的汤。一开始每种类型的汤有 N 毫升。有四种分配操作：

1. 提供 100ml 的汤A 和 0ml 的汤B。
2. 提供 75ml 的汤A 和 25ml 的汤B。
3. 提供 50ml 的汤A 和 50ml的汤B。
4. 提供 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 的范围将被认为是正确的。

题解

动态规划
dp[i][j]表示有i ml A、j ml B 时所求的概率
dp[i][j] = (dp[i-100][j]+dp[i-75][j-25]+dp[i-50][j-50]+dp[i-25][j-75])*0.25;

因为我们每次改变量都是25，不妨我们先将N/25，如果有余数则+1。同时注意不足时应该尽量分
详细过程见代码

代码

	double soupServings(int N) {
        if(N >= 500*25) return 1;		//N足够大时，概率为1
        N = (N/25) + ((N%25 > 0)? 1 : 0);		//按照25ml进行分；若不足25ml，应该尽可能的分，故+1
        vector<vector<double>> dp(N+1,vector<double>(N+1,0));
        dp[0][0] = 0.5;			//都没有，则A、B同时分完的概率为1，题目要求输出/2的结果
        for(int i=1; i<=N; i++)		//若A没要求分配，则A先分配完的概率是1
            dp[0][i] = 1;
        for(int i=1; i<=N; i++){
            for(int j=1; j<=N; j++){
                if(j>=3)    dp[i][j] += dp[i-1][j-3];		//可以按照第4种方案分
                else    dp[i][j] += dp[i-1][0];			//B不够按第4种方案分，故要尽可能的分，也就是分j份

                if(i>=2 && j>=2)    dp[i][j] += dp[i-2][j-2];		//按照第3种方案
                else if(i>=2)   dp[i][j] += dp[i-2][0];		//A不足，则尽可能的分
                else if(j>=2)   dp[i][j] += dp[0][j-2];		//B不足，则尽可能的分
                else    dp[i][j] += dp[0][0];			//A、B不足，则尽可能的分
                
                if(i>=3)    dp[i][j] += dp[i-3][j-1];		//按照第2种方案
                else    dp[i][j] += dp[0][j-1];			//A不足，则尽可能的分
                
                if(i>=4)    dp[i][j] += dp[i-4][j];		//按照第1种方案
                else    dp[i][j] += dp[0][j];		//A不足，则尽可能的分
                dp[i][j] *= 0.25;
            }
        }
        return dp[N][N];
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/soup-servings
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。