leetcode题：799. 香槟塔（中等）

最新推荐文章于 2024-01-25 17:51:30 发布

tankpanv

最新推荐文章于 2024-01-25 17:51:30 发布

阅读量278

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/tankpanv/article/details/100046231

版权

leetcode 专栏收录该内容

133 篇文章 0 订阅

订阅专栏

一、题目描述：799. 香槟塔（中等）

我们把玻璃杯摆成金字塔的形状，其中第一层有1个玻璃杯，第二层有2个，依次类推到第100层，每个玻璃杯(250ml)将盛有香槟。

从顶层的第一个玻璃杯开始倾倒一些香槟，当顶层的杯子满了，任何溢出的香槟都会立刻等流量的流向左右两侧的玻璃杯。当左右两边的杯子也满了，就会等流量的流向它们左右两边的杯子，依次类推。（当最底层的玻璃杯满了，香槟会流到地板上）

例如，在倾倒一杯香槟后，最顶层的玻璃杯满了。倾倒了两杯香槟后，第二层的两个玻璃杯各自盛放一半的香槟。在倒三杯香槟后，第二层的香槟满了 - 此时总共有三个满的玻璃杯。在倒第四杯后，第三层中间的玻璃杯盛放了一半的香槟，他两边的玻璃杯各自盛放了四分之一的香槟，如下图所示。

现在当倾倒了非负整数杯香槟后，返回第 i 行 j 个玻璃杯所盛放的香槟占玻璃杯容积的比例（i 和 j都从0开始）。

示例 1:
输入: poured(倾倒香槟总杯数) = 1, query_glass(杯子的位置数) = 1, query_row(行数) = 1
输出: 0.0
解释: 我们在顶层（下标是（0，0））倒了一杯香槟后，没有溢出，因此所有在顶层以下的玻璃杯都是空的。

示例 2:
输入: poured(倾倒香槟总杯数) = 2, query_glass(杯子的位置数) = 1, query_row(行数) = 1
输出: 0.5
解释: 我们在顶层（下标是（0，0）倒了两杯香槟后，有一杯量的香槟将从顶层溢出，位于（1，0）的玻璃杯和（1，1）的玻璃杯平分了这一杯香槟，所以每个玻璃杯有一半的香槟。
注意:

poured 的范围[0, 10 ^ 9]。
query_glass 和query_row 的范围 [0, 99]。

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

二、解题思路

方法一、动态规划

1、每个杯子满了之后，会往两边溢出。即对每第i，j个杯子满了之后,左下方杯子dp[i+1[j] = (dp[i][j]-1)/2 + dp[i+1][j]，右下方杯子dp[i+1[j+1] = (dp[i][j]-1)/2 + dp[i+1][j+1]。dp[i][j]记录的是总杯数的香槟倒下来之后，流过第i，j个杯子的酒的数量，比如dp[0][0]=n;

2、初始化，所有dp[i][j]=0，然后dp[0][0]=n;从上往下遍历每个杯子，对每个杯子统计流到左下方和右下方的杯子的酒的总数量。当遍历完之后。dp[i][j]就是流到i，j个杯子的旧的数量。如果大于等于1则返回1，如果小于1则返回dp[i][j]

方法二、递归

方法其实和第一个类似，就是递归获取dp[i][j] = （dp[i-1][j]-1）/2 + (dp[i-1][j-1]-1)/2。因为第i，j个杯子的酒只有从左上方和右上方流下来，只要递归获取上层所有左上方和右上方的杯子留下来的酒就可以。这里减1是因为上层的杯子装满了之后才会溢出。

三、代码

class Solution {
public:
    double champagneTower(int poured, int query_row, int query_glass) {
      vector<vector<float>> dp;
      for(int i = 0;i <= query_row; i ++)
      {
        dp.push_back(vector<float>(i+1,0));
      }
      dp[0][0] = poured;
      
      for(int i =0; i<= query_row-1; i ++)
      {
        for(int j = 0; j < dp[i].size(); j++)
        { 
          //cout<<"*****"<<endl;
          if(dp[i][j] > 1)
          {
            dp[i+1][j] = dp[i+1][j]+ (dp[i][j]-1)/2;
            dp[i+1][j+1] = dp[i+1][j+1]+ (dp[i][j]-1)/2;
          }
          //cout<<"i="<<i<<"__"<<"j="<<j<<",dp[i][j]="<<dp[i][j]<<",";
          //cout<<endl;
        }
      }
     // print(dp);
      if(dp[query_row][query_glass] > 1)
        return 1;
      return dp[query_row][query_glass];
    }
  void print(vector<vector<float>> & dp)
  {
    for(int i = 0; i< dp.size();i++)
    {
      for(int j = 0; j < dp[i].size(); j++)
      {
        cout<<dp[i][j]<<",";
      }
      cout<<endl;
    }
  }
};