AcWing：数星星

题目入口

题目跳转

题目描述

天空中有一些星星，这些星星都在不同的位置，每个星星有个坐标。

如果一个星星的左下方（包含正左和正下）有 k 颗星星，就说这颗星星是 k 级的。

例如，上图中星星 5 是 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

思路讲解

因为输入的时候不会有星星重叠。星星按 y 坐标增序给出，y 坐标相同的按 x 坐标增序给出。所以当我们计算坐标为（a，b）的星星是几级的时候，只需考虑已经输入了的星星且x坐标小于等于a的星星即可，因为还暂未输入的星星要么是x坐标大于已有的星星，要么是y坐标大于已有的星星，在计算星星是第几级的时候这些未输入的星星都不在考虑之内，这就转化成了区间查询和单点修改的问题，因此我们可以使用树状数组来解决这个问题，区间查询对应的是x坐标小于等于当前输入星星的x坐标a的星星的个数，我们设A[i]表示x坐标为i的星星的个数，于是转化为A[1]+A[2]+…+A[a],单点修改对应的是，查询过当前输入的星星属于第几级之后将该星星加入进来。

代码

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>

using namespace std;

const int N = 32010;
const int M = 15010;
int n;
int tr[N];
//level[i]表示第i级星星的个数
int level[M];

int lowbit(int x)
{
    return x & -x;
}

void add(int x,int v)
{
    for(int i = x;i < N;i += lowbit(i))
    {
        tr[i] += v;
    }
}

int query(int x)
{
    int res = 0;
    for(int i = x;i > 0;i -= lowbit(i))
    {
        res+= tr[i];
    }
    return res;
}

int main()
{
    scanf("%d",&n);
    for(int i = 0;i < n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        x++;
        level[query(x)]++;
        add(x,1);
    }
    for(int i = 0;i < n;i++)
    {
        printf("%d\n",level[i]);
    }
    return 0;
}