题库链接:
https://nanti.jisuanke.com/t/41387
题目大意
给定n个数,与一个数m,求ai右边最后一个至少比ai大m的数与这个数之间有多少个数
思路
对于每一个数,利用二分的方法求他右边大于等于ai+m的数的最后一个值。
关键在于怎么二分呢?
利用线段树存储区间最大值,看这个区间的最大值是不是比ai+m大
代码:
#include<bits/stdc++.h> using namespace std; #define maxn 1000005 #define mod 1000000007 int a[maxn]; struct node {int l,r,ma; }tree[maxn*4]; void build(int l,int r,int p) {tree[p].l = l;tree[p].r = r;tree[p].ma = -1;if(l == r){tree[p].ma = a[l];return;}int mid = (l+r)/2;build(l,mid,p*2);build(mid+1,r,p*2+1);tree[p].ma = max(tree[p*2].ma,tree[p*2+1].ma); } int query(int x,int y,int p) {if(x == tree[p].l && y == tree[p].r)return tree[p].ma;int mid = (tree[p].l + tree[p].r)/2;if(x > mid)return query(x,y,p*2+1);else if(y <= mid)return query(x,y,p*2);elsereturn max(query(x,mid,p*2),query(mid+1,y,p*2+1));} int main() {int T,i,j,k,n,q,x,y,m,s,l,r,mid;scanf("%d%d",&n,&m);for(i=1;i<=n;++i)scanf("%d",&a[i]);build(1,n,1);for(i=1;i<n;i++){s=m+a[i];l=i+1;r=n;while(l<r){mid=(l+r+1)/2;if(query(mid,r,1)>=s){l=mid;}else if(query(l,mid,1)>=s){r=mid-1;}else {break;}}if(a[l]>=s)cout<<l-i-1<<' ';elsecout<<"-1"<<' ';}cout<<"-1"<<endl;return 0; }