poj 2481 樹狀陣列

caoxiaoran1202發表於2013-08-22

一開始根本沒想到是樹狀陣列,因為有兩個端點需要處理,後來經提醒,排序後只剩下一個端點,接下來樹狀陣列就ok了。

注意:題目中下標從0開始,所以輸入後都要加1,以滿足樹狀陣列下標從1開始的性質。

排序後一個一個處理,前面的比後面強壯,或區間相同,處理時排除區間相同的即可。

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<set>
#include<map>
#include<queue>
#include<stack>
#include<vector>
using namespace std;
inline void RD(int &ret)
{
    char c;
    do
    {
        c=getchar();
    }while(c<'0'||c>'9');
    ret=c-'0';
    while((c=getchar())>='0'&&c<='9')
        ret=ret*10+(c-'0');
}
inline void OT(int a)
{
    char c;
    if(a>=10)
    {
        OT(a/10);
    }
    putchar(a%10+'0');
}
struct node
{
    int b,e;
    int pos;
}s[100005],pre;
int n;
int c[100005];
int re[100005];
bool cmp(node x,node y)
{
    if(x.e==y.e)
        return x.b<y.b;
    else
        return x.e>y.e;
}
int lowbit(int x)
{
    return x&(-x);
}
void add(int i,int detal)
{
    while(i<=100002)
    {
        c[i]+=detal;
        i+=lowbit(i);
    }
}
int sum(int k)
{
    int ret=0;
    while(k>0)
    {
      ret+=c[k];
      k-=lowbit(k);
    }
    return ret;
}
int main()
{
    while(scanf("%d",&n)&&n)
    {
        memset(c,0,sizeof(c));
        //memset(re,0,sizeof(re));
        for(int i=1;i<=n;i++)
        {
            scanf("%d%d",&s[i].b,&s[i].e);
            s[i].pos=i;
            s[i].b+=1;
            s[i].e+=1;
        }
        sort(s+1,s+1+n,cmp);

        pre.b=-1;
        pre.e=-1;
        int ans=0;
        //因為經過排序後後面沒有比前面還強大的,所以一個一個處理
        //ans的記錄也很神奇,排除排序區間相同的
        for(int i=1;i<=n;i++)
        {
            if(s[i].b==pre.b&&s[i].e==pre.e) ans++;
            else
            {
                ans=0;
                pre.b=s[i].b;
                pre.e=s[i].e;
            }
            re[s[i].pos]=sum(s[i].b)-ans;
            add(s[i].b,1);
        }
        for(int i=1;i<=n;i++)
        {
            printf("%d ",re[i]);
        }
        cout<<endl;
    }
    return 0;
}


 

相關文章