POJ 2481 Cows 树状数组

【题目大意】

有一条一维的山脊，山脊上有FJ的牛，每头牛都有自己的领地[Si,Ei]。

如果有[Si,Ei]包含于[Sj,Ej]且不完全相等。则说J牛比I牛强壮。

现在FJ想知道对于每头牛，有多少头牛比他强壮。

【解题思路】

利用树状数组解区间问题。

解法一：

按e的降序排列，e相等时,s小的在前。这样通过查询s可以知道区间的情况。

解法二：

按s的升序排列，s相等时，按照e的降序排列。

【代码】

解法一：

#include<iostream>
#include<string.h>
#include<cstdio>
#include<algorithm>
using namespace std;

struct inter{
int s,e,r;
}p[222222];
int top;
int ta[222222],rec[111111];
inline int lowbit(int x){ return x&-x; }

bool cmp( inter a,inter b ){
if( a.e==b.e )
return a.s<b.s;
return a.e>b.e;
}

int query( int at )
{
int ret=0;
while( at ){
ret+=ta[at];
at-=lowbit(at);
}
return ret;
}

void add( int at,int v )
{
while( at<=100000 ){
ta[at]+=v;
at+=lowbit(at);
}
}

int main()
{
int n;
while( scanf("%d",&n)!=EOF )
{
memset( ta,0,sizeof(ta) );
if( n==0 )
break;
for( int i=0;i<n;i++ )
{
scanf( "%d%d",&p[i].s,&p[i].e );
p[i].s;p[i].e;
p[i].r=i;
}
sort( p,p+n,cmp );
int b;
for( int i=0;i<n;i++ )
{
b=query( p[i].s+1 );
add( p[i].s+1,1 );
if( i!=0 && p[i].s==p[i-1].s && p[i].e==p[i-1].e )
rec[p[i].r]=rec[p[i-1].r];
else
rec[p[i].r]=b;
}
for( int i=0;i<n-1;i++ )
printf( "%d ",rec[i] );
printf( "%d\n",rec[n-1] );
}
return 0;
}
解法二:

#include<iostream>
#include<string.h>
#include<cstdio>
#include<algorithm>
using namespace std;

struct inter{
int s,e,r;
}p[222222];
int top;
int ta[222222],rec[111111];
inline int lowbit(int x){ return x&-x; }

bool cmp( inter a,inter b ){
if( a.s==b.s )
return a.e>b.e;
return a.s<b.s;
}

inline int query( int at )
{
int ret=0;
while( at ){
ret+=ta[at];
at-=lowbit(at);
}
return ret;
}

inline void add( int at,int v )
{
while( at<=top ){
ta[at]+=v;
at+=lowbit(at);
}
}

int main()
{
int n;
while( scanf("%d",&n)!=EOF )
{
memset( ta,0,sizeof(ta) );
memset( rec,0,sizeof(rec) );
if( n==0 )
break;
top=0;
for( int i=0;i<n;i++ )
{
scanf( "%d%d",&p[i].s,&p[i].e );
p[i].s++,p[i].e++;
p[i].r=i;
top=max( top,p[i].e );
}
top=top<<1|1;
sort( p,p+n,cmp );
int a,b;
for( int i=0;i<n;i++ )
{
b=query( p[i].e<<1 );
add( p[i].e<<1|1,-1 );
add( p[i].s<<1,1 );
if( i!=0 && p[i].s==p[i-1].s && p[i].e==p[i-1].e )
rec[p[i].r]=rec[p[i-1].r];
else
rec[p[i].r]=b;
}
for( int i=0;i<n-1;i++ )
printf( "%d ",rec[i] );
printf( "%d\n",rec[n-1] );
}
return 0;
}