线段树(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=' ')