BZOJ4299 : Codechef FRBSUM

Claris發表於2015-10-12

若$[0,i]$的數都可以得到,那麼$[1,所有不大於i+1的數的和]$的數都可以得到。

如此暴力列舉答案,用可持久化線段樹支援查詢,因為每次數字至少翻一倍,所以複雜度為$O(m\log^2n)$。

 

#include<cstdio>
#include<algorithm>
const int N=100010,M=1800000;
int n,m,i,j,c,d,a[N],b[N],g[N],nxt[N],T[N],l[M],r[M],v[M],tot;
inline void read(int&a){char c;while(!(((c=getchar())>='0')&&(c<='9')));a=c-'0';while(((c=getchar())>='0')&&(c<='9'))(a*=10)+=c-'0';}
inline int lower(int x){
  int l=1,r=n,mid,t;
  while(l<=r)if(b[mid=(l+r)>>1]<=x)l=(t=mid)+1;else r=mid-1;
  return t;
}
int ins(int x,int a,int b,int c,int p){
  int y=++tot;v[y]=v[x]+p;
  if(a==b)return y;
  int mid=(a+b)>>1;
  if(c<=mid)l[y]=ins(l[x],a,mid,c,p),r[y]=r[x];else l[y]=l[x],r[y]=ins(r[x],mid+1,b,c,p);
  return y;
}
int ask(int x,int a,int b){
  if(!x)return 0;
  if(c<=a&&b<=d)return v[x];
  int mid=(a+b)>>1,t=0;
  if(c<=mid)t=ask(l[x],a,mid);
  if(d>mid)t+=ask(r[x],mid+1,b);
  return t;
}
inline int query(){for(i=0;;i=j)if((j=ask(T[lower(i+1)],1,n))==i)return i+1;}
int main(){
  for(read(n),i=1;i<=n;i++)read(a[i]),b[i]=a[i];
  for(std::sort(b+1,b+n+1),i=1;i<=n;i++)nxt[i]=g[a[i]=lower(a[i])],g[a[i]]=i;
  for(i=1;i<=n;i++)for(T[i]=T[i-1],j=g[i];j;j=nxt[j])T[i]=ins(T[i],1,n,j,b[i]);
  for(read(m);m--;printf("%d\n",query()))read(c),read(d);
  return 0;
}