题目地址:
https://leetcode.com/problems/russian-doll-envelopes/
给定一个由长度为 2 2 2的数组组成的数组 A A A,要求找到最长链 [ ( a 0 , b 0 ) , ( a 1 , b 1 ) , ( a 2 , b 2 ) , . . . , ( a k , b k ) ] [(a_0,b_0),(a_1,b_1),(a_2,b_2),...,(a_k,b_k)] [(a0,b0),(a1,b1),(a2,b2),...,(ak,bk)]使得 a i < a i + 1 a_i<a_{i+1} ai<ai+1并且 b i < b i + 1 b_i<b_{i+1} bi<bi+1。返回这个最长链的长度。
最优解可以达到 O ( n log n ) O(n\log n) O(nlogn)的时间复杂度,可以参考https://blog.csdn.net/qq_46105170/article/details/108661086。以下介绍一个 O ( n 2 ) O(n^2) O(n2)的动态规划解法。
先将 A A A按照每个数组的第二位由小到大排序,这样如果一个pair接在另一个前面,那么排完序后,在 A A A里它也一定接在另一个前面。这就转化为了一维最长上升子序列问题,可以用动态规划来做。设 f [ i ] f[i] f[i]是以 A [ i ] A[i] A[i]结尾的最长上升子序列的长度,那么 f [ i ] = 1 + max A [ j ] < A [ i ] { f [ j ] } f[i]=1+\max_{A[j]<A[i]}\{f[j]\} f[i]=1+A[j]<A[i]max{f[j]}只需要注意,在判断大小的时候,数组的两个数字都要比较。代码如下:
import java.util.Arrays;
public class Solution {
/**
* @param envelopes: a number of envelopes with widths and heights
* @return: the maximum number of envelopes
*/
public int maxEnvelopes(int[][] envelopes) {
// write your code here
Arrays.sort(envelopes, (e1, e2) -> e1[1] != e2[1] ? Integer.compare(e1[1], e2[1]) : Integer.compare(e1[0], e2[0]));
int[] dp = new int[envelopes.length];
int res = 0;
for (int i = 0; i < envelopes.length; i++) {
dp[i] = 1;
for (int j = 0; j < i; j++) {
if (envelopes[j][0] < envelopes[i][0] && envelopes[j][1] < envelopes[i][1]) {
dp[i] = Math.max(dp[i], 1 + dp[j]);
}
}
res = Math.max(res, dp[i]);
}
return res;
}
}
时间复杂度 O ( n 2 ) O(n^2) O(n2),空间 O ( n ) O(n) O(n)。