【题解】「ZJOI2008」树的统计(树链剖分)

题面

【题目描述】
一棵树上有 n n n个节点,编号分别为 1 1 1 n n n,每个节点都有一个权值 w w w。我们将以下面的形式来要求你对这棵树完成一些操作:
I . I. I. C H A N G E CHANGE CHANGE u u u t t t : 把结点 u u u的权值改为 t t t
I I . II. II. Q M A X QMAX QMAX u u u v v v: 询问从点 u u u到点 v v v的路径上的节点的最大权值
I I I . III. III. Q S U M QSUM QSUM u u u v v v: 询问从点 u u u到点 v v v的路径上的节点的权值和 ,注意:从点 u u u到点 v v v的路径上的节点包括 u u u v v v本身
【输入】
输入的第一行为一个整数 n n n,表示节点的个数。
接下来 n – 1 n – 1 n1行,每行 2 2 2个整数 a a a b b b,表示节点 a a a和节点 b b b之间有一条边相连。
接下来 n n n行,每行一个整数,第 i i i行的整数 w i w_i wi表示节点 i i i的权值。
接下来 1 1 1行,为一个整数 q q q,表示操作的总数。
接下来 q q q行,每行一个操作,以“ C H A N G E CHANGE CHANGE u u u t t t ”或者“ Q M A X QMAX QMAX u u u v v v”或者“ Q S U M QSUM QSUM u u u v v v”的形式给出。
对于 100 % 100% 100的数据,保证 1 < = n < = 30000 1<=n<=30000 1<=n<=30000 0 < = q < = 200000 0<=q<=200000 0<=q<=200000;中途操作中保证每个节点的权值 w w w − 30000 -30000 30000 30000 30000 30000之间。
【输出】
对于每个“ Q M A X QMAX QMAX”或者“ Q S U M QSUM QSUM”的操作,每行输出一个整数表示要求输出的结果。
【样例输入】

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

【样例输出】

4
1
2
2
10
6
5
6
5
16

算法分析

树链剖分模板题。

参考程序

#include<bits/stdc++.h>
#define N 200100 
using namespace std;
int head[N],nxt[N],to[N],w[N],tot;
int n,m;
int father[N],deep[N],size[N];//父节点,深度,子树结点数 
int son[N],top[N];//重儿子,所在重路径的顶部结点(深度最小的结点)
int id[N],rev[N],t;//在线段树中的下标(dfs序),线段树中下标的结点,即rev[id[u]]=u ,dfs序号 
int sum[2*N],Max[2*N]; //线段树 
int ans_sum,ans_max;
void Add(int u,int v)
{
    nxt[++tot]=head[u];
    to[tot]=v;
    head[u]=tot;
}
void dfs1(int u,int dad)
{
    size[u]=1;  //本身结点数为1 
    father[u]=dad;   
    deep[u]=deep[dad]+1;
    for(int i=head[u];i!=-1;i=nxt[i])
    {
        int v=to[i];
        if(v!=dad)
        {
            dfs1(v,u);
            size[u]+=size[v];
            if(size[v]>size[son[u]]) son[u]=v;//找重儿子 
        }
    }
}
void dfs2(int u,int dad)
{
    int v=son[u];
    if(v)  						//优先选择重儿子 
    {
        id[v]=++t;				//dfs序列 
        top[v]=top[u];
        rev[t]=v;
        dfs2(v,u); 
    }
    for(int i=head[u];i!=-1;i=nxt[i])
    {
        v=to[i];
        if(!top[v])
        {
            id[v]=++t;
            top[v]=v;        
            rev[t]=v;
            dfs2(v,u);
        }
    }
}
void built(int k,int l,int r)   //线段树建树
{
    if(l==r)    {Max[k]=sum[k]=w[rev[l]];return;} 	//初始化 
    int mid=(l+r)/2;
    built(2*k,l,mid);
    built(2*k+1,mid+1,r);
    sum[k]=sum[2*k]+sum[2*k+1];
    Max[k]=max(Max[2*k],Max[2*k+1]);
} 
void change(int k,int l,int r,int x,int v)  //单点修改 
{
    if(l>r) return;
    if(l==r&&x==l)  {sum[k]=Max[k]=v;return;}
    int mid=(l+r)/2;
    if(x>=l&&x<=mid) change(2*k,l,mid,x,v);
    if(x>=mid+1&&x<=r) change(2*k+1,mid+1,r,x,v);
    sum[k]=sum[2*k]+sum[2*k+1];
    Max[k]=max(Max[2*k],Max[2*k+1]);
}
void query(int k,int l,int r,int x,int y)
{
    if(x>r||y<l) return ;
    if(x<=l&&r<=y) 
    {
        ans_sum+=sum[k];
        ans_max=max(ans_max,Max[k]);
        return;
    }
    int mid=(l+r)/2;
    if(x<=mid) query(2*k,l,mid,x,y);
    if(y>=mid+1) query(2*k+1,mid+1,r,x,y);
} 
void ask(int u,int v)
{       
    int fu=top[u],fv=top[v];
    while(fu!=fv)   //不在同一条重链上 
    {
        if(deep[fu]<deep[fv]) {swap(u,v);swap(fu,fv);}   //选择深度大的往上跳
        query(1,1,n,id[fu],id[u]);  //访问路径 fu->u 
        u=father[fu];
        fu=top[u]; 
    }
    if(deep[u]>deep[v]) swap(u,v);   //已经跳到同一条重路径上了 
    query(1,1,n,id[u],id[v]);
}
int main()
{
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    int x,y;
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        Add(x,y);
        Add(y,x);
    }
    for(int i=1;i<=n;i++)
        scanf("%d",&w[i]);
    scanf("%d",&m);
    dfs1(1,0);
    id[1]=++t;  //初始化根节点 
    top[1]=1;
    rev[1]=1;
    dfs2(1,0); 
    built(1,1,n); 
    char s[10];
    for(int i=1;i<=m;i++)
    {
        scanf("%s%d%d",s,&x,&y);  
        if(s[0]=='C')   change(1,1,n,id[x],y);
        else
        {
            ans_sum=0;
            ans_max=0x80000000;		//int最小值 
            ask(x,y); 				//路径询问 
            if(s[0]=='Q'&&s[1]=='M')    printf("%d\n",ans_max);
            if(s[0]=='Q'&&s[1]=='S')    printf("%d\n",ans_sum);
        }
         
    }
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值