POJ 2352 树状数组

这是第一道自己想出来的树状数组题目。。好心酸。。

A了老夫一下午，前40分钟在想是树状数组还是线段树，最后终于在树状数组上找到思路了。。嘤嘤。。。

题目大意是让计算每个star左下方星星的个数，并输出拥有相同个数星星的数目。。

难点就是要自己重写sort()方法，一开始Bit数组开小了，找bug找的我心好累。。恩，就这样吧，A了一题，神清气爽~

PS：还要注意x=0时的情况~

#include<stdio.h>
#include<iostream>
#include<math.h>
#include <algorithm>
#include <string.h>
using namespace std;
#define MAX_N 15005
#define MAX_XY 32005
int Bit[MAX_XY];
int temp[MAX_N];
int ans[MAX_N];

typedef pair <int ,int> p;
p source[MAX_N];

int N;

bool cmp (const p a, const p b)
{
if(a.first>b.first)
{
return 0;
}
else if(a.first<b.first)
{
return 1;
}
else if(a.second>b.second)
{
return 0;
}
else
{
return 1;
}
}
int lowBit(int x)
{
return x&(-x);
}

void update(int x,int c)
{
for(int i=x;i<MAX_XY;i+=lowBit(i))
{
Bit[i]+=c;
}
}

int getsum(int x)
{
int ans = 0;
for(int i=x;i>0;i-=lowBit(i))
{
ans += Bit[i];
}
return ans;
}

int main()
{

scanf("%d",&N);
int a,b;
for(int i=1;i<=N;i++)
{
scanf("%d%d",&source[i].second,&source[i].first);             //注意：第0位没有使用

}
sort(source+1,source+N+1,cmp);
//     for(int i=1;i<=N;i++)
//    {
//        printf("%d  %d\n",source[i].first,source[i].second);
//    }
for(int i=1;i<=N;i++)
{
temp[i]=source[i].second;
}
memset(Bit,0,sizeof(int)*MAX_XY);
memset(ans,0,sizeof(int)*N);
for(int i=1;i<=N;i++)
{
//        printf("temp=%d\n",temp[i]);
//        for(int i=1;i<10;i++) printf("%d ",Bit[i]);
//        printf("\n");
//        printf("%d\n",getsum(temp[i]));
++ans[getsum(temp[i]+1)];
update(temp[i]+1,1);
}
for(int i=0;i<N;i++)
{
printf("%d\n",ans[i]);
}
return 0;
}