The Preliminary Contest for ICPC Asia Xuzhou 2019 E. XKC's basketball team （线段树写法）

我只想问出题人，这题是不是在黑我坤坤？？？

题目链接

题意：有n个人，每个人都有一个w值。问每个人右边最远的且大于等于wi+m的位置。

题解：我用了三种写法。这里都说一下。

1.单调序列，从后往前维护一个单调递增的序列，然后对于第i个人在这个序列中二分一下就行了。

原理是，我比你大且比你位置靠后，选我肯定最优。代码就不贴了。

2权值线段树写法。先离散化，建权值线段树，线段树维护区间下标最大值。然后每次查询只要判段查询的值是否在我右边就好了

#include<bits/stdc++.h>
using namespace std;
#define ls id*2
#define rs id*2 + 1
#define mid (l + r) / 2
const int N = 5e5+100;
int w[N], b[N], s[4*N], ans[N];
void up(int id, int l, int r, int pos, int val){
	if(l == r){
		s[id] = val;
		return;
	}
	if(pos <= mid) 
		up(ls, l, mid, pos, val);
	else 	
		up(rs, mid+1, r, pos, val);
	s[id] = max(s[ls], s[rs]);
}

int qurey(int id, int l, int r, int ql, int qr){
	if(ql<=l && qr>=r)
		return s[id];
	int res = 0;
	if(ql <= mid) 
		res = qurey(ls, l, mid, ql, qr);
	if(qr > mid) 
		res = max(res, qurey(rs, mid+1, r, ql, qr));
	return res;
}

int main()
{
	int n, m;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;++i) 
		scanf("%d",&w[i]),b[i] = w[i];
	sort(b+1, b+1+n);
	int sz = unique(b+1, b+1+n)- b - 1;
	for(int i=1; i<=n; ++i){
		int x=lower_bound(b+1, b+1+sz, w[i]) - b;
		up(1, 1, sz, x, i);
	}
	for(int i=1;i<=n;++i){
		int val = lower_bound(b+1, b+1+sz, w[i]+m) - b;
		if(val>sz) 
			ans[i]=-1;
		else {
			int x = qurey(1, 1, sz, val, sz);
			if(x<i) 
				ans[i]=-1;
			else 
				ans[i] = x - i - 1;
		}
	}
	for(int i=1;i<=n;++i)
		printf("%d%c",ans[i],i==n?'\n':' ');
}

3.线段树写法：线段树维护区间最大值。查询时一定要先往右边走，因为往右走答案更优。实际上就是树上二分。

#include<bits/stdc++.h>
using namespace std;
#define ls o*2
#define rs o*2 + 1
#define mid (l + r) / 2
const int N = 5e5+10;
int mx[N*4], w[N];
void build(int o, int l, int r){
	if(l==r){
		scanf("%d", &w[l]);
		mx[o] = w[l];
		return;
	}
	build(ls, l, mid);
	build(rs, mid + 1, r);
	mx[o] = max(mx[ls], mx[rs]);
}
int query(int o, int l, int r, int ql, int qr,int val){
	if(l == r) 
		return l;
	if(mx[rs] >= val)
		return query(rs, mid+1, r, ql, qr, val);
	else 
		if(ql <= mid && mx[ls] >= val)
			return query(ls, l, mid, ql, qr, val);
	else 
		return -1;
}
int main()
{
	int n,m;
	scanf("%d%d", &n, &m);
	build(1, 1, n);
	int ans;
	for(int i=1; i<=n; ++i){
		if(i==n) return puts("-1"), 0;
		ans = query(1, 1, n, i+1, n, w[i] + m);
		if(ans!=-1)
			ans -= i + 1;
		printf("%d ",ans);
	}
}