数据结构 (习题): 树状数组相关

PKU 2352

链接在这里.
题目大意: 一个点的 order 是不包括它在内的所有在它左下方的点的总数, 问你点集中每个 order 的点的数量.

思路

树状数组的牛逼之处在于, 它可以动态地查询比当前值小的元素个数 (要求取值范围不要太大). 在这题当中, 因为有两个维度, 所以我们可以先通过排序 (题目已经提前排好序了, 所以这道题不用排序) 将数据按照两个维度按照从小到大的顺序排好, 比如题目中它已经为你按照 y 轴第一顺序, x 轴第二顺序排好, 那么当你访问到第 n 个点时, 前面 n-1 个点必定在你的下方, 而后面的所有点必定要么在你上方, 要么在你右方, 反正都不可能在你左下方, 那么你只需要统计目前为止所有横坐标小于该点的元素数量. 方法是维护一个数组 (BIT), 每个元素表示一个横坐标的个数, 当遍历到第 n 个点时, 只需要求 $prefix(x_n)$ 的值即可.

代码

// 以下是上一个博客的模板

#define MAXN 40000
int bit[MAXN];

int lowbit(int x) {
   return x&-x;}
//void init() {memset(bit, 0, sizeof(bit));}

void add(int pos, int val) {
   
	while (pos < MAXN) {
   
		bit[pos] += val;
		pos += lowbit(pos);
	}
}

int prefixSum(int pos) {
   
	int sum = 0;
	while (pos > 0) {
   
		sum += bit[pos];
		pos -= lowbit(pos);
	}
	return sum;
}

int rangeSum(int from, int to) {
   
	return prefixSum(to) - prefixSum(from-1);
}

// 以上是上一个博客的模板

#include <stdio.h>
#include <stdlib.h>
int count[15000];
int main() {
   
	int n;
	scanf("%d", &n);
	for (int i = 0; i < n; i++) {