【leetcode354】. 俄罗斯套娃信封问题

给定一些标记了宽度和高度的信封，宽度和高度以整数对形式 (w, h) 出现。当另一个信封的宽度和高度都比这个信封大的时候，这个信封就可以放进另一个信封里，如同俄罗斯套娃一样。

请计算最多能有多少个信封能组成一组“俄罗斯套娃”信封（即可以把一个信封放到另一个信封里面）。

说明:
不允许旋转信封。

示例:

输入: envelopes = [[5,4],[6,4],[6,7],[2,3]]
输出: 3 
解释: 最多信封的个数为 3, 组合为: [2,3] => [5,4] => [6,7]。

 

把宽度和高度分开看，就是LIS（最长递增子序列问题），问题是把两者结合在一起。

首先对问题进行预处理，针对宽度进行升序排序， 针对高进行升序排序，但在相同宽度时，使用降序。

第二步，把高存到数组中，就是经典的LIS（最长递增子序列）问题。

 

package problem0354

import "sort"

func maxEnvelopes(e [][]int) int {
	if len(e) <= 1 {
		return len(e)
	}
    // 第一步，进行预处理，针对宽进行升序排序，针对高进行升序排序，但是在相同宽度时使用降序
	sort.Sort(sortedEnvelopes(e))

	tails := make([]int, len(e))

	for i := 0; i < len(e); i++ {
		tails[i] =  e[i][1]
	}
	// 第二步，把高存到数组中，就是经典的LIS（最长递增子序列）问题
	return lengthOfLIS(tails)
}

func lengthOfLIS(nums []int) int {
	dp := make([]int, len(nums))
	// 数组初始化值都为1
	for i := 0; i < len(nums); i++ {
		dp[i] = 1
	}
	res := 0
	for i := 0; i< len(nums); i++ {
		for j := 0; j < i; j++ {
			if nums[j] < nums[i] {
				dp[i] = max(dp[i], dp[j]+1)
			}
		}
		res = max(res, dp[i])
	}

	return res
}

func max(i int, i2 int) int {
	if i > i2 {
		return i
	} else {
		return i2
	}
}

type sortedEnvelopes [][]int

func (s sortedEnvelopes) Len() int {
	return len(s)
}

func (s sortedEnvelopes) Less(i, j int) bool {
	// 宽度相同的情况下，根据高度进行降序排序
	if s[i][0] == s[j][0] {
		// 同样宽度时，矮信封排在后面
		return s[i][1] > s[j][1]
	}
	return s[i][0] < s[j][0]
}

func (s sortedEnvelopes) Swap(i, j int) {
	s[i], s[j] = s[j], s[i]
}