HDU1541树状数组

Posted 2021-09-03 hesorchen

题目

Stars

二维平面中给出若干点，一个点的贡献是在他最下方的点的数量，
即
$$F(i)=\sum _{j=1}^n(x_i\ge x_j\&\&y_i\ge y_j)$$

对于$i=[0,n-1]$求$F(i)$。

看了好几遍也没看见多组输入… …

解题思路

对x、y其中一维排序，然后维护另一维顺序对即可。

代码

#include <bits/stdc++.h>
using namespace std;

const int N = 32000 + 5;
int c[N];

int n;
int mx;
void add(int i)
{
    while (i < N)
    {
        c[i] += 1;
        i += (i & -i);
    }
}

int get(int i)
{
    int sum = 0;
    while (i)
    {
        sum += c[i];
        i -= (i & -i);
    }
    return sum;
}
struct node
{
    int x, y;
} a[N];
int ans[N];

int main()
{
    while (~scanf("%d", &n))
    {
        memset(c, 0, sizeof c);
        for (int i = 1; i <= n; ++i)
            ans[i - 1] = 0;
        for (int i = 1; i <= n; ++i)
        {
            int x, y;
            scanf("%d %d", &x, &y);
            ++x;
            ++ans[get(x)];
            add(x);
        }
        for (int i = 0; i <= n - 1; ++i)
            printf("%d\\n", ans[i]);
    }
    return 0;
}