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

The Preliminary Contest for ICPC Asia Xuzhou 2019 E. XKC’s basketball team

题意

给 n 个数，和一个 m，对于每个数 i ，找右边最远的满足 j >= m 的 j。
输出两个数中间的数字个数。

分析

n 2e5，很明显要找一个 n log n 的算法，也就是对于每个数查询做到 log n。
自然想到线段树 。对于每个数 x ，可以用线段树找从右边数第一个大于等于 x + m 的数字位置，判断是否在 x 右边，

线段树具体操作，维护每一个区间的最大值。每次查询先找右区间，再找左区间。这样就是右边第一个了。

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define fuck(x) cout << (x) << endl
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
using namespace std;
typedef long long ll;
const int mod = 1e9 + 7;
const int N = 5e5 + 10;
const int M = 1e4 + 10;

int n, m;
int a[N];
int tr[N << 2];

void build(int l, int r, int rt){
    if(l == r){
        tr[rt] = a[l];
        return;
    }
    int m = l + r >> 1;
    build(lson);
    build(rson);
    tr[rt] = max(tr[rt << 1], tr[rt << 1 | 1]); // pushup(rt);
}

int query(int l, int r, int rt, int sum){
    if(l == r){
        return l;
    }
    int m = l + r >> 1;
    if(tr[rt<<1|1] >= sum)       // 先找右边的
        return query(rson, sum);      
    else {						// 右边没有再考虑左边
        if(tr[rt<<1] >= sum)
            return query(lson, sum);
        else {				// 都没有就不行了
            return -1;               
        }
    }    
}

int main(){
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i++){
        scanf("%d", &a[i]);
    }
    build(1, n, 1);
    for(int i = 1; i <= n; i++){
        int ans = query(1, n, 1, a[i]+m);
        if(ans == -1){	// 没有找到
            printf("-1%c", " \n"[i == n]);
        }else {
            if(ans > i)	
                printf("%d%c", ans - i - 1, " \n"[i == n]);
            else	// 找到了，但是不在 i 的右边
                printf("-1%c", " \n"[i == n]);
        }
    }
    return 0;
}