bzoj1036&luogu2590 [ZJOI2008]树的统计

http://www.elijahqi.win/archives/748
题目描述

一棵树上有n个节点，编号分别为1到n，每个节点都有一个权值w。

我们将以下面的形式来要求你对这棵树完成一些操作：

I. CHANGE u t : 把结点u的权值改为t

II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值

III. QSUM u v: 询问从点u到点v的路径上的节点的权值和

注意：从点u到点v的路径上的节点包括u和v本身

输入输出格式

输入格式：

输入文件的第一行为一个整数n，表示节点的个数。

接下来n – 1行，每行2个整数a和b，表示节点a和节点b之间有一条边相连。

接下来一行n个整数，第i个整数wi表示节点i的权值。

接下来1行，为一个整数q，表示操作的总数。

接下来q行，每行一个操作，以“CHANGE u t”或者“QMAX u v”或者“QSUM u v”的形式给出。

输出格式：

对于每个“QMAX”或者“QSUM”的操作，每行输出一个整数表示要求输出的结果。

输入输出样例

输入样例#1：

4
1 2
2 3
4 1
4 2 1 3
12
QMAX 3 4
QMAX 3 3
QMAX 3 2
QMAX 2 3
QSUM 3 4
QSUM 2 1
CHANGE 1 5
QMAX 3 4
CHANGE 3 6
QMAX 3 4
QMAX 2 4
QSUM 3 4
输出样例#1：

4
1
2
2
10
6
5
6
5
16
说明

对于100％的数据，保证1<=n<=30000，0<=q<=200000；中途操作中保证每个节点的权值w在-30000到30000之间。

题目看起来非常裸，其实也确实是一个数据结构的裸题

可以用树链剖分

今天大致学习了下树链剖分，以前从没做过qwq

大概就是把儿子多的点分为重边（意味着从父亲要存儿子）剩下的为轻边（轻边为儿子存父亲）

这么一来我们要查询的时候不断往上走，相当于把树上的点全都映射到线段树的某一个区间，查询的时候要是在跳到重边了就线段树区间统计要是不是重边就采取暴力统计

