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;
}