poj 2352 Stars 树状数组or线段树

题目大意:
在坐标上有n个星星,如果某个星星坐标为(x, y), 它的左下位置为:(x0,y0),x0<=x 且y0<=y。如果左下位置有a个星星,就表示这个星星属于level x
按照y递增,如果y相同则x递增的顺序给出n个星星,求出所有level水平的数量。

分析:
因为输入是按照按照y递增,如果y相同则x递增的顺序给出的, 所以,对于第i颗星星,它的level就是之前出现过的星星中,横坐标x小于等于i星横坐标的那些星星的总数量(前面的y一定比后面的y小)。
所以,需要找到一种数据结构来记录所有星星的x值,方便的求出所有值为0~x的星星总数量。
树状数组和线段树都是很适合处理这种问题。

树状数组:

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=32005;
int c[N],level[N];
int lowbit(int x){return x&(-x);}
int getsum(int x)
{
    int sum =0;
    for(int i=x;i>0;i-=lowbit(i))sum+=c[i];
    return sum;
}
void add(int x)
{
    for(int i=x;i<N;i+=lowbit(i))c[i]++;
}
int main()
{
    int n,x,y;
    while(~scanf("%d",&n)){
        memset(level,0,sizeof(level));
        memset(c,0,sizeof(c));
        for(int i=0;i<n;i++){
            scanf("%d%d",&x,&y);
            ++x;
            level[getsum(x)]++;
            add(x);
        }
        for(int i=0;i<n;i++)
            printf("%d\n",level[i]);
    }
    return 0;
}

线段树:

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int N=15005;
const int M=32009;
int sum[M<<2];
int level[N];
void pushup(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
int query(int a,int b,int l,int r,int rt)
{
    if(a<=l&&r<=b)return sum[rt];
    int m=(l+r)>>1;
    int ans=0;
    if(a<=m)ans=query(a,b,lson);
    if(b>m)ans+=query(a,b,rson);
    return ans;
}
void update(int x,int l,int r,int rt)
{
    if(l==r){sum[rt]++;return;}
    int m=(l+r)>>1;
    if(x<=m)update(x,lson);
    else update(x,rson);
    pushup(rt);
}
int main()
{
    int n,x,y;
    scanf("%d",&n);
        memset(sum,0,sizeof(sum));
        memset(level,0,sizeof(level));
        for(int i=0;i<n;i++){
            scanf("%d%d",&x,&y);
            int t=query(0,x,0,M,1);
            level[t]++;
            update(x,0,M,1);
        }
        for(int i=0;i<n;i++)
            printf("%d\n",level[i]);
    return 0;
}

你可能感兴趣的:(poj 2352 Stars 树状数组or线段树)