Super Mario
给你一个数列(长度n),m次询问,每次询问内有多少个数比小
input
t
n m
output
嘛,大概就这么回事了,考虑主席树,在查询k的的基础上改掉query即可
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 2e5+7;
int t,n,m,a[maxn],root[maxn],cnt,x,y,k;
vector<int> v;
int getid(int x){return lower_bound(v.begin(),v.end(),x)-v.begin()+1;}
struct node{int l,r,sum;}T[maxn*40];
void update(int l,int r,int &x,int y,int k){
T[++cnt] = T[y],T[cnt].sum++,x = cnt;
if(l == r)return;
int mid = (l+r)/2;
if(k<=mid)update(l,mid,T[x].l,T[y].l,k);
else update(mid+1,r,T[x].r,T[y].r,k);
}
int query(int l,int r,int x,int y,int k){
if(r<=k)return T[y].sum-T[x].sum;
int mid = (l+r)/2;
if(k<=mid)return query(l,mid,T[x].l,T[y].l,k);
return T[T[y].l].sum-T[T[x].l].sum+query(mid+1,r,T[x].r,T[y].r,k);
}
int main(){
int Case = 0;
for(scanf("%d",&t);t--;){
v.clear(),cnt = 0;
printf("Case %d:\n",++Case);
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
v.push_back(a[i]);
}
sort(v.begin(),v.end());
v.erase(unique(v.begin(),v.end()),v.end());
for(int i=1;i<=n;i++)update(1,n,root[i],root[i-1],getid(a[i]));
while(m--){
scanf("%d%d%d",&x,&y,&k);
x++,y++;
k = upper_bound(v.begin(),v.end(),k)-v.begin();
if(k)printf("%d\n",query(1,n,root[x-1],root[y],k));
else printf("0\n");
}
}
return 0;
}