(nice!!!!)LeetCode 2926. 平衡子序列的最大和（线段树、二分查找、动态规划）

最新推荐文章于 2024-06-15 09:23:13 发布

岁忧

最新推荐文章于 2024-06-15 09:23:13 发布

阅读量422

点赞数 13

分类专栏： LeetCode 文章标签： leetcode 动态规划算法 c++ 线段树

本文链接：https://blog.csdn.net/weixin_46028214/article/details/139157128

版权

LeetCode 专栏收录该内容

287 篇文章 0 订阅

订阅专栏

2926. 平衡子序列的最大和

在这里插入图片描述

思路：先预处理出nums[i]-i的数组，然后我们只需要找到递增子序列里nums元素和的最大值。
先想一个动态规划dp的思路：状态dp[i]表示以第i个元素结尾的子序列的最大值。那么就得遍历前面一段区间[0,i-1]，来更新dp[i]，这样时间复杂度就是0(n^2)。在这里我们直到，要在前面一段区间里找到小于等于nums[i]-i这个值的最大解，这里就可以通过线段树来维护，这样时间复杂度就降到0(nlogn)。
和这道题的解题思路一样（nice!!!）LeetCode 1626. 无矛盾的最佳球队（动态规划dp || 线段树）

class Solution {
public:
    vector<int> nums2;
    long long f[100005<<2];
    void push_up(int u){
        f[u]=max(f[u<<1],f[u<<1|1]);
    }
    void build(int u,int l,int r){
        if(l>=r){
            f[u]=INT_MIN;
            return ;
        }
        int mid=(l+r)/2;
        build(u<<1,l,mid);
        build(u<<1|1,mid+1,r);
        push_up(u);
    }
    long long query(int u,int l,int r,int L,int R){
        if(L<=l&&r<=R) return f[u];
        if(l>R||r<L) return INT_MIN;
        int mid=(l+r)/2;
        return max(query(u<<1,l,mid,L,R),query(u<<1|1,mid+1,r,L,R));
    }
    void modify(int u,int l,int r,int x,long long y){
        if(l>=r){
            f[u]=max(f[u],y);
            //cout<<"modify"<<l<<":"<<f[u]<<endl;
            return ;
        }
        int mid=(l+r)/2;
        if(x<=mid) modify(u<<1,l,mid,x,y);
        if(x>mid) modify(u<<1|1,mid+1,r,x,y);
        push_up(u);
    }
    long long maxBalancedSubsequenceSum(vector<int>& nums) {
        nums2=nums;
        for(int i=0;i<nums2.size();i++){
            nums2[i]-=i;
        }
        sort(nums2.begin(),nums2.end());
        nums2.erase(unique(nums2.begin(),nums2.end()),nums2.end());
        long long res=INT_MIN;
        build(1,0,nums2.size()-1);
        for(int i=0;i<nums.size();i++){
            auto tmp=lower_bound(nums2.begin(),nums2.end(),nums[i]-i)-nums2.begin();
            long long ans=max(query(1,0,nums2.size()-1,0,tmp),(long long)0)+nums[i];
            //cout<<tmp<<":"<<ans<<endl;
            res=max(res,ans);
            modify(1,0,nums2.size()-1,tmp,ans);
        }
        return res;
    }
};