我们把玻璃杯摆成金字塔的形状,其中第一层有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]
。
这个就比较简单了,先初始化一个二维数组,然后计算数组中每个数的值,返回相应的值就好
class Solution:
def champagneTower(self, poured, query_row, query_glass):
"""
:type poured: int
:type query_row: int
:type query_glass: int
:rtype: float
"""
a= [[0 for i in range(102)] for i in range(103)]//创建数组,多处一行两列为辅助数组
a[1][1]=poured
for i in range(2,query_row+2):
for j in range(1,i+1)://利用下一个杯子水的比重与它上面两个杯子有关
if a[i-1][j]==0 or a[i-1][j-1]==0:
a[i][j]=(a[i-1][j]+a[i-1][j-1]-1)/2
elif a[i-1][j]<1 and a[i-1][j]!=0:
a[i][j]=(a[i-1][j-1]-1)/2
elif a[i-1][j-1]<1 and a[i-1][j-1]!=0:
a[i][j]=(a[i-1][j]-1)/2
else:
a[i][j]=(a[i-1][j]+a[i-1][j-1]-2)/2
if a[query_row+1][query_glass+1]>=1:
return 1
elif a[query_row+1][query_glass+1]<=0:
return 0
else:
return a[query_row+1][query_glass+1]
这是第一次写的,感觉比较蠢,创建了一行两列辅助数组,后来又换了个思路
class Solution:
def champagneTower(self, poured, query_row, query_glass):
"""
:type poured: int
:type query_row: int
:type query_glass: int
:rtype: float
"""
tower = [[0]*(i+1) for i in range(101)]
tower[0][0] = poured
for i in range(query_row+1):
for j in range(i+1):
if tower[i][j] > 1:
remain = (tower[i][j] - 1) / 2.0
tower[i+1][j] += remain
tower[i+1][j+1] += remain
tower[i][j] = 1
return tower[query_row][query_glass]
与第一种方法相比,就是反过来思考,每个上行杯子的水减去自身饱和的部分会平分给下面两个