询问;给定区间中所有小于h的数字个数。
思路:http://www.cnblogs.com/kuangbin/archive/2012/09/23/2699122.html
离线线段树之后,可以边更新,边查询。代码思路很简单,感觉这一手倒向更新太赞了。
此题询问多少个数字,所以可以用简单的线段树来做,但是如果询问第K大的数字是几,就需一个保存历史版本的主席树了
#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<map>
#include<vector>
#include<string.h>
#include<stdlib.h>
#include<math.h>
using namespace std;
const int MAXN=100010;
const int maxn=100010;
struct node
{
int l,r,sum;
}tree[maxn*4];
int s[maxn];
void build(int i,int left,int right)
{
tree[i].l=left;
tree[i].r=right;
tree[i].sum=0;
if(left==right)
{
s[left]=i;
return ;
}
int mid=(left+right)>>1;
build(i<<1,left,mid);
build(i<<1|1,mid+1,right);
}
void update(int t)
{
int m=s[t];
while(m)
{
tree[m].sum++;
m>>=1;
}
}
int query(int i,int left,int right)
{
if(tree[i].l==left&&tree[i].r==right)
return tree[i].sum;
int mid=(tree[i].l+tree[i].r)>>1;
if(right<=mid)
return query(i<<1,left,right);
else if(left>mid)
return query(i<<1|1,left,right);
else return query(i<<1,left,mid)+query(i<<1|1,mid+1,right);
}
struct last
{
int l,r;
int h,index;
}ques[maxn];
int ans[maxn];
struct fir
{
int index,h;
}pre[maxn];
bool cmp(fir a,fir b)
{
return a.h<b.h;
}
bool cmp2(last a,last b)
{
return a.h<b.h;
}
int main()
{
int T;
int m;
int n;
int iCase=0;
scanf("%d",&T);
while(T--)
{
iCase++;
scanf("%d%d",&n,&m);
build(1,1,n);
for(int i=1; i<=n; i++)
{
scanf("%d",&pre[i].h);
pre[i].index=i;
}
for(int i=0; i<m; i++)
{
scanf("%d%d%d",&ques[i].l,&ques[i].r,&ques[i].h);
ques[i].l++;
ques[i].r++;
ques[i].index=i;
}
sort(ques,ques+m,cmp2);
sort(pre+1,pre+1+n,cmp);
int i=1,j=0;
while(j<m)
{
while(i<=n)
{
if(pre[i].h>ques[j].h)
break;
update(pre[i].index);
i++;
}
while(j<m)
{
if(i<=n&&ques[j].h>=pre[i].h)
break;
ans[ques[j].index]=query(1,ques[j].l,ques[j].r);
j++;
}
}
printf("Case %d:\n",iCase);
for(int i=0; i<m; i++)
printf("%d\n",ans[i]);
}
return 0;
}