线段树处理区间最小值(python实现)

线段树(Segment Tree)是一种常见的数据结构,常用于解决区间和、区间最值等问题。

例题:

问题描述
  小蓝有一个序列 a[1], a[2], …, a[n]。
  给定一个正整数 k,请问对于每一个 1 到 n 之间的序号 i,a[i-k], a[i-k+1], …, a[i+k] 这 2k+1 个数中的最小值是多少?当某个下标超过 1 到 n 的范围时,数不存在,求最小值时只取存在的那些值。
输入格式
  输入的第一行包含一整数 n。
  第二行包含 n 个整数,分别表示 a[1], a[2], …, a[n]。
  第三行包含一个整数 k 。
输出格式
  输出一行,包含 n 个整数,分别表示对于每个序号求得的最小值。
样例输入
5
5 2 7 4 3
1
样例输出
2 2 2 3 3
评测用例规模与约定
  对于 30% 的评测用例,1 <= n <= 1000,1 <= a[i] <= 1000。
  对于 50% 的评测用例,1 <= n <= 10000,1 <= a[i] <= 10000。
  对于所有评测用例,1 <= n <= 1000000,1 <= a[i] <= 1000000。
 

class segment_tree:
    def __init__(self,tree,arr):
        self.tree=tree
        self.arr=arr
        self.inf=1e8+7
    def build(self,rt,l,r):
        if l==r:
            self.tree[rt]=self.arr[l]
            return
        mid=(l+r)>>1
        self.build(rt<<1,l,mid)
        self.build(rt<<1|1,mid+1,r)
        self.tree[rt]=min(self.tree[rt<<1],self.tree[rt<<1|1])
    def query(self,rt,l,r,L,R):
        if L<=l and r<=R:
            return self.tree[rt]
        mid=(l+r)>>1
        ans=inf
        if mid>=L:
            ans=min(ans,self.query(rt<<1,l,mid,L,R))
        if mid<R:
            ans=min(ans,self.query(rt<<1|1,mid+1,r,L,R))
        return ans
if __name__=='__main__':
    inf=1e9+7
    n=int(input())
    arr=list(map(int,input().split()))
    tree=[inf]*4*n
    arr.insert(0,0)
    a_segment_tree=segment_tree(tree,arr)
    a_segment_tree.build(1,1,n)
    k=int(input())
    res=[]
    for i in range(1,n+1):
        res.append(a_segment_tree.query(1,1,n,i-k,i+k))
    for j in range(n):
        print(res[j],end=' ')

  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值