https://www.luogu.org/problemnew/show/P2590
题目描述
一棵树上有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”的操作,每行输出一个整数表示要求输出的结果。
输入输出样例
输入样例#1: 复制
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
输出样例#1: 复制
4 1 2 2 10 6 5 6 5 16
说明
对于100%的数据,保证1<=n<=30000,0<=q<=200000;中途操作中保证每个节点的权值w在-30000到30000之间。
很基础的树剖
#include<cstdio>
#include<iostream>
using namespace std;
int read()
{
int ret=0; bool f=0;
char ch=getchar();
while(ch<'0'||ch>'9')
{
if(ch=='-') f=1;
ch=getchar();
}
while(ch>='0'&&ch<='9')
ret=(ret<<1)+(ret<<3)+ch-'0',ch=getchar();
return f?-ret:ret;
}
const int N=2e5+5;
int n,a[N],b[N];
int cnt,to[N],nxt[N],he[N];
int tot,dep[N],dad[N],siz[N],son[N],top[N],id[N];
char s[100];
inline void add(int u,int v)
{
to[++cnt]=v,nxt[cnt]=he[u],he[u]=cnt;
}
void dfs1(int fa,int u)
{
dep[u]=dep[fa]+1,dad[u]=fa,siz[u]=1;
int mx=0;
for(int e=he[u];e;e=nxt[e])
{
int v=to[e];
if(v!=fa)
{
dfs1(u,v);
siz[u]+=siz[v];
if(mx<siz[v]) son[u]=v,mx=siz[v];
}
}
}
void dfs2(int fa,int u,int t)
{
top[u]=t,id[u]=++tot;
if(!son[u]) return;
dfs2(u,son[u],t);
for(int e=he[u];e;e=nxt[e])
{
int v=to[e];
if(v!=son[u]&&v!=fa) dfs2(u,v,v);
}
}
struct NA
{
int c1[N],c2[N];
inline void up(int p)
{
c1[p]=max(c1[p<<1],c1[p<<1|1]);
c2[p]=c2[p<<1]+c2[p<<1|1];
}
void build(int p,int l,int r)
{
if(l==r)
{
c1[p]=c2[p]=b[l];
return;
}
int mid=l+r>>1;
build(p<<1,l,mid);
build(p<<1|1,mid+1,r);
up(p);
}
void add(int p,int l,int r,int x,int k)
{
if(l==r)
{
c1[p]=c2[p]=k;
return;
}
int mid=l+r>>1;
if(mid>=x) add(p<<1,l,mid,x,k);
else add(p<<1|1,mid+1,r,x,k);
up(p);
}
int getmax(int p,int l,int r,int x,int y)
{
if(l==x&&r==y) return c1[p];
int mid=l+r>>1;
if(y<=mid) return getmax(p<<1,l,mid,x,y);
else if(x>mid) return getmax(p<<1|1,mid+1,r,x,y);
else return max(getmax(p<<1,l,mid,x,mid),getmax(p<<1|1,mid+1,r,mid+1,y));
}
int getsum(int p,int l,int r,int x,int y)
{
if(l==x&&r==y) return c2[p];
int mid=l+r>>1;
if(y<=mid) return getsum(p<<1,l,mid,x,y);
else if(x>mid) return getsum(p<<1|1,mid+1,r,x,y);
else return getsum(p<<1,l,mid,x,mid)+getsum(p<<1|1,mid+1,r,mid+1,y);
}
}tree;
int Max(int u,int v)
{
int ret=-2e9;
while(top[u]!=top[v])
{
if(dep[top[u]]>dep[top[v]]) swap(u,v);
ret=max(ret,tree.getmax(1,1,n,id[top[v]],id[v]));
v=dad[top[v]];
}
if(id[u]>id[v]) swap(u,v);
ret=max(ret,tree.getmax(1,1,n,id[u],id[v]));
return ret;
}
int Sum(int u,int v)
{
int ret=0;
while(top[u]!=top[v])
{
if(dep[top[u]]>dep[top[v]]) swap(u,v);
ret+=tree.getsum(1,1,n,id[top[v]],id[v]);
v=dad[top[v]];
}
if(id[u]>id[v]) swap(u,v);
ret+=tree.getsum(1,1,n,id[u],id[v]);
return ret;
}
int main()
{
n=read();
for(int i=1;i<n;i++)
{
int u=read(),v=read();
add(u,v),add(v,u);
}
for(int i=1;i<=n;i++) a[i]=read();
dfs1(0,1); dfs2(0,1,1);
int T=read();
for(int i=1;i<=n;i++) b[id[i]]=a[i];
tree.build(1,1,n);
while(T--)
{
scanf("%s",s+1);
if(s[1]=='C')
{
int x=read(),y=read();
tree.add(1,1,n,id[x],y);
}else
if(s[2]=='M')
{
int x=read(),y=read();
printf("%d\n",Max(x,y));
}else
{
int x=read(),y=read();
printf("%d\n",Sum(x,y));
}
}
return 0;
}