HDU-1541 Starsツリー配列

3297 ワード

http://acm.hdu.edu.cn/showproblem.php?pid=1541
ある点の左下隅の星の数を求めて、ちょうど2次元の木の形の配列でして、結果はきっとメモリが使えないのです.
この問題は、与えられた座標点の並べ替え後、x軸座標に対して1次元配列を確立するだけで、現在の状態に対してx軸で平面をM個の領域に分割し、与えられた点のy軸座標は現在最も高いに違いないので、横座標に対して直接和を求めればよいと正解している.
コードは次のとおりです.
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <map>
#define MAXN 32005
using namespace std;

int N, res[MAXN+5], c[MAXN+5];

inline int lowbit(int x)
{
return x & -x;
}

inline void modify(int pos, int val)
{
for (int i = pos; i <= MAXN; i += lowbit(i))
c[i] += val;
}

inline int getsum(int pos)
{
int s = 0;
for (int i = pos; i > 0; i -= lowbit(i))
{
s += c[i];
}
return s;
}

int main()
{
int x, y;
while (scanf("%d", &N) == 1)
{
memset(c, 0, sizeof (c));
memset(res, 0, sizeof (res));
for (int i = 0; i < N; ++i)
{
scanf("%d %d", &x, &y);
x++;
res[getsum(x)]++;
modify(x, 1);
}
for (int i = 0; i < N; ++i)
printf("%d
", res[i]);
}
return 0;
}