bzoj 1036: [ZJOI2008]树的统计Count 树链剖分+线段树

题目大意：

给你一颗树，每个点有权值，要你求出一个点到另一个点路径的最大权值，或者是权值和，还要支持修改权值的问题。

思路：

树链剖分的模板题。树链剖分大概就是把书上的链给按一定方法拆开，然后每一段去一个hash值，放入线段树中去维护他的区间值。

这里有一个大佬的博客：http://blog.sina.com.cn/s/blog_7a1746820100wp67.html

程序：

#include<cstdio>#include<iostream>#include<cstring>#include<cstdlib> #define inf 0x7fffffff#define maxn 30005#define maxm 60005using namespace std;int pos[maxn],top[maxn],fa[maxn],size[maxn],son[maxn],dep[maxn],a[maxn],last[maxn];int n,cnt,sz;struct data{int w,next;}e[maxm];struct tree{int l,r,mx,sum;}t[100001];void add(int u,int v){    e[++cnt].next=last[u]; e[cnt].w=v; last[u]=cnt;    e[++cnt].next=last[v]; e[cnt].w=u; last[v]=cnt;}void init(){    int u,v;    scanf("%d",&n);    for (int i=1;i<=n-1;i++){        scanf("%d%d",&u,&v);        add(u,v);    }    for (int i=1;i<=n;i++){        scanf("%d",&a[i]);    }}void dfs1(int x){    size[x]=1;    for (int i=last[x];i;i=e[i].next){        if (fa[x]==e[i].w) continue;        fa[e[i].w]=x;        dep[e[i].w]=dep[x]+1;        dfs1(e[i].w);        size[x]+=size[e[i].w];    }}void dfs2(int x,int chain){    int k=0;    sz++;    pos[x]=sz;    top[x]=chain;    for (int i=last[x];i;i=e[i].next){        if (fa[x]!=e[i].w&&size[e[i].w]>size[k]) k=e[i].w;    }    if (k==0) return;     dfs2(k,chain);    for (int i=last[x];i;i=e[i].next){        if (e[i].w!=fa[x]&&e[i].w!=k) dfs2(e[i].w,e[i].w);    }}void build(int d,int l,int r){    t[d].l=l;    t[d].r=r;    if (l==r) return;    int mid=(l+r)/2;    build(d*2,l,mid);    build(d*2+1,mid+1,r);}void change(int d,int x,int y){    if (t[d].l==t[d].r) {        t[d].sum=t[d].mx=y;        return;    }    int mid=(t[d].l+t[d].r)/2;    if (x<=mid) change(d*2,x,y);        else change(d*2+1,x,y);    t[d].sum=t[d*2].sum+t[d*2+1].sum;    t[d].mx=max(t[d*2].mx,t[d*2+1].mx);}int querymax(int d,int x,int y){    if (t[d].l==x&&t[d].r==y) return t[d].mx;    int mid=(t[d].l+t[d].r)/2;    if (y<=mid) return querymax(d*2,x,y);    else if (x>mid) return querymax(d*2+1,x,y);    else return max(querymax(d*2,x,mid),querymax(d*2+1,mid+1,y));}int solvemax(int x,int y){    int mx=-inf;    while (top[x]!=top[y]){        if (dep[top[x]]<dep[top[y]]) swap(x,y);        mx=max(mx,querymax(1,pos[top[x]],pos[x]));        x=fa[top[x]];    }    if (dep[x]>dep[y]) swap(x,y);    mx=max(mx,querymax(1,pos[x],pos[y]));    return mx; }int querysum(int d,int x,int y){    if (t[d].l==x&&t[d].r==y) return t[d].sum;    int mid=(t[d].l+t[d].r)/2;    if (y<=mid) return querysum(d*2,x,y);    else if (x>mid) return querysum(d*2+1,x,y);    else return querysum(d*2,x,mid)+querysum(d*2+1,mid+1,y);}int solvesum(int x, int y){    int sum=0;    while (top[x]!=top[y]){        if (dep[top[x]]<dep[top[y]]) swap(x,y);        sum+=querysum(1,pos[top[x]],pos[x]);        x=fa[top[x]];    }    if (dep[x]>dep[y]) swap(x,y);    sum+=querysum(1,pos[x],pos[y]);    return sum;}int solve(){    build(1,1,n);    for (int i=1;i<=n;i++)      change(1,pos[i],a[i]);    int q;    scanf("%d",&q);    char ch[10]; int x,y;    for (int i=1;i<=q;i++){        scanf("%s%d%d",ch,&x,&y);        if (ch[0]=='C') change(1,pos[x],y);        else {            if (ch[1]=='M') printf("%d\n",solvemax(x,y));                else printf("%d\n",solvesum(x,y));        }    }}int main(){    init();    dfs1(1);    dfs2(1,1);    solve();    return 0;}