Leetcode 799. 香槟塔 C++

Leetcode 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]。

题解

动态规划
先假设杯子的容量无限大，一次性将poured的量导入杯子中，然后根据对半进行流淌

dp[i][j]表示第i+1层第j+1个杯子的香槟量,其中第i+1层应该有i+1个杯子
转移方程为，

• j=0时 dp[i][j] = dp[i-1][j]/2，要保证dp[i-1][j]>1
• j=i时,dp[i][j] = dp[i-1][i-1]/2，要保证dp[i-1][j-1]>1
• 其他情况，dp[i][j] = (dp[i-1][j-1]+dp[i-1][j])/2，需要注意的是只有满的那个杯子才能向外流

初始条件，dp[0][0] = poured，也就是香槟全倒进第1层第1个杯子中。详细过程见代码

代码

	double champagneTower(int poured, int query_row, int query_glass) {
        vector<vector<double>> dp(query_row+1,vector<double>(query_row+1,0));
        dp[0][0] = poured;
        for(int i=1; i<=query_row; i++){
            int j=0;
            if(dp[i-1][j] > 1)		//只有满了才能向外流
                dp[i][j] = (dp[i-1][j]-1)/2;
            for(j=1; j<i; j++){
                if(dp[i-1][j-1]>1)
                    dp[i][j] = (dp[i-1][j-1]-1)/2;
                if(dp[i-1][j]>1)
                    dp[i][j] += (dp[i-1][j]-1)/2;
            }
            if(dp[i-1][j-1]>1)
                dp[i][j] = (dp[i-1][j-1]-1)/2;
        }
        return min(dp[query_row][query_glass],1.0);			//如果杯子满了，它应该向外流
    }

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