BZOJ-3083: 遥远的国度(树链剖分+线段树)

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

树链剖分弄出来的那个序列也是DFS序列,然后就SGT维护一下,对于换根分类讨论一下就可以了。(突然想通sone1怎么写了。。。好开心~)

代码:

#include <cstdio>

#include <algorithm>

#include <cstring>

 

using namespace std ;

 

#define AddEdge( s , t ) Add( s , t ) , Add( t , s )

#define MAXN 100010

#define MAXB 20

#define left( t ) ( t << 1 )

#define right( t ) ( left( t ) ^ 1 )

#define ll long long

 

const ll inf = ( ll )( 0x7fffffff ) * ( ll )( 0x7fffffff ) ;

 

struct edge {

    edge *next ;

    int t ;

} *head[ MAXN ] ;

 

void Add( int s , int t ) {

    edge *p = new( edge ) ;

    p -> t = t , p -> next = head[ s ] ;

    head[ s ] = p ;

}

 

int n , m , roof , up[ MAXN ][ MAXB + 1 ] , B , h[ MAXN ] , beg[ MAXN ] ;

ll w[ MAXN ] ;

bool f[ MAXN ] ;

int arr[ MAXN ] , in[ MAXN ] , out[ MAXN ] , cnt = 0 , size[ MAXN ] , child[ MAXN ] ;

 

void dfs0( int v ) {

    f[ v ] = false , size[ v ] = 1 , child[ v ] = 0 ;

    int ret = 0 ;

    for ( edge *p = head[ v ] ; p ; p = p -> next ) if ( f[ p -> t ] ) {

        h[ p -> t ] = h[ v ] + 1 , up[ p -> t ][ 0 ] = v ;

        dfs0( p -> t ) ;

        if ( size[ p -> t ] > ret ) {

            ret = size[ p -> t ] , child[ v ] = p -> t ;

        }

        size[ v ] += size[ p -> t ] ;

    }

}

 

void dfs1( int v , int u ) {

    f[ v ] = false , arr[ in[ v ] = ++ cnt ] = v , beg[ v ] = u ;

    if ( child[ v ] ) {

        dfs1( child[ v ] , u ) ;

        for ( edge *p = head[ v ] ; p ; p = p -> next ) if ( f[ p -> t ] ) {

            dfs1( p -> t , p -> t ) ;

        }

    }

    out[ v ] = cnt ;

}

 

struct node {

    int l , r ;

    ll Min , flag ;

    node(  ) {

        flag = 0 ;

    }

} sgt[ MAXN << 3 ] ;

 

void pushdown( int t ) {

    if ( sgt[ t ].flag ) {

        sgt[ t ].Min = sgt[ t ].flag ;

        if ( sgt[ t ].l < sgt[ t ].r ) {

            sgt[ left( t ) ].flag = sgt[ right( t ) ].flag = sgt[ t ].flag ;

        }

        sgt[ t ].flag = 0 ;

    }

}

 

void update( int t ) {

    pushdown( t ) ;

    if ( sgt[ t ].l < sgt[ t ].r ) {

        pushdown( left( t ) ) , pushdown( right( t ) ) ;

        sgt[ t ].Min = min( sgt[ left( t ) ].Min , sgt[ right( t ) ].Min ) ;

    }

}

 

void build( int l , int r , int t ) {

    sgt[ t ].l = l , sgt[ t ].r = r ;

    if ( l == r ) {

        sgt[ t ].Min = w[ arr[ l ] ] ; return ;

    }

    int mid = ( l + r ) >> 1 ;

    build( l , mid , left( t ) ) , build( mid + 1 , r , right( t ) ) ;

    update( t ) ;

}

 

void change( int l , int r , ll v , int t ) {

    pushdown( t ) ;

    if ( sgt[ t ].l == l && sgt[ t ].r == r ) {

        sgt[ t ].flag = v ; pushdown( t ) ;

        return ;

    }

    int mid = ( sgt[ t ].l + sgt[ t ].r ) >> 1 ;

    if ( r <= mid ) change( l , r , v , left( t ) ) ; else

    if ( l > mid ) change( l , r , v , right( t ) ) ; else {

        change( l , mid , v , left( t ) ) , change( mid + 1 , r , v , right( t ) ) ;

    }

    update( t ) ;

}

 

