一、思路
- 通过upper_bound() 函数来找到最后一个大于k的数所在的下标，然后查询在某个区间内，【1，h】中所有数出现的次数就可以了。
二、代码
#include <cstdio>
#include <vector>
#include <algorithm>
#include <iostream>
using namespace std;
const int maxn = 1e5 + 7;
struct node {
int l, r, sum; //sum是包含几个元素
} T[maxn * 50];
int n, m, cnt, a[maxn], root[maxn], x, y, k; //root用来保存某个节点的线段树的头节点
vector<int> v;
int getId(int x) {
return lower_bound(v.begin(), v.end(), x) - v.begin() + 1;
}
// pos代表要进行加1的根节点 x是当前构造的线段树 y是上一次构造的线段树
void update(int l, int r, int &x, int y, int pos) {
T[++cnt] = T[y]; //先创造一个节点等于上一课线段树
T[cnt].sum++; //这创造出来的树加上1 因为要包含新的点进来
x = cnt; //root数组 保存下这棵树的头节点
if (l == r) return; //已经到了根节点不用在进行创造了
int mid = (l + r) >> 1;
if (pos <= mid) {
//对左节点进行创造
update(l, mid, T[x].l, T[y].l, pos); //将y线段树的左节点传进去， 因为还要保存另外一个左（右）区间
} else {
update(mid + 1, r, T[x].r, T[y].r, pos);
}
}
//k 是第几小的数 y是后面的树 x是l区间前的树
int query(int l, int r, int x, int y, int k) {
if (r <= k) {
return T[y].sum - T[x].sum;
}
int s = T[T[y].l].sum - T[T[x].l].sum;
int mid = (l + r) >> 1;
if (mid < k) {
return s + query(mid + 1, r, T[x].r, T[y].r, k);
} else {
return query(l, mid, T[x].l, T[y].l, k);
}
}
int main() {
int t, num = 1;
scanf("%d", &t);
while (t--) {
cnt = 0;
v.clear();
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++) {
//根据上一个树 来创造这次的线段树 root[i - 1]上一棵树的根节点 root[i]当前线段树的根节点
update(1, n, root[i], root[i - 1], getId(a[i]));
}
printf("Case %d:\n", num++);
for (int i = 1; i <= m; i++) {
scanf("%d %d %d", &x, &y, &k);
x++;y++;
//upper_bound 返回最后一个小于等于它的数 所以我们在[1, h] 区间内查找符合要求的
int h = upper_bound(v.begin(), v.end(), k) - v.begin();
if (h > 0) {
printf("%d\n", query(1, n, root[x - 1], root[y], h)); //因为v数组是从0开始
} else {
printf("0\n");
}
}
}
return 0;
}
