当前位置: 代码迷 >> 综合 >> AcWing 1265. 数星星 树状数组
  详细解决方案

AcWing 1265. 数星星 树状数组

热度:57   发布时间:2023-11-23 13:48:48.0

AcWing 1265. 数星星
天空中有一些星星,这些星星都在不同的位置,每个星星有个坐标。

如果一个星星的左下方(包含正左和正下)有 k 颗星星,就说这颗星星是 k 级的。

1.png

例如,上图中星星 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是递增的,那这问题就很简单了,我们动态的去求出小于等于x 的点数就可以了。
然后需要值得注意的地方,一定一定要注意,前缀和add(0,k)会导致错误的,因为lowbit(0)=0,所以从零开始的数组变成从一开始的数组,这样减少一些不必要的bug。
然后我们思路就很清晰,我们也可以得出代码了。
代码如下:

#include<iostream>using namespace std;const int N=33000;int d[N];
int n;
int level[15010];int lowbit(int x)
{
    return x&-x;
}void add(int k)
{
    for(int i=k;i<N;i+=lowbit(i)) d[i]++;
}int query(int k)
{
    int res=0;for(int i=k;i;i-=lowbit(i)) res+=d[i];return res;
}
int main(void)
{
    cin>>n;for(int i=1;i<=n;i++){
    int a,b;cin>>a>>b;a++;//需要注意a++int t=query(a);//先查询后添加,避免加上自己。level[t]++;add(a);}for(int i=0;i<n;i++)cout<<level[i]<<endl;
}