题目描述
354. 俄罗斯套娃信封问题
给定一些标记了宽度和高度的信封,宽度和高度以整数对形式 (w, h) 出现。当另一个信封的宽度和高度都比这个信封大的时候,这个信封就可以放进另一个信封里,如同俄罗斯套娃一样。
请计算最多能有多少个信封能组成一组“俄罗斯套娃”信封(即可以把一个信封放到另一个信封里面)。
说明:
不允许旋转信封。
示例:
输入: envelopes = [[5,4],[6,4],[6,7],[2,3]]
输出: 3
解释: 最多信封的个数为 3, 组合为: [2,3] => [5,4] => [6,7]。
解题方法
状态定义:
dp[i] 为前i个元素以第i个元素结尾时的状态(最多组成几个套娃)。
尝试1
class Solution:
def maxEnvelopes(self, envelopes: List[List[int]]) -> int:
if len(envelopes) <= 1:
return len(envelopes)
envelopes.sort(key=lambda item: item[0])
dp = [1 for i in range(len(envelopes))]
max_env = 0
for i in range(1,len(envelopes)):
max_i = 1
for j in range(0,i):
if envelopes[i][0] > envelopes[j][0] and envelopes[i][1] > envelopes[j][1]:
max_i = max(max_i,dp[j]+1)
dp[i] = max_i
print(dp)
return max(dp)
运行结果:超时
尝试2:超时的改进
尝试1 O ( n 2 ) O(n^2) O(n2)时间复杂度,考虑到LIS问题的改进nlogn算法,dp[i] 为构成的i个上升序列中最后一个 元素的最小值。
- 排序:二分法查找时,对于二维的情况不好比较判断,应对信封排序,将其变成一个一维的LIS问题。排序算法尽可能将小的信封放在前面。
- 只是简单的按照x[0]排序还不够,对于x[0]相等的元素继续按照x[1]降序排序。降序的作用是算法上升子序列中x[0]相等的元素只挑选其中一个。
import functools
class Solution:
def maxEnvelopes(self, envelopes: List[List[int]]) -> int:
if len(envelopes) <= 0:
return len(envelopes)
cmp = lambda x,y:x[0]-y[0] if x[0] != y[0] else y[1] - x[1]
envelopes.sort(key= functools.cmp_to_key(cmp))
print(envelopes)
dp = [envelopes[0]]
for i in range(1, len(envelopes)):
left,right = 0, len(dp)
while left < right:
mid = (left + right) // 2
if envelopes[i][1] > dp[mid][1]:
left = mid + 1
else:
right = mid
if len(dp) == left:
dp.append(envelopes[i])
else:
dp[left] = envelopes[i]
return len(dp)
总结
- 之前对于LIS和最大子序和问题定义的状态理解有偏差。两者dp[i]都是代表以第i个元素结尾是的子序或者子串的状态。
- 最大子序和中没有明确指定i元素之前的元素的起始位置以及都是哪几个元素的组合
- 子串问题中指定了子串的末位置i,没有明确指定子串的起始位置。