HDU 4417 Super Mario // 线段树

题目描述

解题思路

参考代码

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
#define lson rt<<1
#define rson rt<<1|1
#define mid ((l+r)>>1)
using namespace std;
typedef vector<int> vi;
const int MAX_N = 100010;
int num[MAX_N];
vi v[MAX_N << 2];

void build(int l, int r, int rt) {//建树
v[rt].clear();
for(int i = l; i <= r; i++)// 把 l 到 r 的元素都存在vector中
v[rt].push_back(num[i]);
sort(v[rt].begin(), v[rt].end());//排序方便后面二分查找
if(l == r) return ;
build(l, mid, lson);
build(mid+1, r, rson);
}

int query(int L, int R, int val, int l, int r, int rt) {//查询
if(L <= l && r <= R) {//当查待询区间[L, R] 完全包含 [l, r] 则返回该区间的结果
vi::iterator it;
it = upper_bound(v[rt].begin(), v[rt].end(), val);
//找到第一个比val大的位置
return it - v[rt].begin();//相减得到小于等于val的个数
}
int ans = 0;
if(L <= mid) ans += query(L, R, val, l, mid, lson);
// 有一部分区间在 [l, mid] 之间
if(mid < R) ans += query(L, R, val, mid+1, r, rson);
//有一部分在 [mid+1, r] 之间
return ans;
}
int main() {
int T, l, r, h, n, m;
while(~scanf("%d", &T)) {
for(int k = 1; k <= T; k++) {
scanf("%d %d", &n, &m);
for(int i = 1; i <= n; i++)
scanf("%d", &num[i]);
build(1, n, 1);
printf("Case %d:\n", k);
while(m--) {
scanf("%d %d %d", &l, &r, &h);
printf("%d\n", query(l+1, r+1, h, 1 , n , 1));
}
}
}
return 0;
}