【BZOJ 1036】 树的统计count

题目

一棵树上有n个节点，编号分别为1到n，每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成
一些操作： I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 I
II. QSUM u v: 询问从点u到点v的路径上的节点的权值和 注意：从点u到点v的路径上的节点包括u和v本身

分析

树链剖分

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<vector>
using namespace std;
#define MN 30000
#define fINF -30000000
int fa[MN+],W[MN+],ans[MN+],dfn[MN+],sons[MN+],fl[MN+],head[MN+],ccnt=,cnt=;
struct data{int to,next;}e[MN*+];
void ins(int u,int v){
    e[++ccnt].to=v;e[ccnt].next=head[u];head[u]=ccnt;
    e[++ccnt].to=u;e[ccnt].next=head[v];head[v]=ccnt;
}
struct TREE{int val,max;}t[MN*+];
int n,q;
void update(int k,int l,int r,int q,int v){
    if(l==r) {t[k].val=t[k].max=v;return;}
    int mid=(l+r)/;
    if(q<=mid) update(k<<,l,mid,q,v);
    if(q>mid) update(k<<|,mid+,r,q,v);
    t[k].val=t[k<<].val+t[k<<|].val;
    t[k].max=max(t[k<<].max,t[k<<|].max);
}
void dfs1(int x){
    sons[x]=;
    for(int i=head[x];i;i=e[i].next){
        if(e[i].to==fa[x]) continue;
        fl[e[i].to]=fl[x]+; fa[e[i].to]=x;
        dfs1(e[i].to);
        sons[x]+=sons[e[i].to];
    }
}
void dfs2(int x,int chain){
    int k=;
    dfn[x]=++cnt;
    ans[x]=chain;
    for(int i=head[x];i;i=e[i].next)
        if(fl[e[i].to]>fl[x]&&sons[e[i].to]>sons[k]) k=e[i].to;
    if(k==) return;
    dfs2(k,chain);
    for(int i=head[x];i;i=e[i].next)
        if(fl[e[i].to]>fl[x]&&k!=e[i].to)
            dfs2(e[i].to,e[i].to);
}
int query_max(int k,int l,int r,int a,int b){
    if(a<=l&&r<=b) return t[k].max;
    int m=(l+r)/,anss=fINF;
    if(a<=m) anss=max(anss,query_max(k<<,l,m,a,b));
    if(m<b) anss=max(anss,query_max(k<<|,m+,r,a,b));
    return anss;
}
int query_sum(int k,int l,int r,int a,int b){
    if(a<=l&&r<=b) return t[k].val;
    int m=(l+r)/,anss=;
    if(a<=m) anss+=query_sum(k<<,l,m,a,b);
    if(m<b) anss+=query_sum(k<<|,m+,r,a,b);
    return anss;
}
int find_sum(int x,int y){
    int sum=;
    while(ans[x]!=ans[y]){
        if(fl[ans[x]]<fl[ans[y]]) swap(x,y);
        sum+=query_sum(,,n,dfn[ans[x]],dfn[x]);
        x=fa[ans[x]];
    }
    if(dfn[x]>dfn[y]) swap(x,y);
    sum+=query_sum(,,n,dfn[x],dfn[y]);
    return sum;
}
int find_max(int x,int y){
    int mx=fINF;
    while(ans[x]!=ans[y]){
        if(fl[ans[x]]<fl[ans[y]]) swap(x,y);
        mx=max(mx,query_max(,,n,dfn[ans[x]],dfn[x]));
        x=fa[ans[x]];
    }
    if(dfn[x]>dfn[y]) swap(x,y);
    mx=max(mx,query_max(,,n,dfn[x],dfn[y]));
    return mx;
}
void solve(int k,int a,int b){
    if(k==) printf("%d\n",find_max(a,b));
    if(k==) printf("%d\n",find_sum(a,b));
    if(k==) update(,,n,dfn[a],b);
}
int main(){
    int u,v,ro;
    scanf("%d",&n);
    for(int i=;i<n;i++) scanf("%d%d",&u,&v),ins(u,v);
    dfs1(); dfs2(,);
    for(int i=;i<=n;i++) scanf("%d",&W[i]),update(,,n,dfn[i],W[i]);
    scanf("%d",&q);
    while(q--){
        char ch=getchar(); int k;
        int x1=,f1=,x2=,f2=;
        while(ch<''||ch>''){
            if(ch=='X') k=; if(ch=='U') k=;
            if(ch=='H') k=; if(ch=='-') f1=-;
            ch=getchar();
        }
        while(ch>=''&&ch<='') x1=x1*+ch-'',ch=getchar();
        while(ch<''||ch>'') f2=ch=='-'?-:,ch=getchar();
        while(ch>=''&&ch<='') x2=x2*+ch-'',ch=getchar();
        solve(k,x1*f1,x2*f2);
    }
    return ;
}

————————————————————————————————————

来自PaperCloud的博客，未经允许，请勿转载，谢谢。