有个马戏团正在设计叠罗汉的表演节目,一个人要站在另一人的肩膀上。出于实际和美观的考虑,在上面的人要比下面的人矮一点且轻一点。已知马戏团每个人的身高和体重,请编写代码计算叠罗汉最多能叠几个人。
示例:
输入:height = [65,70,56,75,60,68] weight = [100,150,90,190,95,110]
输出:6
解释:从上往下数,叠罗汉最多能叠 6 层:(56,90), (60,95), (65,100), (68,110), (70,150), (75,190)
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/circus-tower-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
——————————————
解题思路:因为要考虑相互之间的大小关系,所以得基于体重和身高进行排序。可以选择从大到小排序,也可以选择从小到大排序。当选择基于身高按从大到小排序之后,可以基于体重将问题变为求解体重的最长上升子序列的动态规划问题。
这里需要记录的一个点是,基于zip将体重和身高放到一个列表中,然后使用列表的sort()函数对列表进行排序,其中key规定了排序的条件。代码中nums.sort(key=lambda x:(x[0],-x[1]))表示将身高从小到大排序,当遇到身高一样的几组数据时,将这几组数据按照体重从大到小的顺序进行排序。
这里注意当身高相同时将按照体重从大到小排序,这样做的原因是为了方便求解最长的上升序列。
在得到排序后的数据之后基于贪心算法和二分查找寻找数据中的最小体重上升子序列。
class Solution:
def bestSeqAtIndex(self, height: List[int], weight: List[int]) -> int:
nums = list(zip(height,weight))
nums.sort(key=lambda x:(x[0],-x[1]))
print(nums)
Num = []
res = 0
for n in nums:
if not Num or n[1]>Num[-1][1]:
Num.append(n)
res += 1
else:
i,j = 0,res-1
while i<j:
mid = (i+j)//2
if n[1]<=Num[mid][1]:
j = mid
else:
i = mid + 1
Num[j] = n
return res