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;
}
};