【CodeForces】【338E】Optimize!

线段树

---

　　先搞出来每个a[i]能连多少条边记为w[i]……如果对一组s[i]，都满足w[i]-rank[i]>=0则这是一组合法方案，然后线段树维护w[i]-rank[i]（第一个元素出去的时候后面所有的rank要-1，加入最后一个元素的时候后面的元素rank+1，建关于a[i]的权值线段树，记得离散化）

　　线段树维护的时候modify（区间修改）忘加push_down了……so sad

 #include<cstdio>
 #include<cstring>
 #include<cstdlib>
 #include<iostream>
 #include<algorithm>
 #define rep(i,n) for(int i=0;i<n;++i)
 #define F(i,j,n) for(int i=j;i<=n;++i)
 #define D(i,j,n) for(int i=j;i>=n;--i)
 using namespace std;
 typedef long long LL;
 inline int getint(){
     int r=,v=; char ch=getchar();
     for(;!isdigit(ch);ch=getchar()) if(ch=='-')r=-;
     for(; isdigit(ch);ch=getchar()) v=v*+ch-'';
     return r*v;
 }
 const int N=,INF=~0u>>;
 //#define debug
 /********************template*******************/
 struct node{
     int size,min,add;
 }t[N<<];
 #define L (o<<1)
 #define R (o<<1|1)
 #define mid (l+r>>1)
 int n,m,len,H,w[N],a[N],b[N],c[N];
 void maintain(int o,int l,int r){
     if (l==r) return;
     t[o].size=t[L].size+t[R].size;
     if (t[o].size) t[o].min=min(t[L].min,t[R].min);
     else t[o].min=INF,t[o].add=;
 }
 int ql,qr;
 void push_down(int o,int l,int r){
     if (!t[o].add) return;
     t[L].add+=t[o].add; t[R].add+=t[o].add;
     t[L].min+=t[o].add; t[R].min+=t[o].add;t[o].add=;
 }
 void modify(int o,int l,int r,int num){
     if (ql>qr) return;
     if (ql<=l && qr>=r) t[o].add+=num,t[o].min+=num;
     else{
         push_down(o,l,r);
         if (ql<=mid) modify(L,l,mid,num);
         if (qr>mid ) modify(R,mid+,r,num);
         maintain(o,l,r);
     }
 }
 void update(int o,int l,int r,int pos,int rank){
     if (l==r){
         t[o].size^=;
         t[o].min=t[o].size?w[pos]-rank-t[o].size:INF;
     }else{
         push_down(o,l,r);
         if (a[pos]<=mid) update(L,l,mid,pos,rank);
         if (a[pos]>mid) update(R,mid+,r,pos,rank+t[L].size);
         maintain(o,l,r);
     }
 }
 void out(int o,int l,int r){
     printf("%d %d %d %d %d %d\n",
         o,l,r,t[o].size,t[o].add,t[o].min);
     if (l==r) return;
     else {out(L,l,mid); out(R,mid+,r);}
 }
 inline bool cmp(int x,int y){return a[x]<a[y];}
 int main(){
 #ifndef ONLINE_JUDGE
     freopen("CF338E.in","r",stdin);
     freopen("CF338E.out","w",stdout);
 #endif
     n=getint(); len=getint(); H=getint();
     F(i,,len) b[i]=getint();
     sort(b+,b+len+);b[len+]=INF;
     F(i,,n){
         c[i]=i; a[i]=getint();
         w[i]=len-(lower_bound(b+,b+len+,H-a[i])-b-);
     }
     sort(c+,c+n+,cmp);
     F(i,,n) a[c[i]]=i;
     F(i,,n*) t[i].min=INF;
     F(i,,len){
         ql=a[i]+; qr=n;
         modify(,,n,-);
         update(,,n,i,);
     }

     #ifdef debug
     F(i,,n) printf("%d %d\n",a[i],w[i]);
     out(,,n);puts("");
     #endif
     int ans=t[].min>=;
     F(i,len+,n){
         ql=a[i-len]+; qr=n;
         update(,,n,i-len,);
         modify(,,n,);

         ql=a[i]+; qr=n;
         update(,,n,i,);
         modify(,,n,-);
         if (t[].min>=) ans++;
         #ifdef debug
         printf("%d\n",i);
         out(,,n);puts("");
         #endif
     }
     printf("%d\n",ans);
     return ;
 }