描述
给定数组arr,设长度为n,输出arr的最长递增子序列。(如果有多个答案,请输出其中 按数值(注:区别于按单个字符的ASCII码值)进行比较的 字典序最小的那个)
困惑:如何用二分查找找到第一个大于目标值T的数值的下标
分析
- 结尾数组end[],end[i]表示长度是i的最长子序列的末尾最小数值,end[i]的初始值是arr[1],记录长度的数组dp[],dp[i]表示arr[i]为结尾的最长递增子序列的长度。
- 遍历数组arr[],若当前值arr[i]大于end[i]时,则当前值赋值给end[i+1],并且end数组结尾数字个数赋值给dp[i]。
- 若当前值arr[i]小于等于end[]数组最后面的数值时,用二分查找方法找到end[]数组中等于或第一个大于当前值的数值,替换这个找到的数值,dp[i]取这个数值的下标。
- 遍历结束后,从后向前遍历dp[]数组,找dp[i]与最长子序列长度一一匹配的下表i(dp[i]与最长子序列长度都每次递减一),用这个下表i获取arr[i],最终汇成字典序列最小的最长子序列。
- 思想是让end[]中存储比较小的元素。这样,end[]未必是真实的最长上升子序列,但长度是对的。也就是说,end[]是一个单调数组,这样就可以使用二分查找的快速明确以当前数值arr[i]为结尾的最长子序列的长度。
import java.util.*;
public class Solution {
public int[] LIS (int[] arr) {
int n = arr.length;
int[] dp = new int[n+1];//以arr[i]为结尾的最长递增子序列的长度
int[] end = new int[n+1];//end[i]指长度是i的最长子序列的末尾最小数值
int len = 1;
dp[0] = 1;
end[1] = arr[0];
for(int i = 0; i < n; i++){
if(arr[i] > end[len]){
end[++len] = arr[i];
dp[i] = len;
}else{
int left = 1, right = len;
while(left <= right){
int mid = (right - left) / 2 + left;
if(end[mid] >= arr[i]){
right = mid - 1;
}else{
left = mid + 1;
}
}
end[left] = arr[i];
dp[i] = left;
}
}
int[] res = new int[len];
for(int i = n - 1; i >= 0; i--){
if(dp[i] == len){
res[--len] = arr[i];
}
}
return res;
}
}