POJ 2352 Stars(树状数组)

来源:互联网 发布:救援夫妇知乎 编辑:程序博客网 时间:2024/05/21 10:03

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

/*刚学完树状数组 采用树状数组做

    c[i]记录的是 i管辖区域内元素的个数
    level[i] 记录的是左下角星星个数为i的点的个数总和 
    因为输入是按照Y的递增输入的 如果输入的Y的大小相同,那么后者输入的X一定比前者大
    按照这个规律所以后者输入的不可能会出现在前者的左下角区域
    解题采取边输入边处理更新的办法 
    还有一个细节问题就是坐标轴是从0开始的,但是树状数组下标是从1开始的,因此需要整体右移1
*/

#include <iostream>
#include "stdio.h"
#include "string.h"

using namespace std;

int level[32005],c[32005];
int lowbit(int x) { return x&(-x);}

void add(int x)  //增加坐标为X的元素个数即c[x]++,同时继续更改后面会影响到的元素 
{
      while(x<32005)
      {
            c[x]++;
            x+=lowbit(x);
      }
}
int sum(int x)  //求新增元素的等级(左下角所有元素的总和)
{
      int ans=0;
      while(x>0)
      {
            ans+=c[x];
            x-=lowbit(x);
      }
      return ans;
}
int main()
{
      int n;
      scanf("%d",&n);
      memset(level,0,sizeof(level));
      memset(c,0,sizeof(c));
      for(int i=0;i<n;i++)
      {
            int a,b;
            scanf("%d%d",&a,&b);
            level[sum(++a)]++; //树状数组是从下标1开始的,但是坐标是从0开始的,所以整体右移1
            add(a);
      }
      for(int i=0;i<n;i++)
            printf("%d\n",level[i]);
      return 0;
}

0 0
原创粉丝点击