力扣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 的范围将被认为是正确的。

---

解题思路

四种分配方式的AB分配量都是25的整倍数，为了便于计算，将其数据全都除以25，即：

1. 4A+0B
2. 3A+1B
3. 2A+2B
4. 1A+3B

而对于输入的数据n也要进行相应的处理，即对其除以25，若不能整除，则多出来的部分以1计入，即：

n=(n+24)/25;

定义函数gailv(int a,int b)，传入的参数a和b分别代表汤A和汤B的剩余量，起初将n赋值给a和b代入函数。
因为每一次选择后，只要AB汤都还有剩，就要继续从等概率的四种分配方式中选择其中一种继续下去，以此循环，所以用递归的方式解题。
当我想到这里并把代码提交后发现超出时间限制，原因可能是该递归运算涉及的数据过多导致性能下降。
于是再定义了一个二维数组用于缓存中间计算结果，提高性能。
修改好代码再次提交后发现执行错误，显示最后执行的输入为660295675，于是思考着，当n足够大时，输出是趋近于1的，所以我测试了多个数据，最终发现当n>4800时，输出为1。因此，当n>4800时，直接返回1，极大地提高了效率。

---

代码

double cache[5000][5000];
double gailv(int a,int b){
    double hc;
    if(a<=0){
        if(b>0)return 1;
        else return 0.5;
    }else if(b<=0)return 0;
    if(cache[a][b])return cache[a][b];
    hc=(0.25*(gailv(a-4,b)+gailv(a-3,b-1)+gailv(a-2,b-2)+gailv(a-1,b-3)));
    cache[a][b]=hc;
    return hc;
}
double soupServings(int n){
    if(n>4800)return 1;
    n=(n+24)/25;
    return gailv(n,n);
}

链接：https://leetcode-cn.com/problems/soup-servings/solution/li-kou-808fen-tang-by-competent-mccarthy-kfz8/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。