BZOJ1036--树链剖分模板

来源：互联网发布：网络玄幻小说合集下载编辑：程序博客网时间：2024/06/05 00:55

Description
　　一棵树上有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本身

代码如下：

#include<cstdio>#include<cstring>#include<algorithm>#define maxn 30006using namespace std;struct data{    int l,r,max,sum,key;}tre[maxn*4];int n,tot,tem,tet,lnk[maxn],nxt[maxn*2],son[maxn*2],dep[maxn],H_son[maxn],top[maxn],siz[maxn],fa[maxn],id[maxn];bool vis[maxn];int _read(){    int sum=0,p;char ch=getchar();    while((!(ch>='0'&&ch<='9'))&&(ch!='-'))ch=getchar();    if(ch=='-')p=-1,ch=getchar();else p=1;    while(ch>='0'&&ch<='9')sum=sum*10+ch-48,ch=getchar();    return sum*p;}void add(int x,int y){    nxt[++tot]=lnk[x];son[tot]=y;lnk[x]=tot;}void build(int p,int l,int r){    int mid=(l+r)>>1;    tre[p].l=l;tre[p].r=r;tre[p].max=-1e9;tre[p].key=tre[p].sum=0;    if(l==r)return;    build(p<<1,l,mid);build(p<<1|1,mid+1,r);}void update(int p,int pos,int k){    if(pos>tre[p].r||pos<tre[p].l)return;    if(tre[p].l==tre[p].r){        tre[p].key=tre[p].max=tre[p].sum=k;        return;    }    update(p<<1,pos,k);update(p<<1|1,pos,k);    tre[p].max=max(tre[p<<1].max,tre[p<<1|1].max);    tre[p].sum=tre[p<<1].sum+tre[p<<1|1].sum;}int query_max(int p,int l,int r){    if(l>tre[p].r||r<tre[p].l)return -1e9;    if(l<=tre[p].l&&r>=tre[p].r)return tre[p].max;    return max(query_max(p<<1,l,r),query_max(p<<1|1,l,r));}int query_sum(int p,int l,int r){    if(l>tre[p].r||r<tre[p].l)return 0;    if(l<=tre[p].l&&r>=tre[p].r)return tre[p].sum;    return query_sum(p<<1,l,r)+query_sum(p<<1|1,l,r);}void dfs1(int x){    vis[x]=0;siz[x]=1;    for(int j=lnk[x];j;j=nxt[j]) if(vis[son[j]]){        fa[son[j]]=x;dep[son[j]]=dep[x]+1;        dfs1(son[j]);        if(siz[son[j]]>siz[H_son[x]])H_son[x]=son[j];        siz[x]+=siz[son[j]];    }}void dfs2(int x,int lst){    vis[x]=0;top[x]=lst;id[x]=++tem;    if(H_son[x])dfs2(H_son[x],lst);    for(int j=lnk[x];j;j=nxt[j]) if(vis[son[j]]&&(son[j]!=H_son[x]))dfs2(son[j],son[j]);}int get_max(int x,int y){    int Max=-1e9;    for(;top[x]!=top[y];x=fa[top[x]]){        if(dep[top[x]]<dep[top[y]])swap(x,y);        Max=max(Max,query_max(1,id[top[x]],id[x]));    }    if(dep[x]<dep[y])swap(x,y);    return max(Max,query_max(1,id[y],id[x]));}int get_sum(int x,int y){    int sum=0;    for(;top[x]!=top[y];x=fa[top[x]]){        if(dep[top[x]]<dep[top[y]])swap(x,y);        sum+=query_sum(1,id[top[x]],id[x]);    }    if(id[x]>id[y])swap(x,y);    return sum+query_sum(1,id[x],id[y]);}int main(){    freopen("count.in","r",stdin);    freopen("count.out","w",stdout);    n=_read();    for(int i=1,x,y;i<n;i++)x=_read(),y=_read(),add(x,y),add(y,x);    memset(vis,1,sizeof(vis));    dfs1(1);    memset(vis,1,sizeof(vis));    dfs2(1,1);    build(1,1,tem);    for(int i=1,x;i<=n;i++)x=_read(),update(1,id[i],x);    tet=_read();    while(tet--){        char s[30];        scanf("%s",s);int x=_read(),y=_read();        if(s[0]=='C')update(1,id[x],y);else        if(s[1]=='M')printf("%d\n",get_max(x,y));else        if(s[1]=='S')printf("%d\n",get_sum(x,y));    }    return 0;}

0 0