BZOJ-2588: Spoj 10628. Count on a tree(树上路径第K最值=LCA+可持久化线段树)

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

思路:

每个节点上建立一棵维护权值的可持久化线段树(维护从根到这个节点的权值),以他的父节点为历史版本建立,每次查询时直接在线段树上二分即可,所以只需要联立三棵可持久化线段树T[u],T[v],T[lca(u,v)]即可快捷查询。复杂度O(n log n)****

****代码:****

#include <cstdio>

#include <algorithm>

#include <cstring>

#include <queue>

  

using namespace std;

  

#define MAXN 100001

  

int a[MAXN];

int n,m;

  

struct edge {

    int t;

    bool flag;

    edge *next;

    edge (){

        next=NULL;

        flag=false;

    }

} *head[MAXN];

  

void INSERT(int s,int t){

    edge *p=new(edge);

    p->t=t;

    p->next=head[s];

    head[s]=p;

}

  

struct Node {

    int l,r,mint;

} RMQ[MAXN*6];

  

int DFS[MAXN*2],V=0;

int h[MAXN],first[MAXN];

bool f[MAXN];

  

void dfs(int v){

    f[v]=false;

    DFS[++V]=v;

    first[v]=V;

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

        if (f[p->t]){

            p->flag=true;

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

            dfs(p->t);

            DFS[++V]=v;

        }

    }

}

  

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

    RMQ[t].l=l;

    RMQ[t].r=r;

    if (l==r){

        RMQ[t].mint=DFS[l];

        return ;

    }

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

    Build_RMQ(l,mid,t<<1);

    Build_RMQ(mid+1,r,(t<<1)^1);

    if (h[RMQ[t<<1].mint]<h[RMQ[(t<<1)^1].mint]){

        RMQ[t].mint=RMQ[t<<1].mint;

    } else RMQ[t].mint=RMQ[(t<<1)^1].mint;

}

  

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

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

        return RMQ[t].mint;

    }

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

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

    if (l>mid) return query(l,r,(t<<1)^1);

    int L=query(l,mid,t<<1),R=query(mid+1,r,(t<<1)^1);

    if (h[L]<h[R]) return L; else return R;

}

  

int lca(int s,int t){

    return query(min(first[s],first[t]),max(first[s],first[t]),1);

}

  

struct saver {

    int v,t;

} b[MAXN];

  

int c[MAXN],N=0;

int r[MAXN];

  

bool cmp(saver x,saver y){

    return x.v<y.v;

}

  

struct node {

    int l,r,s;

    node *left,*right;

    node (){

        s=0;

        left=right=NULL;

    }

} *T[MAXN];

  

void Build0(int l,int r,node *t){

    t->l=l;

    t->r=r;

    if (l==r) return ;

    Build0(l,(l+r)>>1,t->left=new(node));

    Build0(((l+r)>>1)+1,r,t->right=new(node));

}

  

  

void Add(int x,node *u,node *t){

    t->l=u->l;

    t->r=u->r;

    t->s=u->s+1;

    if (t->l==t->r) return ;

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

    if (x<=mid) {

        t->right=u->right;

        t->left=new(node);

        Add(x,u->left,t->left);

    } else {

        t->left=u->left;

        t->right=new(node);

        Add(x,u->right,t->right);

    }

}

  

queue<int>Q;

  

void Build(){

    Build0(1,N,T[0]=new(node));

    INSERT(0,1);

    head[0]->flag=true;

    while (!Q.empty()) Q.pop();

    Q.push(0);

    while (!Q.empty()) {

        int v=Q.front();

        Q.pop();

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

            if (p->flag){

                Add(r[p->t],T[v],T[p->t]=new(node));

                Q.push(p->t);

            }

        }

    }

}

  

int main(){

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

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

        scanf("%d",&a[i]);

    }

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

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

        int s,t;

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

        INSERT(s,t);

        INSERT(t,s);

    }

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

    h[1]=0;

    dfs(1);

    Build_RMQ(1,V,1);

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

        b[i].v=a[i];

        b[i].t=i;

    }

    sort(b+1,b+n+1,cmp);

    c[++N]=b[1].v;

    r[b[1].t]=N;

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

        if (b[i].v!=b[i-1].v){

            c[++N]=b[i].v;

        }

        r[b[i].t]=N;

    }

    Build();

    int lastans=0;

    while (m--){

        int u,v,k;

        scanf("%d%d%d",&u,&v,&k);

        u^=lastans;

        int L=lca(u,v);

        node *s=T[u],*t=T[v],*l=T[L];

        while (s->l!=s->r) {

            int sum=s->left->s+t->left->s-2*l->left->s;

            if (a[L]<=c[(s->l+s->r)>>1]&&a[L]>=c[s->l]) sum++;

            if (k<=sum) {

                s=s->left;

                t=t->left;

                l=l->left;

            } else {

                k-=sum;

                s=s->right;

                t=t->right;

                l=l->right;

            }

        }

        lastans=c[s->l];

        printf("%d",lastans);

        if (m) printf("\n");

    }

    return 0;

}


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

推荐阅读更多精彩内容

  • 这个不错分享给大家,从扣上看到的,就转过来了 《电脑专业英语》 file [fail] n. 文件;v. 保存文...
    麦子先生R阅读 6,546评论 5 24
  • 《 如 果 》 如果能量浮化力量 就让个体力量聚焦成能量场 爱就是一场转化 造化着你我他 如果智商决定你的下限 ...
    谷主_7e37阅读 106评论 0 0
  • 在喜马拉雅电台上有一档节目,它的收听次数达到1.4亿,是由吴伯凡老师和梁冬老师共同支持的,这也是喜马拉雅上我最喜欢...
    燕落菩提阅读 1,341评论 2 2
  • 代之以根 树木有了圣贤 他们离开其晚餐桌 进入地下 跪下祈祷。 唯有他们的光环 留在了外面, 这些树, 这些花。 ...
    东丰林波阅读 249评论 0 1
  • 15和16年的流水账还躺在我的备忘录 那么就让我沿承这个习俗 继续和17年说拜拜吧 1月终于休了婚假去了日本 说实...
    蓝雪寒冰Yoyo阅读 163评论 0 0