版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/liufengwei1/article/details/83017516
上半年湘潭的C题,当时因为太菜了忘记了主席树,于是用莫队二分树状数组水过去了n(sqrt(n)logn+log^2n)的复杂度,现在用主席树就是nlog^2n的复杂度,结果也用了1.1s,当时肯定是因为数据水,没卡莫队到nsqrt(n)所以3s水过去了。
把权值主席树,二分一下答案,然后就记录一蛤从>=mid的地方到nn在L到R区间有多少个数字就行了,主席树裸题。
#include<bits/stdc++.h>
#define maxl 100010
using namespace std;
int n,nn,q,tot,cas;
int a[maxl],b[maxl],rt[maxl];
struct node
{
int ls,rs,sum;
}tree[maxl<<5];
inline void insert(int &x,int y,int num,int l,int r)
{
tree[++tot]=tree[y];x=tot;
tree[x].sum++;
if(l==r)
return;
int mid=(l+r)>>1;
if(num<=mid)
insert(tree[x].ls,tree[y].ls,num,l,mid);
else
insert(tree[x].rs,tree[y].rs,num,mid+1,r);
}
inline void prework()
{
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]),b[i]=a[i];
rt[i]=0;
tree[0].ls=tree[0].rs=tree[0].sum=0;
}
sort(b+1,b+1+n);
nn=unique(b+1,b+1+n)-b-1;
for(int i=1;i<=n;i++)
a[i]=lower_bound(b+1,b+1+nn,a[i])-b;
tot=0;
for(int i=1;i<=n;i++)
insert(rt[i],rt[i-1],a[i],1,nn);
}
inline int query(int i,int j,int L,int R,int l,int r)
{
if(l==L && r==R)
return tree[j].sum-tree[i].sum;
int mid=(l+r)>>1,sum;
if(R<=mid)
sum=query(tree[i].ls,tree[j].ls,L,R,l,mid);
else
if(L>mid)
sum=query(tree[i].rs,tree[j].rs,L,R,mid+1,r);
else
{
sum=query(tree[i].ls,tree[j].ls,L,mid,l,mid);
sum+=query(tree[i].rs,tree[j].rs,mid+1,R,mid+1,r);
}
return sum;
}
inline int find(int L,int R)
{
int l=1,r=b[nn],mid,sum,num;
while(l+1<r)
{
mid=(l+r)>>1;
num=lower_bound(b+1,b+1+nn,mid)-b;
sum=query(rt[L-1],rt[R],num,nn,1,nn);
if(sum>=mid)
l=mid;
else
r=mid;
}
num=lower_bound(b+1,b+1+nn,r)-b+1;
sum=query(rt[L-1],rt[R],num,nn,1,nn);
if(sum>=r)
return r;
else
return r-1;
}
inline void mainwork()
{
int l,r;
for(int i=1;i<=q;i++)
{
scanf("%d%d",&l,&r);
printf("%d\n",find(l,r));
}
}
int main()
{
while(~scanf("%d%d",&n,&q))
{
prework();
mainwork();
}
return 0;
}