Xuzhou网络赛19 E【线段树】

最新推荐文章于 2021-03-16 16:21:33 发布

zhmeng_

最新推荐文章于 2021-03-16 16:21:33 发布

阅读量133

点赞数

分类专栏：线段树

本文链接：https://blog.csdn.net/qq_43482299/article/details/100614523

版权

线段树专栏收录该内容

2 篇文章 0 订阅

订阅专栏

XKC’s basketball team

传送门。

题意：

第一行两个数 $n$ $m$ $(2\leq n\leq 5*10^5, 0\leq m \leq 10^9)$
第二行 $n$ 个数的序列 $w_1..w_n(0\leq w_i \leq 10^9)$
输出序列中每个元素的愤怒值
第i个人的愤怒值= $j - i - 1$ （ $j$ :满足 $a [j] > = a [i] + m$ 中最大的 $j$ ，即最右一个大于等于 $a [i] + m$ 的元素下标）
若不存在 $j$ 则 $a [i] = - 1$ 。

思路：

维护线段树最大值
若查询区间最大值大于等于 $a [i] + m$ 则答案在该区间
优先查询右子树，保证答案为最右
query()函数返回下标

基础线段树，算是今天的签到题。

#include<cstdio>
#include<algorithm>
using namespace std;

const int Mn=5e5+17;
struct tree{	//区间[l,r]的最大值maxx
	int maxx,l,r;
}T[Mn*4];
int n,mm,a[Mn],ans[Mn];

void pushup(int rt){
	T[rt].maxx=max(T[rt<<1].maxx,T[rt<<1|1].maxx);
}

void update(int L,int R,int ud,int l,int r,int rt){
	T[rt].l=l,T[rt].r=r;
	if(L<=l&&r<=R){
		T[rt].maxx=ud;
		return;
	}
	int mid=(l+r)>>1;
	if(L<=mid) update(L,R,ud,l,mid,rt<<1);
	if(mid<R) update(L,R,ud,mid+1,r,rt<<1|1);
	pushup(rt);
}

int query(int L,int v,int rt){	//询问区间[L,n]中>=v的位置，优先向右子树查询
	if(T[rt].l==T[rt].r) return T[rt].r;
	int mid=(T[rt].l+T[rt].r)>>1;
	if(T[rt<<1|1].maxx>=v) return query(L,v,rt<<1|1);
	else if(T[rt<<1].maxx>=v&&mid>=L) return 	query(L,v,rt<<1);
	return	-1;
}

int main(){
	scanf("%d%d",&n,&mm);
	for(int i=1;i<=n;++i){scanf("%d",&a[i]);update(i,i,a[i],1,n,1);}//建树
	for(int i=n-1;i>0;--i){
		ans[i]=query(i,a[i]+mm,1);
		if(ans[i]==i) ans[i]=-1;
		else if(ans[i]!=-1) ans[i]=ans[i]-i-1;
	}
	ans[n]=-1;
	for(int i=1;i<n;++i) printf("%d ",ans[i]);printf("%d\n",ans[n]);
	return	0;
}