题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4417
题意:给n个数,数中有重复的。还有m个询问,问的是[L,R] 区间内有多少个数小于h,有多次询问。
想到是树状数组但是没想出来怎么做。。。也不是第一次做离线处理的题了,可就是没往这里想。。。。不过队友用主席树也过了,应该不太简洁。
贴一份比较清晰的解释:
① 先把所有位置的高度都存下来,然后排序,注意保存下标;
② 把所有询问存下来,然后按照询问的高度进行排序,同注意保存下标;
③ 对于排序后的每次询问的处理:由于每个位置的高度都已经存了下来并且进行了排序,所以可以按照顺序将每个点插入到线段树的对应位置(保存的下标),并更新线段树,直到要插入的位置的高度大于这次询问的高度H;最后处理区间查询,由于刚才已经把小于等于该次查询高度的位置都已经插入到了线段树中,所以询问的结果就是查询区间中被更新过的叶子节点的个数,也就是区间求和问题。
参考博客:http://www.cnblogs.com/H-Vking/p/4296171.html
#include <bits/stdc++.h>
using namespace std;
const int maxn = 100100;
int sum[maxn], ans[maxn];
int n, m;
struct stone {
int h, pos;
};
stone sto[maxn];
struct query {
int l, r, h, id;
};
query que[maxn];
bool cmp1(stone a, stone b) {
return a.h < b.h;
}
bool cmp2(query a, query b) {
return a.h < b.h;
}
int lowbit(int x) {
return x & (-x);
}
void update(int index, int val) {
int i;
for(i = index; i <= n; i += lowbit(i)) {
sum[i] += val;
}
}
int getsum(int index) {
int i;
int res = 0;
for(i = index; i > 0; i -= lowbit(i)) {
res += sum[i];
}
return res;
}
int main() {
int t;
scanf("%d", &t);
for(int kase = 1; kase <= t; kase++) {
memset(sum, 0, sizeof(sum));
printf("Case %d:\n", kase);
scanf("%d %d", &n, &m);
for(int i = 1; i <= n; i++) {
scanf("%d", &sto[i].h);
sto[i].pos = i;
}
sort(sto + 1, sto + n + 1, cmp1);
for(int i = 1; i <= m; i++) {
scanf("%d %d %d", &que[i].l, &que[i].r, &que[i].h);
que[i].l++;
que[i].r++;
que[i].id = i;
}
sort(que + 1, que + m + 1, cmp2);
int j = 1;
for(int i = 1; i <= m; i++) {
while(j <= n && sto[j].h <= que[i].h) {
update(sto[j].pos, 1);
j++;
}
ans[que[i].id] = getsum(que[i].r) - getsum(que[i].l - 1);
}
for(int i = 1; i <= m; i++) printf("%d\n", ans[i]);
}
return 0;
}