POJ 2352|URAL 1028|Stars|树状数组

http://acm.timus.ru/problem.aspx?space=1&num=1028

题目大意

棋盘中有n个棋子，一个棋子的等级定义为其左下的所有棋子的数目（包括正左方和正下方的棋子）。输出等级0~(n-1)的棋子的数目

题目

天文学家经常检查星图，即用一个点表示一颗星，一个平面表示宇宙，故每颗星都有自己的坐标。定义一颗星的等级为一些不高于且不在这颗星右边的星星的数目。天文学家希望知道每个星星等级内有多少颗星星。

比如，观察上面这个地图。5号星的等级为3（因为1、2、4在其左下方）。2和4号星的等级为1，只有1个星星的等级为0，2个为等级1，1个为等级3。
你要写个程序计算给定的星图里每个等级的星星的数目。

输入

输入的第一行一个整数$N(1\leq N\leq 15000)$，表示星星的个数。
接下来N行每行2个正数$X,Y(0\leq X,Y\leq 32000)$表示一个星星在平面上的坐标。输入数据按Y坐标为第一关键字，X坐标为第二关键字已经排好序了。

输出

输出N行，每行一个数，表示等级0~n-1各等级有多少颗星星。

样例输入

5
1 1
5 1
7 1
3 3
5 5

样例输出

1
2
1
1
0

题解

如果棋盘比较小，我们可以定义$sum_{i,j}$表示左下方棋子数目的和然后暴力一下就好了。但是棋盘比较大，考虑扫描线。扫描i，那么对于每一层，我们用某种数据结构维护i层所有棋子的左下方棋子的数目，再推出第i+1层即可。显然每次我们只需要查询1~i层的所有棋子数目，所以用一个树状数组维护即可：a[j]表示1~i行，1~j列的棋子数目，每次查询一下就好。

#include <cstdio>
#define FOR(i,j,k) for(i=j;i<=k;++i)
const int N = 16384;
int c[32768], x[N], y[N], ans[N];

void add(int i, int x) {
    for (; i <= 33000; i += i & -i)
        c[i] += x;
}

int sum(int i) {
    int s = 0;
    for (; i > 0; i -= i & -i)
        s += c[i];
    return s;
}

int main() {
    int i, n;

    scanf("%d", &n);
    FOR(i,1,n) scanf("%d%d", &x[i], &y[i]), ++x[i], ++y[i];
    FOR(i,1,n) {
        ++ans[sum(x[i])];
        add(x[i], 1);
    }
    FOR(i,0,n-1) printf("%d\n", ans[i]);

    return 0;
}