BZOJ 3083 遥远的国度 树链剖分

题目大意:给定一棵有根树,有以下操作:换根,链上修改,子树查询最小值。

树链剖分配合线段树可以轻松解决子树查询与链上修改,换根怎么办?

当然不能真的换根。设初始根为 root (也就是说树剖处理时是以 root 为根的),当前根为 cur ,对于每一次查询子树x进行讨论:
1.如果 x 就是 cur ,取所有点最小即可。
2.如果 x 的子树包括 cur ,则对包括 cur 的子树取补集查询最小值。
3.不满足以上两种情况(即 x 在 cur 的子树中 或 cur的子树中没有 x 且 x 的子树中没有 cur ),那么 x 的子树没有变化,正常查询即可。

还有一些细节,若查询一段空区间(l>r) 应该返回一个不影响答案的值否则RE

#include <cstdio>
#include <cstring>
#define N 100005
#define INF 2147483647
using namespace std;
inline int Min(int x,int y) {return x<y?x:y;}
inline void Swap(int& x,int& y) {int z=x;x=y;y=z;}
struct Edge {
    int to,nex;
}e[N<<1];
int n,m,root,fir[N];
int T,val[N],dpt[N],siz[N],son[N],top[N],pa[N],seq[N],pos[N];
void dfs1(int x) {
    dpt[x]=dpt[pa[x]]+1;
    siz[x]=1;
    for(int i=fir[x];i!=-1;i=e[i].nex) {
        if(e[i].to==pa[x]) continue;
        pa[e[i].to]=x;
        dfs1(e[i].to);
        siz[x]+=siz[e[i].to];
        if(siz[e[i].to]>siz[son[x]]) son[x]=e[i].to;
    }
    return ;
}
void dfs2(int x) {
    seq[++T]=x; pos[x]=T;
    if(son[pa[x]]==x) top[x]=top[pa[x]];
    else top[x]=x;
    if(son[x]) dfs2(son[x]);
    for(int i=fir[x];i!=-1;i=e[i].nex) {
        if(e[i].to==pa[x] || e[i].to==son[x]) continue;
        dfs2(e[i].to);
    }
    return ;
}
struct Segment_Tree {
    int l,r,minn,mark,flag;
}s[N*4];
void pushdown(int x) {
    if(s[x].flag==-1) return ;
    s[x].minn=s[x].mark;
    if(s[x].l!=s[x].r && s[x].flag>s[x*2].flag) {
        s[x*2].flag=s[x].flag;
        s[x*2].mark=s[x].mark;
    }
    if(s[x].l!=s[x].r && s[x].flag>s[x*2+1].flag) {
        s[x*2+1].flag=s[x].flag;
        s[x*2+1].mark=s[x].mark;
    }
    s[x].flag=-1;
    return ;
}
void maintain(int x) {
    pushdown(x); pushdown(x*2); pushdown(x*2+1);
    s[x].minn=Min(s[x*2].minn,s[x*2+1].minn);
    return ;
}
void init(int x,int l,int r) {
    s[x].l=l; s[x].r=r; s[x].flag=-1;
    if(l==r) { s[x].minn=val[seq[l]]; return ; }
    int mid=l+r>>1;
    init(x*2,l,mid); init(x*2+1,mid+1,r);
    maintain(x);
    return ;
}
void change(int x,int l,int r,int v) {
    pushdown(x);
    if(s[x].l==l && s[x].r==r) {
        s[x].flag=T;
        s[x].mark=v;
        return ;
    }
    int mid=s[x].l+s[x].r>>1;
    if(r<=mid) change(x*2,l,r,v);
    else if(l>mid) change(x*2+1,l,r,v);
    else change(x*2,l,mid,v) , change(x*2+1,mid+1,r,v);
    maintain(x);
    return ;
}
int Get_Min(int x,int l,int r) {
    if(l>r) return INF;
    pushdown(x);
    if(s[x].l==l && s[x].r==r) return s[x].minn;
    int mid=s[x].l+s[x].r>>1;
    if(r<=mid) return Get_Min(x*2,l,r);
    if(l>mid) return Get_Min(x*2+1,l,r);
    return Min(Get_Min(x*2,l,mid),Get_Min(x*2+1,mid+1,r));
}
void To_change(int x,int y,int v) {
    T++;
    while(top[x]!=top[y]) {
        if(dpt[top[x]]<dpt[top[y]]) Swap(x,y);
        change(1,pos[top[x]],pos[x],v);
        x=pa[top[x]];
    }
    if(dpt[x]<dpt[y]) Swap(x,y);
    change(1,pos[y],pos[x],v);
    return ;
}
bool In_Tree(int x,int y) { //yx的子树中返回true
    if(pos[x]<=pos[y] && pos[y]<=pos[x]+siz[x]-1) return true;
    return false;
}
int Query_Min(int x) {
    if(x==root) return Get_Min(1,1,n);
    if(In_Tree(root,x) || !In_Tree(x,root)) return Get_Min(1,pos[x],pos[x]+siz[x]-1);
    int rootl,rootr;
    for(int i=fir[x];i!=-1;i=e[i].nex)
        if(e[i].to!=pa[e[i].to] && In_Tree(e[i].to,root)) {
            rootl=pos[e[i].to];
            rootr=pos[e[i].to]+siz[e[i].to]-1;
            break;
        }
    return Min(Get_Min(1,1,rootl-1),Get_Min(1,rootr+1,n));
}
int main() {
    memset(fir,-1,sizeof fir);
    scanf("%d%d",&n,&m);
    for(int i=0;i<n-1;i++) {
        int x,y,u;
        scanf("%d%d",&x,&y);
        u=i<<1; e[u].to=y; e[u].nex=fir[x]; fir[x]=u;
        u^=1; e[u].to=x; e[u].nex=fir[y]; fir[y]=u;
    }
    for(int i=1;i<=n;i++) scanf("%d",&val[i]);
    scanf("%d",&root);
    dfs1(1); dfs2(1);
    init(1,1,n);
    T=0;
    while(m--) {
        int mode;
        scanf("%d",&mode);
        if(mode==1) scanf("%d",&root);
        if(mode==2) {
            int x,y,v;
            scanf("%d%d%d",&x,&y,&v);
            To_change(x,y,v);
        }
        if(mode==3) {
            int x;
            scanf("%d",&x);
            printf("%d\n",Query_Min(x));
        }
    }
    return 0;
}
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值