码迷,mamicode.com
首页 > 其他好文 > 详细

【GDOI2017模拟12.9】最近公共祖先

时间:2018-05-21 21:55:48      阅读:126      评论:0      收藏:0      [点我收藏+]

标签:就会   模拟   ace   name   turn   src   odi   The   using   

题目

技术分享图片

分析

首先,将这些节点按dfs序建一棵线段树。
因为按dfs序,所以在同一子树上的节点会放在线段树相邻的位置。
发现,对于一个位置x,它的权值只会对以x为根的子树造成影响。
当修改x时,用w[x]更新子树x的最大值,
接着从x向上跳,用w[fa[x]]更新子树fa[x]-子树x最大值,
因为当用w[fa[x]]来更新过子树fa[x]-子树x时,再用它更新就会没有意义,所以打个标记,不再更新。这样就最多只会更新n次。

#include <cmath>
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
const int N=100005;
using namespace std;
int d[N],aft[N],fa[N],mx[N*10],son[N],size[N],n,m,last[N*2],next[N*2],to[N*2],tot,v1[N],ans,end[N],lazy[N*10];
int bz[N];
int bj(int x,int y)
{
    next[++tot]=last[x];
    last[x]=tot;
    to[tot]=y;
}
int down(int v)
{
    if(!lazy[v]) return 0;
    lazy[v*2]=max(lazy[v],lazy[v*2]);
    lazy[v*2+1]=max(lazy[v],lazy[v*2+1]);
    mx[v*2]=max(mx[v*2],lazy[v]);
    mx[v*2+1]=max(mx[v*2+1],lazy[v]);
}
int dg(int x)
{
    d[++tot]=x;
    aft[x]=tot;
    for(int i=last[x];i;i=next[i])
    {
        int j=to[i];
        if(j!=fa[x])
        {
            fa[j]=x;
            dg(j);
            end[j]=tot;
        }
    }
}
int change(int v,int l,int r,int x,int y,int z)
{
    if(x>y && x && y) return 0;
    if(l==x && y==r)
    {
        mx[v]=max(mx[v],z);
        lazy[v]=max(lazy[v],z);
        return 0;
    }
    down(v);
    int mid=(l+r)/2;
    if(y<=mid) change(v*2,l,mid,x,y,z);
    else 
    if(x>mid) change(v*2+1,mid+1,r,x,y,z);
    else change(v*2,l,mid,x,mid,z),change(v*2+1,mid+1,r,mid+1,y,z);
    mx[v]=max(mx[v*2],mx[v*2+1]);
}
int find(int v,int l,int r,int x)
{
    if(l==r) return mx[v];
    down(v);
    int mid=(l+r)/2,j;
    if(x<=mid) j=find(v*2,l,mid,x);
    else j=find(v*2+1,mid+1,r,x);
    mx[v]=max(mx[v*2],mx[v*2+1]);
    return j;
}
int up(int x)
{
    if(!fa[x]) return 0;
    change(1,1,tot,max(aft[fa[x]],1),aft[x]-1,v1[fa[x]]);
    change(1,1,tot,end[x]+1,end[fa[x]],v1[fa[x]]);
    if(!bz[x]) up(fa[x]);
    bz[x]=true;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&v1[i]);
    for(int i=1;i<=n-1;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        bj(x,y);
        bj(y,x);
    }
    tot=0;
    dg(1);
    end[1]=tot;
    for(int i=1;i<=m;i++)
    {
        char c=getchar();
        int x;
        while(c!='Q' && c!='M') c=getchar();
        if(c=='M')
        {
            scanf("odify %d",&x);
            change(1,1,tot,aft[x],end[x],v1[x]);
            up(x);
        }
        else
        {
            scanf("uery %d",&x);
            ans=find(1,1,tot,aft[x]);
            printf("%d\n",(ans)?ans:-1);
        }
    }
}

【GDOI2017模拟12.9】最近公共祖先

标签:就会   模拟   ace   name   turn   src   odi   The   using   

原文地址:https://www.cnblogs.com/chen1352/p/9069420.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!