力扣(LeetCode)808. 分汤(C++)

清墨韵染

已于 2022-11-21 10:54:26 修改

阅读量298

点赞数 2

分类专栏：墨染leetcode 文章标签： leetcode c++ 算法动态规划 dp

于 2022-11-21 10:41:17 首次发布

本文链接：https://blog.csdn.net/Innocence02/article/details/127958296

版权

墨染leetcode 专栏收录该内容

138 篇文章 2 订阅

订阅专栏

动态规划

在这里插入图片描述

如图，本题的状态表示，是二维 $d p$ $f [i, j]$ ， $i$ 表示剩余的 $a$ ， $j$ 表示剩余的 $b$ ， $f [i, j]$ 表示 $a$ 先取完的概率。

按照 $i / j$ 的剩余数量做集合划分
①当 $i\le 0,j\le0$ ， $a / b$ 同时取完，线性概率 $f[i][j]=1.0\div2=0.5$
②当 $i\le 0,j>0$ ， $a$ 先取完， $f [i] [j] = 1.0$
③当 $0,j\le0$ ， $b$ 先取完， $f [i] [j] = 0.0$
④当 $i > 0, j > 0$ ， $a / b$ 都有剩余，有均等概率进行 $4$ 种取法，为了便于操作，将所有取法的取值 $÷25 \div 25$ ，那么初始汤量 $n\div25$ ， $f [i] [j] = (f [i - 4] [j] + f [i - 3] [j - 1] + f [i - 2] [j - 2] + f [i - 1] [j - 3]) / 4$

状态转移方程
$\begin{cases} 0.5&\text{if }i\le 0,j\le0\\ 1.0&\text{if }i\le 0,j>0\\ 0.0&\text{if }i> 0,j\le0\\ (f[i-4][j]+f[i-3][j-1]+f[i-2][j-2]+f[i-1][j-3])\div4 &\text{if }i> 0,j>0 \end{cases}$

数学期望 : 取 $a=(4+3+2+1)\div4=2.5$ 取 $b=(3+2+1+0)\div 4=1.5$ ，数学期望 $a > b$ ，当 $n$ 很大时， $a$ 先取完的概率趋于 $1$ 。经测试， $n\div 25 \ge 189$ 时， $a$ 先取完的概率近似为 $1$

朴素dp

class Solution {
public:
    int g(int x){//分汤后小于0，归为0
        return max(0,x);
    }
    double soupServings(int n) {
        // n = (n+24)/25;//向上取整
        n = n / 25 + (n%25!=0);
        if(n>=189) return 1;
        vector<vector<double>> f(n+1,vector<double>(n+1));
        for(int i = 0;i<=n;i++)
            for(int j = 0;j<=n;j++){
                if(!i&&!j) f[i][j] = 0.5;
                else if(!i&&j) f[i][j] = 1.0;
                else if (i&&!j) f[i][j] = 0.0;
                else{//(i&&j)
                    f[i][j] = (f[g(i-4)][j]+f[g(i-3)][g(j-1)]+f[g(i-2)][g(j-2)]+f[g(i-1)][g(j-3)])/4;
                }
            }
        return f[n][n];
    }
};

时间复杂度 : $O(n^2)$ ，状态转移的时间复杂度 $O(n^2)$ 。
空间复杂度 : $O(n^2)$ ， $f$ 数组的空间复杂度 $O(n^2)$ 。