【Leetcode】808. Soup Servings

题目地址：

https://leetcode.com/problems/soup-servings/description/

有两种汤：$A$型和$B$型。初始时有$n$毫升的每种类型的汤。有四种操作方式：
1、提供$100$毫升$A$型汤和$0$毫升$B$型汤；
2、提供$75$毫升$A$型汤和$25$毫升$B$型汤；
3、提供$50$毫升$A$型汤和$50$毫升$B$型汤；
4、提供$25$毫升$A$型汤和$75$毫升$B$型汤。
我们将以$0.25$的相等概率从四种操作中选择一种。如果剩余的汤量不足以完成操作，将全部提供。一旦汤不够了就停止。返回$A$型汤首先用完的概率，加上$\frac{1}{2}$的A型和B型同时用完的概率。在实际答案的$10^{-5}$范围内的答案都可以接受。

思路是记忆化搜索，令$f(i,j)$表示当前还剩下$i,j$的数量的$A,B$的汤的情况下，$A$型汤首先用完的概率，加上$\frac{1}{2}$的A型和B型同时用完的概率。然后用全概率公式进行记忆化搜索。代码如下：

class Solution {
 public:
  using PII = pair<int, int>;
  double soupServings(int n) {
    if (n >= 4800) return 1;
    n = (n + 24) / 25;
    auto ha = [&](auto& p) {
      return hash<int>()(p.first) ^ hash<int>()(p.second);
    };
    unordered_map<PII, double, decltype(ha)> mp(0, ha);
    return dfs(n, n, mp);
  }

  double dfs(int x, int y, auto& mp) {
    if (x <= 0 && y <= 0) return 0.5;
    if (x <= 0) return 1;
    if (y <= 0) return 0;
    if (mp.count({x, y})) return mp[{x, y}];
    return mp[{x, y}] = (dfs(x - 4, y, mp) + dfs(x - 3, y - 1, mp) +
                         dfs(x - 2, y - 2, mp) + dfs(x - 1, y - 3, mp)) /
                        4.0;
  }
};

时空复杂度$O(n^2)$，实际时间不会这么多。