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