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

棋子等级

坐标系平面上有好多棋子，每个整点上至多有一个棋子。
假定棋子的等级是左下方的棋子个数，现在给出若干棋子的位置，求不同等级的棋子各有多少个。左下方包含正下和正左。说明（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;
}

转载于:https://www.cnblogs.com/fisherss/p/10443292.html