2407. 最长递增子序列 II

最新推荐文章于 2024-10-10 22:59:02 发布

云儿乱飘

最新推荐文章于 2024-10-10 22:59:02 发布

阅读量50

点赞数

分类专栏：算法题文章标签：算法 leetcode 数据结构

本文链接：https://blog.csdn.net/Mz_yuner/article/details/134297009

版权

算法题专栏收录该内容

168 篇文章 0 订阅

订阅专栏

2407. 最长递增子序列 II - 力扣（LeetCode）

线段树练习

class Solution {
public:
    vector<int> tree;
    int get(int node, int l, int r, int L, int R) {
        if (L <= l && r <= R) return tree[node];
        int mid = (l + r) >> 1;
        int ret = 0;
        if (L <= mid) ret = get(node * 2, l, mid, L, R);
        if (R >  mid) return max(ret, get(node * 2 + 1, mid + 1, r, L, R));
        return ret;
    }
    void update(int node, int l, int r, int pos, int val) {
        tree[node] = max(tree[node], val);
        if (l == r) return;
        int mid = (l + r) >> 1;
        if (pos <= mid) update(node * 2, l, mid, pos, val);
        else update(node * 2 + 1, mid + 1, r, pos, val);
        return;
    }
    int lengthOfLIS(vector<int>& nums, int k) {
        int u = *max_element(nums.begin(), nums.end());
        tree.resize(4 * u);
        for (int i = 0; i < nums.size(); i++) {
            int f = 1;
            if (nums[i] != 1){
                int x = nums[i] - k > 0 ? nums[i] - k : 1;
                f = get(1, 1, u, x, nums[i] - 1) + 1;    
            }
            update(1, 1, u, nums[i], f); 
        }
        return tree[1];
    }
};