题目链接:https://www.luogu.com.cn/problem/P2590
题目大意:
把模板的求和,加一个最大值
#include <bits/stdc++.h>
#define Rint register int
using namespace std;
#define mid ((l+r)>>1)
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
#define len (r-l+1)
const int maxn=100005;
int n,m;
//见题意
int tot,head[maxn],nex[maxn],to[maxn],w[maxn],wt[maxn];
//链式前向星数组,w[]、wt[]初始点权数组
int a[maxn<<2],laz[maxn<<2], Max[maxn<<2];
//线段树数组、lazy操作 MAX 最大值
int son[maxn],id[maxn],fa[maxn],cnt,dep[maxn],siz[maxn],top[maxn];
//son[]重儿子编号,id[]新编号,fa[]父亲节点,cnt dfs_clock/dfs序,dep[]深度,siz[]子树大小,top[]当前链顶端节点
int res=0;
//查询答案
inline void add(int x,int y){//链式前向星加边
to[++tot]=y;
nex[tot]=head[x];
head[x]=tot;
}
//-------------------------------------- 以下为线段树
inline void pushdown(int rt,int lenn){
laz[rt<<1]+=laz[rt];
laz[rt<<1|1]+=laz[rt];
a[rt<<1]+=laz[rt]*(lenn-(lenn>>1));
a[rt<<1|1]+=laz[rt]*(lenn>>1);
Max[rt<<1]+=laz[rt];
Max[rt<<1|1]+=laz[rt];
laz[rt]=0;
}
inline void build(int rt,int l,int r){
if(l==r){
a[rt]=wt[l];
Max[rt]=wt[l];
return;
}
build(lson);
build(rson);
a[rt]=(a[rt<<1]+a[rt<<1|1]);
Max[rt]=max(Max[rt<<1], Max[rt<<1|1]);
}
int MAX;
inline void queryMax(int rt,int l,int r,int L,int R){
if(L<=l&&r<=R){MAX=max(MAX, Max[rt]);return;}
else{
if(laz[rt])pushdown(rt,len);
if(L<=mid)queryMax(lson,L,R);
if(R>mid)queryMax(rson,L,R);
}
}
inline void querySum(int rt,int l,int r,int L,int R){
if(L<=l&&r<=R){res+=a[rt];return;}
else{
if(laz[rt])pushdown(rt,len);
if(L<=mid)querySum(lson,L,R);
if(R>mid)querySum(rson,L,R);
}
}
inline void update(int rt,int l,int r,int L,int R,int k){
if(L<=l&&r<=R){
laz[rt]+=k;
a[rt]+=k*len;
Max[rt]+=k;
}
else{
if(laz[rt])pushdown(rt,len);
if(L<=mid)update(lson,L,R,k);
if(R>mid)update(rson,L,R,k);
a[rt]=(a[rt<<1]+a[rt<<1|1]);
Max[rt]=max(Max[rt<<1], Max[rt<<1|1]);
}
}
inline void updRange(int x,int y,int k){//x-y + k
while(top[x]!=top[y]){
if(dep[top[x]]<dep[top[y]])swap(x,y);
update(1,1,n,id[top[x]],id[x],k);
x=fa[top[x]];
}
if(dep[x]>dep[y])swap(x,y);
update(1,1,n,id[x],id[y],k);
}
inline int qRangeSUM(int x,int y){//x-y和
int ans=0;
while(top[x]!=top[y]){//当两个点不在同一条链上
if(dep[top[x]]<dep[top[y]])swap(x,y);//把x点改为所在链顶端的深度更深的那个点
res=0;
querySum(1,1,n,id[top[x]],id[x]);//ans加上x点到x所在链顶端 这一段区间的点权和
ans+=res;
x=fa[top[x]];//把x跳到x所在链顶端的那个点的上面一个点
}
//直到两个点处于一条链上
if(dep[x]>dep[y])swap(x,y);//把x点深度更深的那个点
res=0;
querySum(1,1,n,id[x],id[y]);//这时再加上此时两个点的区间和即可
ans+=res;
return ans;
}
inline int qRangeMAX(int x,int y){//x-y最大值
MAX=-50000000;
while(top[x]!=top[y]){//当两个点不在同一条链上
if(dep[top[x]]<dep[top[y]])swap(x,y);//把x点改为所在链顶端的深度更深的那个点
queryMax(1,1,n,id[top[x]],id[x]);//ans加上x点到x所在链顶端 这一段区间的点权和
x=fa[top[x]];//把x跳到x所在链顶端的那个点的上面一个点
}
//直到两个点处于一条链上
if(dep[x]>dep[y])swap(x,y);//把x点深度更深的那个点
queryMax(1,1,n,id[x],id[y]);//这时再加上此时两个点的区间和即可
return MAX;
}
//当然,区间修改就和区间查询一样的啦~~
inline void updSon(int x,int k){//x +k
update(1,1,n,id[x],id[x]+siz[x]-1,k);
}
inline int qSonSum(int x){
res=0;
querySum(1,1,n,id[x],id[x]+siz[x]-1);
return res;
}
inline int qSonMax(int x){
MAX=-50000000;
queryMax(1,1,n,id[x],id[x]+siz[x]-1);
return MAX;
}
inline void dfs1(int x,int f,int deep){//x当前节点,f父亲,deep深度
dep[x]=deep;//标记每个点的深度
fa[x]=f;//标记每个点的父亲
siz[x]=1;//标记每个非叶子节点的子树大小
int maxson=-1;//记录重儿子的儿子数
for(Rint i=head[x];i;i=nex[i]){
int y=to[i];
if(y==f)continue;//若为父亲则continue
dfs1(y,x,deep+1);//dfs其儿子
siz[x]+=siz[y];//把它的儿子数加到它身上
if(siz[y]>maxson)son[x]=y,maxson=siz[y];//标记每个非叶子节点的重儿子编号
}
}
inline void dfs2(int x,int topf){//x当前节点,topf当前链的最顶端的节点
id[x]=++cnt;//标记每个点的新编号
wt[cnt]=w[x];//把每个点的初始值赋到新编号上来
top[x]=topf;//这个点所在链的顶端
if(!son[x])return;//如果没有儿子则返回
dfs2(son[x],topf);//按先处理重儿子,再处理轻儿子的顺序递归处理
for(Rint i=head[x];i;i=nex[i]){
int y=to[i];
if(y==fa[x]||y==son[x])continue;
dfs2(y,y);//对于每一个轻儿子都有一条从它自己开始的链
}
}
int main(){
scanf("%d", &n);
for(Rint i=1;i<n;i++){
int a,b;
scanf("%d%d", &a, &b);
add(a,b);add(b,a);
}
for(Rint i=1;i<=n;i++){
scanf("%d", &w[i]);//点权值
}
dfs1(1,0,1);
dfs2(1,1);
build(1,1,n);
scanf("%d", &m);
while(m--){
char k[10];
int x, y;
scanf("%s", k);
if(k[1]=='H'){
scanf("%d%d", &x, &y);//x点变y
int t=y;
y=y-w[x];
w[x]=t;
updRange(x,x,y);//a-b +c
}
else if(k[1]=='M'){
scanf("%d%d", &x, &y);
printf("%d\n", qRangeMAX(x,y));//x-y 最大值
}
else if(k[1]=='S'){
scanf("%d%d", &x, &y);
printf("%d\n", qRangeSUM(x,y));//x-y 和
}
}
}