C++数星星(树状数组)

    科技2022-07-17  124

    天空中有一些星星,这些星星都在不同的位置,每个星星有个坐标。 如果一个星星的左下方(包含正左和正下)有 k 颗星星,就说这颗星星是 k级的。

    例如,上图中星星 是 3 级的(1,2,4 在它左下),星星 2,4 是 1级的。 例图中有 1个 0 级,2 个 1 级,1 个 2 级,1 个 3级的星星。 给定星星的位置,输出各级星星的数目。换句话说,给定 N个点,定义每个点的等级是在该点左下方(含正左、正下)的点的数目,试统计每个等级有多少个点。 输入格式 第一行一个整数 N,表示星星的数目;接下来 N行给出每颗星星的坐标,坐标用两个整数 x,y表示;不会有星星重叠。星星按 y坐标增序给出,y 坐标相同的按 x坐标增序给出。 输出格式 N行,每行一个整数,分别是 0 级,1 级,2 级,……,N−1级的星星的数目。 数据范围 1≤N≤15000,0≤x,y≤32000 输入样例: 5 1 1 5 1 7 1 3 3 5 5 输出样例: 1 2 1 1 0

    读题知每个星星左下方的星星数据一定先于这颗星星给出,可以每读一颗星星就算这颗星星的级数,这样将一层一层地从下往上、从左往右计算每颗星星的级数,这颗星星的级数即除它自己以外的以x为坐标的前缀和,故要先查询前缀和再添加这颗星星的数据。显然这是一个动态维护前缀和问题,可以用树状数组。

    AC代码:

    #include<stdio.h> const int X=32010; const int N=15010; int level[N],tr[X]; int lowbit(int x) { return x & (-x); } void add(int x)//在x的位置加1 { for(int i=x;i<X;i+=lowbit(i)) ++tr[i]; } int query(int x)//查询从第1项到第x项的加和 { int sum=0; for(int i=x;i>0;i-=lowbit(i)) sum+=tr[i]; return sum; } int main() { int n; scanf("%d",&n); int x,y; for(int i=0;i<n;++i) { scanf("%d%d",&x,&y); ++x;//树状数组从1开始计数 ++level[query(x)]; add(x); } for(int i=0;i<n;++i) printf("%d\n",level[i]); return 0; }
    Processed: 0.009, SQL: 8