#include<cstdio>
#include<cstring>
#define N 33000
inline int read(){
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9') {
        if(ch=='-') f=-1;ch=getchar();
    }
    while (ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int a[N];
inline void swap(int &x,int &y){
    x^=y;y^=x;x^=y;
}
struct node{
    int y,next;
}data[N<<1];
struct node1{
    int l,r,sum,max,left,right;
}tree[N<<2];
int size[N],id[N],son[N],fa[N],h[N],dep[N],dfn=0,tp[N],n,root=0,m;
inline int max(int x,int y){
    return x>y?x:y;
}
void dfs1(int x){
    size[x]=1;
    for (int i=h[x];i;i=data[i].next){
        int y=data[i].y;
        if (fa[x]==y) continue;
        dep[y]=dep[x]+1;fa[y]=x;dfs1(y);size[x]+=size[y];
        if (size[y]>size[son[x]]) son[x]=y;
    }
}
void dfs2(int x,int top){
    id[x]=++dfn;tp[x]=top;
    if (son[x]) dfs2(son[x],top);
    for (int i=h[x];i;i=data[i].next){
        int y=data[i].y;
        if (y==son[x]||y==fa[x]) continue;
        dfs2(y,y);
    }
}
int num=0;
inline void update(int x){
    int l=tree[x].left,r=tree[x].right;
    tree[x].sum=tree[l].sum+tree[r].sum;
    tree[x].max=max(tree[l].max,tree[r].max);
}
void build(int &x,int l,int r){
    tree[x=++num].l=l;tree[x].r=r;
    if(l==r) {//tree[x].sum=tree[x].max=a[tree[x].l];
    return ;}
    int mid=(l+r)>>1;
    build(tree[x].left,l,mid);build(tree[x].right,mid+1,r);
    update(x);
}
void print(int x){
    if (tree[x].left) print(tree[x].left);
    printf("%d %d %d %d\n",tree[x].l,tree[x].r,tree[x].sum,tree[x].max);
    if (tree[x].right) print(tree[x].right);
}
void print1(int x){
    if (son[x]) print1(son[x]); 
    printf("%d %d %d\n",x,id[x],dep[x]);
}
void change(int rt,int x,int val){
    if (tree[rt].l==tree[rt].r) {
        tree[rt].sum=tree[rt].max=val;return;
    }
    int mid=(tree[rt].l+tree[rt].r)>>1;
    if (x<=mid) change(tree[rt].left,x,val);else change(tree[rt].right,x,val);
    update(rt);
}
int qs(int x,int l,int r){
    int l1=tree[x].l,r1=tree[x].r;
    if (l<=l1&&r>=r1)return tree[x].sum;
    int mid=(l1+r1)>>1,tmp=0;
    if (l<=mid) tmp+=qs(tree[x].left,l,r);
    if (r>mid) tmp+=qs(tree[x].right,l,r);
    return tmp;
}
int qm(int x,int l,int r){
    int l1=tree[x].l,r1=tree[x].r;
    if (l<=l1&&r>=r1) return tree[x].max;
    int mid=l1+r1>>1,tmp=-N;
    if (l<=mid) tmp=max(tmp,qm(tree[x].left,l,r));
    if (r>mid) tmp=max(tmp,qm(tree[x].right,l,r));
    return tmp;
}
int main(){
    freopen("2590.in","r",stdin);
    n=read();int x,y;
    for (int i=1;i<n;++i){
        x=read();y=read();
        data[++num].y=y;data[num].next=h[x];h[x]=num;
        data[++num].y=x;data[num].next=h[y];h[y]=num;
    } 
    char str1[10];dep[1]=1;dfs1(1);
    dfs2(1,1);num=0;build(root,1,n);
    for (int i=1;i<=n;++i) a[i]=read(),change(root,id[i],a[i]);
    m=read();
    for (int i=1;i<=m;++i){
        scanf("%s",str1);x=read();y=read();
        if (str1[1]=='H')change(root,id[x],y),a[x]=y;
        if (str1[1]=='S'){
            int tmp=0;
            while (tp[x]!=tp[y]){
                if (dep[tp[x]]<dep[tp[y]]) swap(x,y);
                if(tp[x]==x) tmp+=a[x];else tmp+=qs(root,id[tp[x]],id[x]);
                x=fa[tp[x]]; 
            }
            if(id[x]>id[y]) swap(x,y);
            tmp+=qs(root,id[x],id[y]);
            printf("%d\n",tmp);
        }
        if (str1[1]=='M'){
            int tmp=-N;
            while (tp[x]!=tp[y]){
                if (dep[tp[x]]<dep[tp[y]]) swap(x,y);
                tmp=max(tmp,qm(root,id[tp[x]],id[x]));
                x=fa[tp[x]];
            }
            if (id[x]>id[y]) swap(x,y);
            tmp=max(tmp,qm(root,id[x],id[y]));
            printf("%d\n",tmp);
        }
    }
    //print(root);
    //print1(1);
    return 0;
}

LCT动态树做

#include<cstdio>
#define N 33000
long long w[N],max[N],sum[N];
bool rev[N];
int father[N],child[N][2],q[N],u[N],v[N];
int n,m;
inline int read(){
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9') {if (ch=='-') f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
inline long long max1(int x,int y){
    return x>y?x:y;
}
inline void swap(int &x,int &y){
    int t=x;x=y;y=t;
}
inline void pushdown(int x){
    int l=child[x][0],r=child[x][1];
    if (rev[x]){
        rev[x]^=1;rev[l]^=1;rev[r]^=1;
        swap(child[x][0],child[x][1]);
    }
}
void print(int x){
    pushdown(x);
    if (child[x][0]) print(child[x][0]);
    printf("%lld %lld %lld\n",w[x],sum[x],max[x]);
    if (child[x][1]) print(child[x][1]);
}
inline void updata(int x){
    int l=child[x][0],r=child[x][1];
    sum[x]=sum[l]+sum[r]+w[x];
    max[x]=max1(max[l],max[r]);
    max[x]=max1(max[x],w[x]);
}
inline bool isroot(int x){
    int fa=father[x];
    return child[fa][1]!=x&&child[fa][0]!=x;
}
void rotate(int x){
    int y=father[x],z=father[y];
    int l=(child[y][1]==x),r=l^1;
    if (!isroot(y)) child[z][child[z][1]==y]=x;
    father[child[x][r]]=y;father[y]=x;father[x]=z;
    child[y][l]=child[x][r];child[x][r]=y;
    updata(y);updata(x);
}
void splay(int x){
    int top=0;q[++top]=x;int i=x;
    while (!isroot(i)){
        q[++top]=father[i];i=father[i];
    }
    while (top) pushdown(q[top--]);
    while (!isroot(x)){
        int y=father[x],z=father[y];
        if (!isroot(y))
            if (child[y][1]==x^child[z][1]==y) rotate(x);else rotate(y);
        rotate(x);
    }
}
void access(int x){
    int y=0;
    while (x!=0){
        splay(x); 
        child[x][1]=y;
        updata(x);y=x;
        x=father[x];
    }
}
void makeroot(int x){
    access(x);//print(x);printf("*****\n");
    splay(x);rev[x]^=1;
}
void split(int x,int y){
    //print(x);
    makeroot(x);//print(x);
//    printf("123321\n");
    access(y);//print(3);
    splay(y);
    //print(y);
}
void link(int x,int y){
    makeroot(x);father[x]=y;
}
int main(){
    //freopen("2590.in","r",stdin);
    //freopen("2590.out","w",stdout);
    n=read();max[0]=-0x7fffffff;
    for (int i=1;i<n;++i) u[i]=read(),v[i]=read();
    for (int i=1;i<=n;++i) w[i]=read();
    //for (int i=1;i<=n;++i) printf("%lld ",w[i]);
    for (int i=1;i<n;++i) link(u[i],v[i]);
    //for (int i=1;i<=n;++i) printf("%d ",father[i]);
    //printf("\n");
    m=read();char ch[10];int x,y;
    while (m--){
        scanf("%s",ch);
        x=read(),y=read();
        if (ch[1]=='H'){
            splay(x);w[x]=y;updata(x);
        }
        if (ch[1]=='M'){
            split(x,y);printf("%lld\n",max[y]);
            //print(y);
        }
        if (ch[1]=='S'){
            split(x,y);printf("%lld\n",sum[y]);
        }
    }
    return 0;
}