题目大意:给定一个序列,求区间内小于等于某数的元素数量
首先区间排名一看就是划分树 不过和第k小不一样 我们需要做一些处理
第一种处理方式是二分答案 然后转换成区间第k小 好方法我喜欢 但是这里说的不是这种方法
首先建树,然后对于每个询问,我们讨论k与a[mid]的关系
若k<a[mid],则右子树一定没有小于等于k的数,我们进入左子树查找
若k>=a[mid],则左子树内一定所有数都小于等于k,于是我们将查询区间中进入左子树的元素的数量记入ans,然后查找右区间
递归退出条件是查询区间为空或节点为叶节点
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define M 100100
using namespace std;
int n,m,cnt,a[M],b[M],c[M];
int s[20][M];
void Build_Tree(int l,int r,int dpt)
{
int i,mid=l+r>>1;
int l1=l,l2=mid+1;
int left=mid-l+1;
if(l==r)
return ;
for(i=l;i<=r;i++)
left-=(a[i]<c[mid]);
for(i=l;i<=r;i++)
{
if(a[i]>c[mid]||a[i]==c[mid]&&!left)
b[l2++]=a[i],s[dpt][i]=(i==l?0:s[dpt][i-1] );
else
b[l1++]=a[i],s[dpt][i]=(i==l?1:s[dpt][i-1]+1),left-=(a[i]==c[mid]);
}
memcpy( a+l , b+l , sizeof(a[0])*(r-l+1) );
Build_Tree(l,mid,dpt+1);
Build_Tree(mid+1,r,dpt+1);
}
int Get_Ans(int l,int r,int dpt,int x,int y,int k)
{
int mid=l+r>>1;
int l1=(x==l?0:s[dpt][x-1]),l2=s[dpt][y];
if(x>y)
return 0;
if(l==r)
return a[mid]<=k;
if(k<c[mid])
return Get_Ans(l,mid,dpt+1,l+l1,l+l2-1,k);
else
return l2-l1+Get_Ans(mid+1,r,dpt+1,(mid+1)+(x-l-l1),(mid+1)+(y-l+1-l2)-1,k);
}
int main()
{
int T,i,x,y,k;
for(cin>>T;T;T--)
{
printf("Case %d:\n",++cnt);
cin>>n>>m;
for(i=1;i<=n;i++)
scanf("%d",&a[i]),c[i]=a[i];
sort(c+1,c+n+1);
Build_Tree(1,n,0);
for(i=1;i<=m;i++)
scanf("%d%d%d",&x,&y,&k),printf("%d\n", Get_Ans(1,n,0,x+1,y+1,k) );
}
}
后来看了看题解发现这题树状数组也能做。。。离线思想,先OTZ一下
我们把询问按照询问的值排序,对于每一个值,我们把小于等于这个值的元素位置都插进树状数组,然后查询(x-1,y]区间内有多少个元素即可
写完是140MS 划分树是109MS 没办法两次排序的常数有点大0.0
把反复memset改成时间标记的话是125MS 能快一些
然后别忘了输出Case 1...因为这点P事还贡献了一次WA
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define M 100100
using namespace std;
struct abcd{
int x,y,num,pos;
bool operator < (const abcd &x)const
{
return num < x.num;
}
}q[M];
pair<int,int>a[M];
int n,m,cnt,ans[M];
int c[M],tim[M];
void update(int x)
{
for(;x<=n;x+=x&-x)
{
if(tim[x]!=cnt)
c[x]=0;
tim[x]=cnt;
c[x]++;
}
}
int getans(int x)
{
int re=0;
for(;x;x-=x&-x)
if(tim[x]==cnt)
re+=c[x];
return re;
}
int main()
{
int T,i,j,x,y,k;
for(cin>>T;T;T--)
{
printf("Case %d:\n",++cnt);
cin>>n>>m;
for(i=1;i<=n;i++)
scanf("%d",&a[i].first),a[i].second=i;
sort(a+1,a+n+1);
for(i=1;i<=m;i++)
scanf("%d%d%d",&q[i].x,&q[i].y,&q[i].num),q[i].pos=i;
sort(q+1,q+m+1);
j=1;
for(i=1;i<=m;i++)
{
for(;j<=n&&a[j].first<=q[i].num;j++)
update(a[j].second);
ans[q[i].pos]=getans(q[i].y+1)-getans(q[i].x);
}
for(i=1;i<=m;i++)
printf("%d\n",ans[i]);
}
}