[hoj 2678]Stars[二维树状数组]

iYUNDI

于 2013-07-29 19:35:16 发布

阅读量808

点赞数

CC 4.0 BY-SA版权

分类专栏：树状数组

本文链接：https://blog.youkuaiyun.com/iyundi/article/details/9621105

树状数组专栏收录该内容

2 篇文章

订阅专栏

本文介绍了一种使用三维树状数组解决类似POJ Stars问题的方法。通过定义二维树状数组的含义及其在一维上的扩展，文章详细展示了如何利用树状数组进行前缀和查询与更新操作，最终解决三维空间中的问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

二维树状数组的含义:

sum[i][j]表示从lowbit(i)<=x<=i,lowbit(j)<=y<=j之间的元素之和.

题意:

类似poj那道Stars,扩展到了三维.

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int MAXN = 15005;
const int N = 1005;
typedef struct point{
    int x,y,z;
}point;
point p[MAXN];
int c[N][N],s[MAXN];
bool cmp(point a,point b)
{
    if(a.z!=b.z)  return a.z<b.z;
    if(a.y!=b.y)  return a.y<b.y;
    return a.x<b.x;
}
inline int lowbit(int i)
{
    return i&(-i);
}

int GetSum(int x,int y)
{
    int ans = 0;
    for(int i=x; i>0; i-=lowbit(i))
        for(int j=y; j>0; j-=lowbit(j))
            ans += c[i][j];
    return ans;
}

void add(int x,int y, int v)
{
    for(int i=x;i<N;i+=lowbit(i))
        for(int j=y;j<N;j+=lowbit(j))
            c[i][j] += v;
}
int main()
{
    int n;
    while(cin>>n)
    {
        memset(c,0,sizeof(c));
        memset(s,0,sizeof(s));
        for(int i=1;i<=n;i++)
            cin>>p[i].x>>p[i].y>>p[i].z;
        sort(p+1,p+n+1,cmp);
        for(int k=1;k<=n;k++)
        {
            s[GetSum(p[k].x+1,p[k].y+1)]++;
            add(p[k].x+1,p[k].y+1,1);
        }
        for(int i=0;i<n-1;i++)
            cout<<s[i]<<" ";
        cout<<s[n-1]<<endl;
    }
    return 0;
}