棋子等级|树状数组单点更新

棋子等级

坐标系平面上有好多棋子,每个整点上至多有一个棋子。
假定棋子的等级是左下方的棋子个数,现在给出若干棋子的位置,求不同等级的棋子各有多少个。左下方包含正下和正左。说明(0,0)坐标的位置在左下角。
输入格式
第一行一个整数N(1≤N≤100000)接下来N行,一行两个整数X,Y(0≤X,Y<100000),表示坐标。
数据保证坐标先按Y排序,再按X排序。
输出格式
N行,每行一个整数,从0到N-1等级的棋子数量。

思路:Y不断增大,所以可以使用树状数组维护“横坐标小于x的棋子数量”。另外注意当x==0的时候 ,change更新函数会死循环,因此需要判断0

代码:

#include<bits/stdc++.h>
using namespace std;

const int MAX_N = 100000 + 5;
int n;
int C[MAX_N]; 
int cnt[MAX_N];
 
int lowbit(int x){
	return x & -x;
}

//求和:各个横坐标下的棋子数量----棋子等级 
int getsum(int x){
	int res = 0;
	while(x){
		res += C[x];
		x -= lowbit(x);	
	}
	return res;
}

//更新 
void change(int x,int c){
	//注意:change(x=0)的时候程序会死循环
	if(x==0){
		C[x] += c;
		x++;
	} 
	while(x<=MAX_N){
		C[x] +=c;
		x += lowbit(x);
	}
}

int main(){
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		int x,y;
		scanf("%d%d",&x,&y);
		cnt[getsum(x)]++;
		change(x,1);//小于x的横坐标上 数量加1 
	}
	//查询每个 x上的点管辖范围内的 棋子数量 
	for(int i=0;i<=n-1;i++){
		printf("%d\n",cnt[i]);
	}
	return 0;
}
posted @ 2019-02-27 13:59  fishers  阅读(325)  评论(0编辑  收藏  举报