注册 登录  
 加关注
   显示下一条  |  关闭
温馨提示!由于新浪微博认证机制调整,您的新浪微博帐号绑定已过期,请重新绑定!立即重新绑定新浪微博》  |  关闭

GreenCloudS

 
 
 

日志

 
 

BZOJ-3631: [JLOI2014]松鼠的新家(LCA)  

2014-07-13 12:41:00|  分类: oi,bzoj,数据结构 |  标签: |举报 |字号 订阅

  下载LOFTER 我的照片书  |

题目:http://www.lydsy.com/JudgeOnline/problem.php?id=3631


裸裸的求个LCA,然后树上前缀和维护一下就好啦~


代码(倍增+DFS似乎有点慢,其实这题可以完全O(n)的额):

  • #include <cstdio>

  • #include <algorithm>

  • #include <cstring>

  •  

  • using namespace std ;

  •  

  • #define REP( i , l , r ) for ( int i = l ; i <= r ; ++ i )

  •  

  • const int maxn = 301000 ;

  • const int maxm = maxn << 1 ;

  •  

  • struct edge {

  •     edge *next ;

  •     int t ;

  • } E[ maxm ] ;

  •  

  • edge *pt = E , *head[ maxn ] ;

  •  

  • #define Init_edge memset( head , 0 , sizeof( head ) ) ;

  •  

  • inline void add( int s , int t ) {

  •     pt -> t = t , pt -> next = head[ s ] ; head[ s ] = pt ++ ;

  • }

  •  

  • inline void addedge( int s , int t ) {

  •     add( s , t ) , add( t , s ) ;

  • }

  •  

  • int up[ maxn ][ 21 ] , a[ maxn ] , n , h[ maxn ] ;

  •  

  • #define travel( x ) for ( edge *p = head[ x ] ; p ; p = p -> next )

  •  

  • void getfa( int now , int fa ) {

  •     up[ now ][ 0 ] = fa , h[ now ] = h[ fa ] + 1 ;

  •     travel( now ) if ( p -> t != fa ) getfa( p -> t , now ) ;

  • }

  •  

  • #define DOWN( i , r , l ) for ( int i = r ; i >= l ; -- i )

  •  

  • inline int getlca( int v , int u ) {

  •     if ( h[ v ] < h[ u ] ) swap( v , u ) ;

  •     DOWN( i , 20 , 0 ) if ( h[ up[ v ][ i ] ] >= h[ u ] ) v = up[ v ][ i ] ;

  •     if ( v == u ) return v ;

  •     DOWN( i , 20 , 0 ) if ( up[ v ][ i ] != up[ u ][ i ] ) {

  •         v = up[ v ][ i ] , u = up[ u ][ i ] ;

  •     }

  •     return up[ v ][ 0 ] ;

  • }

  •  

  • int sm[ maxn ] ;

  •  

  • void getsm( int now , int fa ) {

  •     travel( now ) if ( p -> t != fa ) {

  •         getsm( p -> t , now ) ; sm[ now ] += sm[ p -> t ] ;

  •     }

  • }

  •  

  • int main(  ) {

  •     scanf( "%d" , &n ) ;

  •     REP( i , 1 , n ) scanf( "%d" , a + i ) ;

  •     Init_edge ;

  •     REP( i , 2 , n ) {

  •         int s , t ; scanf( "%d%d" , &s , &t ) ; addedge( s , t ) ;

  •     }

  •     h[ 0 ] = 0 ; getfa( 1 , 0 ) ;

  •     REP( i , 1 , 20 ) REP( j , 1 , n ) up[ j ][ i ] = up[ up[ j ][ i - 1 ] ][ i - 1 ] ;

  •     memset( sm , 0 , sizeof( sm ) ) ;

  •     REP( i , 2 , n ) {

  •         int lca = getlca( a[ i - 1 ] , a[ i ] ) ;

  •         sm[ a[ i - 1 ] ] ++ , sm[ a[ i ] ] ++ , sm[ lca ] -- , sm[ up[ lca ][ 0 ] ] -- ;

  •     }

  •     getsm( 1 , 0 ) ;

  •     REP( i , 1 , n ) printf( "%d\n" , sm[ i ] - ( i != a[ 1 ] ) ) ;

  •     return 0 ;

  • }



  评论这张
 
阅读(3)| 评论(0)
推荐 转载

历史上的今天

评论

<#--最新日志,群博日志--> <#--推荐日志--> <#--引用记录--> <#--博主推荐--> <#--随机阅读--> <#--首页推荐--> <#--历史上的今天--> <#--被推荐日志--> <#--上一篇,下一篇--> <#-- 热度 --> <#-- 网易新闻广告 --> <#--右边模块结构--> <#--评论模块结构--> <#--引用模块结构--> <#--博主发起的投票-->
 
 
 
 
 
 
 
 
 
 
 
 
 
 

页脚

网易公司版权所有 ©1997-2017