BZOJ-2599: [IOI2011]Race(树-点分治+SBT)

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

裸的点分治的题,看到k这么小其实还有其他方法,我偷懒直接就SBT O(n log^2 n)水过了。

代码:

#include <cstdio>

#include <algorithm>

#include <cstring>

 

using namespace std ;

 

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

#define MAXN 200100

#define L( t ) left[ t ]

#define R( t ) right[ t ]

#define K( t ) key[ t ]

#define S( t ) size[ t ]

#define inf 0x7fffffff

 

struct node {

    int k , v ;

    bool operator < ( const node &a ) const {

        return k < a.k ;

    }

    bool operator == ( const node &a ) const {

        return k == a.k ;

    }

    bool operator > ( const node &a ) const {

        return k > a.k ;

    }

};

 

node make( int _k , int _v ) {

    node u ;

    u.k = _k , u.v = _v ;

    return u ;

}

 

struct SBT {

    int left[ MAXN ] , right[ MAXN ] , size[ MAXN ] , V , roof ;

    node key[ MAXN ] ;

    void Init(  ) {

        V = roof = 0 ;

        L( 0 ) = R( 0 ) = S( 0 ) = 0 ;

    }

    void update( int t ) {

        S( t ) = S( L( t ) ) + S( R( t ) ) + 1 ;

    }

    void Left( int &t ) {

        int k = R( t ) ;

        R( t ) = L( k ) ; update( t ) ;

        L( k ) = t ; update( k ) ;

        t = k ;

    }

    void Right( int &t ) {

        int k = L( t ) ;

        L( t ) = R( k ) ; update( t ) ;

        R( k ) = t ; update( k ) ;

        t = k ;

    }

    void maintain( int &t ) {

        if ( S( L( L( t ) ) ) > S( R( t ) ) ) {

            Right( t ) ;

            maintain( R( t ) ) ; maintain( t ) ;

            return ;

        }

        if ( S( R( L( t ) ) ) > S( R( t ) ) ) {

            Left( L( t ) ) ; Right( t ) ;

            maintain( L( t ) ) , maintain( R( t ) ) ; maintain( t ) ;

            return ;

        }

        if ( S( R( R( t ) ) ) > S( L( t ) ) ) {

            Left( t ) ;

            maintain( L( t ) ) ; maintain( t ) ;

            return ;

        }

        if ( S( L( R( t ) ) ) > S( L( t ) ) ) {

            Right( R( t ) ) ; Left( t ) ;

            maintain( L( t ) ) , maintain( R( t ) ) ; maintain( t ) ;

            return ;

        }

    }

    int search( node k , int t ) {

        if ( ! t ) return 0 ;

        if ( k == K( t ) ) return t ;

        return search( k , k < K( t ) ? L( t ) : R( t ) ) ;

    }

    void Insert( node k , int &t ) {

        if ( ! t ) {

            t = ++ V ;

            L( t ) = R( t ) = 0 , S( t ) = 1 , K( t ) = k ;

            return ;

        }

        Insert( k , k < K( t ) ? L( t ) : R( t ) ) ;

        update( t ) ; maintain( t ) ;

    }

    int query( int k ) {

        int t = search( make( k , 0 ) , roof ) ;

        if ( ! t ) return inf ;

        return K( t ).v ;

    }

    void Push( node k ) {

        int t = search( k , roof ) ;

        if ( ! t ) Insert( k , roof ) ; else {

            if ( k.v < K( t ).v ) K( t ).v = k.v ;

        }

    }

} sbt ;

 

struct edge {

    edge *next ;

    int t , d ;

} *head[ MAXN ] ;

 

void Add( int s , int t , int d ) {

    edge *p = new( edge ) ;

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

    head[ s ] = p ;

}

 

int n , len , ans = inf ;

 

int size[ MAXN ] , h[ MAXN ] , dep[ MAXN ] , rt , roof , b[ MAXN ] , bn ;

bool f[ MAXN ] ;

 

void dfs0( int v , int u ) {

    size[ v ] = 1 ;

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

        dfs0( p -> t , v ) ;

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

    }

}

 

void dfs1( int v , int u ) {

    if ( roof ) return ;

    bool flag = true ;

    if ( size[ rt ] - size[ v ] > size[ rt ] / 2 ) flag = false ;

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

        dfs1( p -> t , v ) ;

        if ( size[ p -> t ] > size[ rt ] / 2 ) flag = false ;

    }

    if ( flag ) roof = v ;

}

 

void dfs2( int v , int u ) {

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

        h[ p -> t ] = h[ v ] + 1 , dep[ p -> t ] = dep[ v ] + p -> d ;

        dfs2( p -> t , v ) ;

    }

}

 

void dfs3( int v , int u ) {

    b[ ++ bn ] = v ;

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

        dfs3( p -> t , v ) ;

    }

}

 

void Solve( int v ) {

    dfs0( v , 0 ) ;

    roof = 0 , rt = v ;

    dfs1( v , 0 ) ;

    h[ roof ] = dep[ roof ] = 0 ;

    dfs2( roof , 0 ) ;

    sbt.Init(  ) ;

    sbt.Push( make( 0 , 0 ) ) ;

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

        bn = 0 ;

        dfs3( p -> t , roof ) ;

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

            int temp = sbt.query( len - dep[ b[ i ] ] ) ;

            if ( temp < inf ) ans = min( ans , h[ b[ i ] ] + temp ) ;

        }

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

            sbt.Push( make( dep[ b[ i ] ] , h[ b[ i ] ] ) ) ;

        }

    }

    f[ roof ] = false ;

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

}

 

int main(  ) {

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

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

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

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

        AddEdge( s + 1 , t + 1 , d ) ;

    }

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

    Solve( 1 ) ;

    printf( "%d\n" , ans < inf ? ans : - 1 ) ;

    return 0 ;

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

推荐阅读更多精彩内容

  • 题目:http://www.lydsy.com/JudgeOnline/problem.php?id=1468 题...
    AmadeusChan阅读 408评论 0 0
  • 这个不错分享给大家,从扣上看到的,就转过来了 《电脑专业英语》 file [fail] n. 文件;v. 保存文...
    麦子先生R阅读 6,545评论 5 24
  • 时间过得那么快,三年的高中生涯就仅仅只剩196天,我感到惶恐,并不那么优秀的我就快要接受高考的洗礼。 看着自己惨淡...
    喵唔呼阅读 252评论 0 1
  • 注:原文来自网络,有大面积删改。 第一场 背景:落叶枯黄,九月的浅秋终于来临.故事就发生在这美丽的季节.仁心医院 ...
    光影尘迹阅读 2,994评论 3 17
  • 一直以来,都以为自己踏实把事情办好,会不会说话,其实并没有那么重要。后来的自己渐渐发现,这样的想法是不是显得太过天...
    纠缠的离骚阅读 1,028评论 17 12