有個馬戲團正在設計疊羅漢的表演節目,一個人要站在另一人的肩膀上。出於實際和美觀的考慮,在上面的人要比下面的人矮一點且輕一點。已知馬戲團每個人的身高和體重,請編寫代碼計算疊羅漢最多能疊幾個人。
示例:
輸入: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