codeforces #271(div2) E

  1 #include<bits/stdc++.h>
  2 using namespace std;
  3 typedef long long ll;
  4 const int INF = 0x3f3f3f3f;
  5 const int N = 100005;
  6 
  7 ll h[N],hh[N];
  8 struct node{
  9     int l,r,Max;
 10 }e[N*4];
 11 map<ll ,int >a;
 12 map<ll ,int >::iterator it;
 13 ll dp[N];
 14 
 15 void build(int L,int R,int x){
 16     e[x].l=L;
 17     e[x].r=R;
 18     e[x].Max=0;
 19     if(L==R) return ;
 20     int mid=(L+R)>>1;
 21     build(L,mid,2*x);
 22     build(mid+1,R,2*x+1);
 23 }
 24 
 25 void update(int y,int val,int x){
 26     if(e[x].l==y&&e[x].r==y) {
 27         e[x].Max=val;return ;
 28     }
 29     int mid=(e[x].l+e[x].r)>>1;
 30     if(y<=mid) update(y,val,2*x);
 31     else update(y,val,2*x+1);
 32     e[x].Max=max(e[x*2].Max,e[2*x+1].Max);
 33 }
 34 
 35 ll query(int L,int R,int x){
 36     if(e[x].l>=L&&e[x].r<=R) {
 37         return e[x].Max;
 38     }
 39 
 40     int mid=(e[x].l+e[x].r)>>1;
 41 
 42     if(mid>=R) return query(L,R,2*x);
 43     else if(L>mid) return query(L,R,2*x+1);
 44     else return max(query(L,mid,2*x),query(mid+1,R,2*x+1));
 45     e[x].Max=max(e[x*2].Max,e[2*x+1].Max);
 46 }
 47 int wei[N];
 48 int main(){
 49     int n,d;
 50     scanf("%d%d",&n,&d);
 51     for(int i=1;i<=n;i++){
 52         scanf("%lld",&h[i]);
 53         hh[i]=h[i];
 54     }
 55     sort(h+1,h+1+n);
 56     int x=0;
 57     for(int i=1;i<=n;i++){
 58         if(a[h[i]]==0) a[h[i]]=++x;
 59     }
 60     build(1,x,1);
 61     for(int i=1;i<=n;i++){
 62         dp[i]=1;
 63         it=a.upper_bound(hh[i]-d);//得找hh[i]-d及以前的数造成的最大值,upper指向第一个>该值的位置//
 64         if(it!=a.begin()){
 65             it--;
 66             int xx=it->second;
 67             ll Max=query(1,xx,1);
 68             dp[i]=max(dp[i],Max+1);
 69         }
 70         it=a.lower_bound(hh[i]+d);//得找hh[i]+d及后面的数字,lower指向第一个>=该值的位置//
 71         if(it!=a.end()){
 72             int xx=it->second;
 73             ll Max=query(xx,x,1);
 74             dp[i]=max(dp[i],Max+1);
 75         }
 76         update(a[hh[i]],dp[i],1);
 77     }
 78     ll ss=1;
 79     int z;
 80     for(int i=1;i<=n;i++){
 81         if(dp[i]>ss){
 82             ss=dp[i];z=i;
 83         }
 84     }
 85     printf("%d\n",ss);
 86     int t=0;
 87     wei[++t]=z;
 88     for(int i=n;i>=1;i--){
 89      // cout<<dp[i]<<" "<<hh[i]<<" "<<
 90         if(dp[i]==ss-1&&abs(hh[z]-hh[i])>=d){
 91             wei[++t]=i;
 92             z=i;
 93             ss--;
 94         }
 95     }
 96     for(int i=t;i>=1;i--){
 97         printf("%d ",wei[i]);
 98     }
 99     printf("\n");
100     return 0;
101 }