题意:
有n个数字 m个查询 每个查询询问区间[l,r]内有几个数字小于h
思路:
由于n个数字是不变的 所以可以利用划分树在nlogn复杂度内求出任意区间的第k小值(这里指建树)
每次查询可以二分查找该区间第几小的数小于等于h 那么答案就是二分出的个数
总复杂度 nlogn + m(logn)^2 即 建树复杂度 + m次询问*每次查询logn*二分复杂度logn
n和m都只有10^5 完全可过
代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 100010
#define M 20
int tree[M][N],toleft[M][N],sorted[N];
int n,m,t,T;
void build(int l,int r,int dep)
{
if(l==r) return ;
int i,mid=(l+r)>>1;
int y=sorted[mid],same=mid-l+1,lpos=l,rpos=mid+1;
for(i=l;i<=r;i++)
{
if(tree[dep][i]<y) same--;
}
for(i=l;i<=r;i++)
{
if(tree[dep][i]<y) tree[dep+1][lpos++]=tree[dep][i];
else if(tree[dep][i]==y&&same>0)
{
tree[dep+1][lpos++]=tree[dep][i];
same--;
}
else tree[dep+1][rpos++]=tree[dep][i];
toleft[dep][i]=toleft[dep][l-1]+lpos-l;
}
build(l,mid,dep+1);
build(mid+1,r,dep+1);
}
int query(int L,int R,int l,int r,int dep,int k)
{
if(l==r) return tree[dep][l];
int mid=(L+R)>>1,amt=toleft[dep][r]-toleft[dep][l-1];
if(amt>=k)
{
int fl=L+toleft[dep][l-1]-toleft[dep][L-1];
int fr=fl+amt-1;
return query(L,mid,fl,fr,dep+1,k);
}
else
{
int fr=r+toleft[dep][R]-toleft[dep][r];
int fl=fr-(r-l-amt);
return query(mid+1,R,fl,fr,dep+1,k-amt);
}
}
int main()
{
int i,j,l,r,h,fr,ed,md,tmp,ans;
while(~scanf("%d",&T))
{
for(t=1;t<=T;t++)
{
printf("Case %d:\n",t);
scanf("%d%d",&n,&m);
for(i=1;i<=n;i++)
{
scanf("%d",&sorted[i]);
tree[0][i]=sorted[i];
}
sort(sorted+1,sorted+1+n);
build(1,n,0);
while(m--)
{
scanf("%d%d%d",&l,&r,&h);
l++; r++;
fr=1; ed=r-l+1; ans=0;
while(fr<=ed)
{
md=(fr+ed)>>1;
tmp=query(1,n,l,r,0,md);
if(tmp<=h)
{
fr=md+1;
ans=md;
}
else ed=md-1;
}
printf("%d\n",ans);
}
}
}
return 0;
}