线段树 算法

文章目录

基础知识

线段树和树状数组都只是一个工具来的,题目并不会一下子就告诉你这个题目用到线段树和树状数组,这个取决于你想使用的数据结构以及所要优化的方向

线段树和树状数组(也称为二叉索引树)是两种常用的数据结构,主要用于处理数组的区间查询和更新操作。它们的主要区别如下:

适用场景

  • 线段树
    • 适用于需要复杂区间操作的场景,如区间最大值、区间最小值、区间更新等。
    • 适合动态性较强的问题。
python 复制代码
# 点的更新
class Tree:
    
    def __init__(self,n):
        self.st = [0]*(4*n)
    
    def update(self,o,l,r,index,val):
        # l,r 是当前区间的范围,index 是要插入的数据的索引
        if l==r:
            self.st[o] = val
            return 
        mid = (l+r)//2
        if index<=mid:
            self.update(2*o,l,mid,index,val)
        else:
            self.update(2*o+1,mid+1,r,index,val)
        self.st[o] = max(self.st[2*o],self.st[2*o+1])
    
    def query(self,o,l,r,L,R):
        if l >= L and r <= R:
            return self.st[o]
        mid = (l+r)//2
        res = 0
        if L<=mid:
            res = max(res,self.query(2*o,l,mid,L,R))
        if R > mid:
            res = max(res,self.query(2*o+1,mid+1,r,L,R))
        return res
  • 树状数组
    • 适用于简单的前缀和查询和单点更新问题。
    • 适合静态或半静态问题,且代码实现更简洁。

小结

特性 线段树 树状数组
结构 二叉树 基于数组的树形结构
功能 支持复杂区间操作 主要用于前缀和查询
时间复杂度 ( O ( log ⁡ n ) (O(\log n) (O(logn)) ( O ( log ⁡ n ) ) (O(\log n)) (O(logn))
空间复杂度 ( O ( 4 n ) (O(4n) (O(4n)) ( O ( n ) ) (O(n)) (O(n))
适用场景 动态区间操作 简单查询和更新

题目概述

2407.最长递增子序列 II

题目详解

300.最长递增子序列

这个题目是方便我们做下面的 最长递增子序列

思路分析:这个最长递增子序列,我们采用动态规划进行求解,
dp[i]的定义:以nums[i]为结尾的子序列的最长的长度

有递推公式:
d p [ i ] = m a x ( d p [ i ] , d p [ j ] + 1 ) dp[i] = max(dp[i],dp[j]+1) dp[i]=max(dp[i],dp[j]+1),当我们遍历前 i − 1 i-1 i−1个元素,当满足 n u m s [ j ] < = n u m s [ i ] nums[j]<=nums[i] nums[j]<=nums[i]的时候更新
时间复杂度分析:是o(n^2)

2407.最长递增子序列 II


思路分析:这个题目首先想到可以使用动态规划,但是如果内层的的判断i之前的满足的情况的时候还是使用暴力进行求解的话,就会出现超时的情况,所以我们就得考虑使用线段树,线段树对于求解区间的最值的时间复杂度只有o(logn)

python 复制代码
class Solution:
    def lengthOfLIS(self, nums: List[int], k: int) -> int:
        ans = 1
        
        # dp[i][j] 表示前i个元素中,以值j结尾的长度的子序列长度的最大值
        # 注意,这里的j是值域
        tr = Tree(100005)
        # 这个求解的是外层循环i
        for va in nums:
            if va != 1:
            	# 确保这个zuo没有超过范围
                zuo = max(1,va-k)
                you = va-1
                # 1为根节点
                # 这个是求解的内层循环j
                pre = tr.query(1,1,100000,zuo,you)
                now = pre+1
                ans = max(ans,now)
                tr.update(1,1,100000,va,now)
            else:
                tr.update(1,1,100000,1,1)
        return ans

# 线段树模版
class Tree:

    def __init__(self,n):
        self.t = [0] * (n*4)
    def update(self,o,l,r,index,va):
        if l==r:
            self.t[o] = va
            return 
        m = (l+r) //2
        if index<=m:
            self.update(o*2,l,m,index,va)
        else:
            self.update(o*2+1,m+1,r,index,va)
        self.t[o] = max(self.t[o*2],self.t[o*2+1])

    def query(self,o,l,r,L,R):
        if l>=L and r<= R:
            return self.t[o]
        m = (l+r) // 2
        res = 0
        if m >= L:
            res = max(res,self.query(o*2,l,m,L,R))
        if R>m:
            res = max(res,self.query(o*2+1,m+1,r,L,R))
        return res
相关推荐
NAGNIP1 天前
AI训练要的数据这么多,怎么存?
算法
CoovallyAIHub1 天前
冻结比微调更好?YOLOv8/V10迁移学习最佳实践发布,GPU内存直降28%
深度学习·算法·计算机视觉
爱编程的化学家1 天前
代码随想录算法训练营第21天 -- 回溯4 || 491.非递减子序列 / 46.全排列 /47.全排列 II
数据结构·c++·算法·leetcode·回溯·全排列·代码随想录
console.log('npc')1 天前
前端性能优化,给录音播放的列表加个播放按键,点击之后再播放录音。减少页面的渲染录音文件数量过多导致加载缓慢
前端·javascript·vue.js·算法
_不会dp不改名_1 天前
leetcode_138 随机链表的复制
算法·leetcode·链表
小欣加油1 天前
leetcode 129 求根节点到叶节点数字之和
数据结构·c++·算法·leetcode
DuHz1 天前
Stable Video Diffusion:将潜在视频扩散模型扩展到大规模数据集——论文阅读
论文阅读·人工智能·深度学习·神经网络·算法·音视频
彩云回1 天前
决策树(Decision Tree)
算法·决策树·机器学习
铭哥的编程日记1 天前
C++优选算法精选100道编程题(附有图解和源码)
开发语言·c++·算法
不枯石1 天前
Matlab通过GUI实现点云的最远点下采样(Farthest point sampling)
开发语言·图像处理·算法·计算机视觉·matlab