ll query( int l , int r , int t ) {

    pushdown( t ) ;

    if ( l == sgt[ t ].l && r == sgt[ t ].r ) return sgt[ t ].Min ;

    int mid = ( sgt[ t ].l + sgt[ t ].r ) >> 1 ;

    if ( r <= mid ) return query( l , r , left( t ) ) ;

    if ( l > mid ) return query( l , r , right( t ) ) ;

    return min( query( l , mid , left( t ) ) , query( mid + 1 , r , right( t ) ) ) ;

}

 

int Lca( int u , int v ) {

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

    for ( int i = MAXB ; i >= 0 ; -- i ) {

        if ( h[ up[ u ][ i ] ] >= h[ v ] ) {

            u = up[ u ][ i ] ;

        }

    }

    if ( u == v ) return u ;

    for ( int i = MAXB ; i >= 0 ; -- i ) {

        if ( up[ u ][ i ] != up[ v ][ i ] ) {

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

        }

    }

    return up[ v ][ 0 ] ;

}

 

void Change( int v , int u , ll c ) {

    while ( h[ v ] >= h[ u ] ) {

        if ( h[ beg[ v ] ] > h[ u ] ) {

            change( in[ beg[ v ] ] , in[ v ] , c , 1 ) ;

            v = up[ beg[ v ] ][ 0 ] ;

        } else {

            change( in[ u ] , in[ v ] , c , 1 ) ;

            break ;

        }

    }

}

 

ll Solve( int v ) {

    if ( v == roof ) return query( 1 , n , 1 ) ;

    int lca = Lca( v , roof ) ;

    if ( lca == v ) {

        ll ret = inf ;

        int t = roof ;

        for ( int i = MAXB ; i >= 0 ; -- i ) if ( h[ up[ t ][ i ] ] > h[ v ] ) {

            t = up[ t ][ i ] ;

        }

        if ( in[ t ] - 1 ) ret = min( ret , query( 1 , in[ t ] - 1 , 1 ) ) ;

        if ( out[ t ] < n ) ret = min( ret , query( out[ t ] + 1 , n , 1 ) ) ;

        return ret ;

    } else return query( in[ v ] , out[ v ] , 1 ) ;

}

 

int main(  ) {

    memset( head , 0 , sizeof( head ) ) ;

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

    for ( int i = 1 ; i < n ; ++ i ) {

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

        AddEdge( s , t ) ;

    }

    for ( int i = 0 ; i ++ < n ; ) scanf( "%lld" , w + i ) ;

    scanf( "%d" , &roof ) ;

    memset( f , true , sizeof( f ) ) ;

    h[ 1 ] = 1 ;

    dfs0( 1 ) ;

    memset( f , true , sizeof( f ) ) ;

    dfs1( 1 , 1 ) ;

    for ( int i = 0 ; i ++ < MAXB ; ) {

        for ( int j = 0 ; j ++ < n ; ) {

            up[ j ][ i ] = up[ up[ j ][ i - 1 ] ][ i - 1 ] ;

        }

    }

    build( 1 , n , 1 ) ;

    while ( m -- ) {

        int x , v , u , lca ;

        ll c ;

        scanf( "%d" , &x ) ;

        switch ( x ) {

            case 1 :

                scanf( "%d" , &v ) ;

                roof = v ;

                break ;

            case 2 :

                scanf( "%d%d%lld" , &u , &v , &c ) ;

                lca = Lca( u , v ) ;

                Change( u , lca , c ) , Change( v , lca , c ) ;

                break ;

            case 3 :

                scanf( "%d" , &v ) ;

                printf( "%lld\n" , Solve( v ) ) ;

                break ;

        }

    }

    return 0 ;

}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,258评论 6 498
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,335评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 162,225评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,126评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,140评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,098评论 1 295
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,018评论 3 417
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,857评论 0 273
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,298评论 1 310
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,518评论 2 332
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,678评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,400评论 5 343
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,993评论 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,638评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,801评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,661评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,558评论 2 352

推荐阅读更多精彩内容