POJ2352-Stars【树状数组】

正题

题目链接:
http://poj.org/problem?id=2352


题意

有n个坐标不同的星星,一个星星的等级等于在它左下的所有星星数量,求各个等级的星星数。
注:输入顺序保证Y坐标是升序,Y坐标相同的情况下X坐标也是升序的。


解题思路

用树状数组表示在x坐标上有多少个星星,因为Y坐标升序输入所以不用考虑Y坐标,然后用树状数组求1到x坐标上的星星总数就好了。


代码

#include<cstdio>
#include<algorithm>
using namespace std;
int c[82010],w[15001],n,s,pre,a;
int lowbit(int x)
{return x&(x^(x-1));}
void change(int x,int num)//修改
{
    int i=x;
    while(i<=82000)
    {
        c[i]+=num;
        i+=lowbit(i);
    }
}
int getsum(int x)//求值
{
    int sum=0;
    while (x>0)
    {
        sum+=c[x];
        x-=lowbit(x);
    }
    return sum;
}
int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
    {
        scanf("%d%d",&a,&w[i]);
        a++;//避免读入0死循环
        w[i]=getsum(a);//就等级
        change(a,1);//插入该星星
    }
    sort(w+1,w+1+n);//排序
    pre=w[1];s=1;
    for (int i=2;i<=n;i++)
    {
        if (w[i]!=pre)//等级不相同
        {
            printf("%d\n",s);
            for (int j=pre+1;j<w[i];j++) //输出中间
              printf("0\n");
            pre=w[i];
            s=1;
        }
        else s++;
    }
    printf("%d\n",s);
    for (int i=pre+1;i<n;i++) printf("0\n");//输出剩余的
}

猜你喜欢

转载自blog.csdn.net/Mr_wuyongcong/article/details/80285844