HDU1541 Stars 树状数组

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1541


题目大意:有n颗星星,每颗星星的权值为它左下角的星星的个数(不包括它本身),现在按y升序(y上按x升序)给出你n个星星的坐标,输出权值为0,1,...,n-1的星星的数目。


分析:刚开始用二维树状数组做的,MLE了,没用到题中给出的输入次序。按照题中的输入次序,由于是按y升序输入的,那么对于当前的星星来说,它前面是没有比它的高的星星,那么很显然,它的权值就是前面的星星的个数。这样我们只需要用一个一维的树状数组存下i前面x轴上有多少颗星星几颗。另外注意下坐标是从0开始的。


实现代码如下:

#include <cstdio>
#include <cstring>
using namespace std;
const int M=32005;
int a[M],n,ans[M];
int lowbit(int i)
{
    return i&(-i);
}
void update(int i,int x)
{
    while(i<=M)
    {
        a[i]+=x;
        i+=lowbit(i);
    }
}
int query(int n)
{
    int sum=0;
    while(n>0)
    {
        sum+=a[n];
        n-=lowbit(n);
    }
    return sum;
}
int main()
{
    while(scanf("%d",&n)!=-1)
    {
        memset(a,0,sizeof(a));
        memset(ans,0,sizeof(ans));
        for(int i=0;i<n;i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            ans[ query(x+1) ]++;
            update(x+1,1);
        }
        for(int i=0;i<n;i++)
          printf("%d\n",ans[i]);
    }
    return 0;
}


你可能感兴趣的:(HDU1541 Stars 树状数组)