牛客题霸 [ 最长递增子序列] C++题解/答案

牛客题霸 [ 最长递增子序列] C++题解/答案

题目描述

给定数组arr，设长度为n，输出arr的最长递增子序列。（如果有多个答案，请输出其中字典序最小的）

题意：

直接暴力会超时
应该用二分+贪心

题解：

class Solution {
public:
    /**
     * retrun the longest increasing subsequence
     * @param arr int整型vector the array
     * @return int整型vector
     */
    vector<int> LIS(vector<int>& arr) {
        // write code here
         int n = arr.size();
        if (n < 2) return arr;
        int ans = 0;
        vector<int> ret(n);
        vector<int> st;
        for (int i = 0; i < n; i++){
            if (st.size() == 0 || arr[i] > st.back()){
                st.push_back(arr[i]);
                ret[i] = ++ans;
            }
            else{
                int l = 0, h = ans-1;
                while (l < h){
                    int m = (l+h)/2;
                    if (st[m] < arr[i]) l = m+1;
                    else h = m;
                }
                st[h] = arr[i];
                ret[i] = h+1;
            }
        }
        vector<int> res(ans);
        for (int i = n-1; i >= 0; i--){
            if (ret[i] == ans){
                res[--ans] = arr[i];
            }
        }
        return res;
    }
};