题目描述
一棵树上有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”的操作,每行输出一个整数表示要求输出的结果。
题解:树链剖分的模板题,第一遍dfs确认轻重孩子,第二遍dfs拉出轻重链,每一条轻重链便成了一条条的序列,然后用线段树维护区间最大值和区间和即可。
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+7;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
int _dfs[maxn],son[maxn],far[maxn],siz[maxn],sum[maxn];
int dep[maxn],tot,top[maxn],n,a[maxn],id[maxn],tree[maxn];
vector<int>G[maxn];
void dfs1(int u,int fa,int depth)
{
far[u]=fa;
siz[u]=1;
dep[u]=depth;
int sz=G[u].size();
for(int i=0;i<sz;i++)
{
int v=G[u][i];
if(v==fa) continue;
dfs1(v,u,depth+1);
siz[u]+=siz[v];
if(siz[v]>siz[son[u]]) son[u]=v;
}
}
void dfs2(int x,int t)
{
_dfs[x]=++tot;
top[x]=t;
id[tot]=x;
if(son[x]) dfs2(son[x],t);
int sz=G[x].size();
for(int i=0;i<sz;i++)
{
int v=G[x][i];
if(v!=far[x] && v!=son[x])
dfs2(v,v);
}
}
void build(int l=1,int r=n,int rt=1)
{
if(l==r)
{
tree[rt]=a[id[l]];
sum[rt]=a[id[l]];
return;
}
int m=(l+r)>>1;
build(lson);build(rson);
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
tree[rt]=max(tree[rt<<1],tree[rt<<1|1]);
}
int querys(int L,int R,int l=1,int r=n,int rt=1)
{
if(l>=L && r<=R) return sum[rt];
int m=(l+r)>>1,ans=0;
if(L<=m) ans+=querys(L,R,lson);
if(R>m) ans+=querys(L,R,rson);
return ans;
}
int querym(int L,int R,int l=1,int r=n,int rt=1)
{
if(l>=L && r<=R) return tree[rt];
int m=(l+r)>>1,ans=-0x3f3f3f3f;
if(L<=m) ans=max(ans,querym(L,R,lson));
if(R>m) ans=max(ans,querym(L,R,rson));
return ans;
}
void update(int o,int v,int l=1,int r=n,int rt=1)
{
if(l==r)
{
tree[rt]=sum[rt]=v;
return;
}
int m=(l+r)>>1;
if(o<=m) update(o,v,lson);
else update(o,v,rson);
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
tree[rt]=max(tree[rt<<1],tree[rt<<1|1]);
}
int calm(int u,int v)
{
int ans=-0x3f3f3f3f;
while(top[u]!=top[v])
{
if(dep[top[u]]<dep[top[v]])
swap(u,v);
ans=max(ans,querym(_dfs[top[u]],_dfs[u]));
u=far[top[u]];
}
if(dep[u]>dep[v])
swap(u,v);
ans=max(ans,querym(_dfs[u],_dfs[v]));
return ans;
}
int cals(int u,int v)
{
int ans=0;
while(top[u]!=top[v])
{
if(dep[top[u]]<dep[top[v]])
swap(u,v);
ans+=querys(_dfs[top[u]],_dfs[u]);
u=far[top[u]];
}
if(dep[u]>dep[v])
swap(u,v);
ans+=querys(_dfs[u],_dfs[v]);
return ans;
}
int main()
{
scanf("%d",&n);
for(int i=0;i<n-1;i++)
{
int u,v;
scanf("%d%d",&u,&v);
G[u].push_back(v);
G[v].push_back(u);
}
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
dfs1(1,1,0);dfs2(1,1);
build();
int q;scanf("%d",&q);
while(q--)
{
char s[15]={0};
scanf("%s",s);
int u,v;
scanf("%d%d",&u,&v);
if(s[0]=='C')
update(_dfs[u],v);
else if(s[1]=='M')
{
int ans=calm(u,v);
printf("%d\n",ans);
}
else
{
int ans=cals(u,v);
printf("%d\n",ans);
}
}
return 0;
}