leetcode:2926. 平衡子序列的最大和 【树状数组维护最大前缀和】

题目链接

lc2926

题目描述

题目思路

定义b[i] = nums[i] - i

目标是从b中找到一个非降子序列使得元素和最大

b[i] = nums[i] - i

找到b的一个非降子序列使得元素和最大

f[i]: 子序列最后一个数下标是i,对应的最大子序列

f[i] = max (max f[j], 0) + nums[i] (j < i and b[j] <= b[i])

也就是维护f[j]的前缀最大值

10 ** 9: 离散化处理 + 重新标序号

ac code

python 复制代码
# 树状数组模板(维护前缀最大值)
class BIT:
    def __init__(self, n: int):
        self.tree = [-inf] * n

    def update(self, i: int, val: int) -> None:
        while i < len(self.tree):
            self.tree[i] = max(self.tree[i], val)
            i += i & -i

    def pre_max(self, i: int) -> int:
        mx = -inf
        while i > 0:
            mx = max(mx, self.tree[i])
            i -= i & -i
        return mx


class Solution:
    def maxBalancedSubsequenceSum(self, nums: List[int]) -> int:
        n = len(nums)
        # b[i] = nums[i] - i
        # 找到b的一个非降子序列使得元素和最大
        # f[i]: 子序列最后一个数下标是i,对应的最大子序列
        # f[i] = max (max f[j], 0) + nums[i] (j < i and b[j] <= b[i])
        # 也就是维护f[j]的前缀最大值
        # 10 ** 9: 离散化处理 + 重新标序号
        b = sorted(set(x - i for i, x in enumerate(nums)))
        t = BIT(len(b) + 1) # 经典初始化
        for i, x in enumerate(nums):
            j = bisect_left(b, x - i) + 1 # 找到b[i]离散化后的位置
            f = max(t.pre_max(j), 0) + x # 计算f[i]
            t.update(j, f)
        return t.pre_max(len(b)) # 所有f的最大值
            

树状数组维护前缀最大值模版

python 复制代码
# 树状数组模板(维护前缀最大值)
class BIT:
    def __init__(self, n: int):
        self.tree = [-inf] * n

    def update(self, i: int, val: int) -> None:
        while i < len(self.tree):
            self.tree[i] = max(self.tree[i], val)
            i += i & -i

    def pre_max(self, i: int) -> int:
        mx = -inf
        while i > 0:
            mx = max(mx, self.tree[i])
            i -= i & -i
        return mx
相关推荐
დ旧言~19 分钟前
【高阶数据结构】图论
算法·深度优先·广度优先·宽度优先·推荐算法
张彦峰ZYF24 分钟前
投资策略规划最优决策分析
分布式·算法·金融
The_Ticker39 分钟前
CFD平台如何接入实时行情源
java·大数据·数据库·人工智能·算法·区块链·软件工程
Lenyiin1 小时前
02.06、回文链表
数据结构·leetcode·链表
爪哇学长1 小时前
双指针算法详解:原理、应用场景及代码示例
java·数据结构·算法
Dola_Pan1 小时前
C语言:数组转换指针的时机
c语言·开发语言·算法
繁依Fanyi2 小时前
简易安卓句分器实现
java·服务器·开发语言·算法·eclipse
烦躁的大鼻嘎2 小时前
模拟算法实例讲解:从理论到实践的编程之旅
数据结构·c++·算法·leetcode
C++忠实粉丝2 小时前
计算机网络socket编程(4)_TCP socket API 详解
网络·数据结构·c++·网络协议·tcp/ip·计算机网络·算法
祁思妙想2 小时前
10.《滑动窗口篇》---②长度最小的子数组(中等)
leetcode·哈希算法