给出N个数,和K
求这N个数的所有满足条件(最大数-最小数<k)的区间个数
数组b记录以当前位置开始,到最右端最多满足条件的数的个数,b数组的值可通过二分+线段树查找区间最大最小值得到
对于第二组数据
10 5
0 3 4 5 2 1 6 7 8 9
B数组为
3 7 7 7 3 1 4 3 2 1
由于当前I点取最右端值可能会导致i+1点和后面取到的点不满足条件,所有应有:b[i]<=b[i+1]+1
得B数组
3 5 4 3 2 1 4 3 2 1
ans=28
#include "stdio.h" #include "string.h" struct node { int l,r; __int64 ma,mi; }data[400010]; __int64 a[100010],b[100010]; __int64 Max(__int64 a,__int64 b) { if (a<b) return b; else return a; } __int64 Min(__int64 a,__int64 b) { if (a<b) return a; else return b; } void build(int l,int r,int k) { int mid; data[k].l=l; data[k].r=r; if (l==r) { data[k].ma=data[k].mi=a[data[k].l]; return ; } mid=(data[k].l+data[k].r)/2; build(l,mid,k*2); build(mid+1,r,k*2+1); data[k].ma=Max(data[k*2].ma,data[k*2+1].ma); data[k].mi=Min(data[k*2].mi,data[k*2+1].mi); } __int64 query(int l,int r,int k,int op) { int mid; if (data[k].l==l && data[k].r==r) { if (op==0) return data[k].mi; else return data[k].ma; } mid=(data[k].l+data[k].r)/2; if (r<=data[k*2].r) return query(l,r,k*2,op); else if (l>data[k*2].r) return query(l,r,k*2+1,op); else { if (op==0) return Min(query(l,mid,k*2,op),query(mid+1,r,k*2+1,op)); else return Max(query(l,mid,k*2,op),query(mid+1,r,k*2+1,op)); } } int main() { int t,n,k,i,l,r,mid; __int64 ans,ma,mi; scanf("%d",&t); while (t--) { scanf("%d%d",&n,&k); for (i=1;i<=n;i++) scanf("%I64d",&a[i]); build(1,n,1); for (i=1;i<n;i++) { l=i;r=n; if (data[1].ma<a[i]+k && data[1].mi>a[i]-k) l=n+1; while (l<=r) { mid=(l+r)/2; ma=query(l,mid,1,1); mi=query(l,mid,1,0); if (ma>=a[i]+k || mi<=a[i]-k) r=mid-1; else l=mid+1; } b[i]=l-i; } b[n]=1; for (i=n-1;i>=1;i--) if (b[i]>b[i+1]+1) b[i]=b[i+1]+1; ans=0; for (i=1;i<=n;i++) ans+=b[i]; printf("%I64d\n",ans); } return 0; }
版权声明:本文为博主原创文章,未经博主允许不得转载。
时间: 2024-10-27 18:38:06