题目很简单 求区间内部小于等于k 的数字个数
还是主席树思想,n个线段树每个维护 1-i ,之后本题和求第k大很像,我们需要对输入数据预处理一下
首先要对输入的数据进行一下离散化,保存到新数组里,每个线段树维护的是某个数字出现过的次数,之后记录出原数组每个数字离散后的大小,之后先找出第k大的数字,之后在整个数据结构里找出' 1 - k 之间的所有数字数量就好了
以下是AC代码
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 7;
int tot;
int a[maxn], b[maxn];
int rt[maxn];
struct node
{
int lc, rc, v;
}t[maxn*20];
int build(int l, int r)
{
int now = ++ tot;
t[now].lc = t[now].rc = t[now].v = 0;
if(l == r)
return now;
int mid = (l + r) >> 1;
t[now].lc = build(l, mid);
t[now].rc = build(mid+1, r);
return now;
}
int update(int last, int l, int r, int pos)
{
int now = ++ tot;
t[now] = t[last];
t[now].v ++;
if(l == r)
return now;
int mid = (l + r) >> 1;
if(pos <= mid)
{
t[now].lc = update(t[last].lc, l, mid, pos);
}
else
{
t[now].rc = update(t[last].rc, mid+1, r, pos);
}
return now;
}
int query(int st, int ed, int L, int R, int l, int r)
{
if(L <= l && r <= R)
return t[ed].v - t[st].v;
int mid = (l + r) >> 1;
int res = 0;
if(L <= mid)
res += query(t[st].lc, t[ed].lc, L, R, l, mid);
if(R > mid)
res += query(t[st].rc, t[ed].rc, L, R, mid+1, r);
return res;
}
int main()
{
int T, Case = 0;
scanf("%d", &T);
int n, q;
while(T --)
{
scanf("%d%d", &n, &q);
for(int i = 1;i <= n;i ++)
{
scanf("%d", &a[i]);
b[i] = a[i];
}
sort(b+1, b+1+n);
int m = unique(b+1, b+1+n) - (b+1);
for(int i = 1;i <= n;i ++)
a[i] = lower_bound(b+1, b+1+m, a[i]) - b; //离散化操作
tot = 0;
rt[0] = build(1, m);
for(int i = 1;i <= n;i ++)
rt[i] = update(rt[i-1], 1, m, a[i]);
printf("Case %d:\n", ++Case);
while(q --)
{
int l, r, H;
scanf("%d%d%d", &l, &r, &H);
l ++, r ++;
int pos = upper_bound(b+1, b+1+m, H) - b;
pos --;
if(!pos)
{
printf("0\n");
}
else
{
printf("%d\n", query( rt[l-1], rt[r], 1, pos, 1, m ) );
}
}
}
return 0;
}