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本身
Input
输入的第一行为一个整数n,表示节点的个数。接下来n – 1行,每行2个整数a和b,表示节点a和节点b之间有一条边相连。接下来n行,每行一个整数,第i行的整数wi表示节点i的权值。接下来1行,为一个整数q,表示操作的总数。接下来q行,每行一个操作,以“CHANGE u t”或者“QMAX u v”或者“QSUM u v”的形式给出。 对于100%的数据,保证1<=n<=30000,0<=q<=200000;中途操作中保证每个节点的权值w在-30000到30000之间。
Output
对于每个“QMAX”或者“QSUM”的操作,每行输出一个整数表示要求输出的结果。
Sample Input
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 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
Sample Output
4
1
2
2
10
6
5
6
5
16
题解
一道裸的树链剖分。下面说说树链剖分的思路和注意的地方吧。
树链剖分支持树上修改和查询。就是把树的剖分成一条一条链连起来,然后用线段树在链上进行操作,这条链上的节点的编号是根据先重儿子后轻儿子的原则,重链又由轻链连接,重儿子就是节点数最多的儿子。这个过程要两遍dfs。求两点之间的距离是由轻链跳到重链,重链跳到头,直到两点在一条重链上,因为重链上的序号是连续的,线段树可以算。
1
2
2
10
6
5
6
5
16
题解
一道裸的树链剖分。下面说说树链剖分的思路和注意的地方吧。
树链剖分支持树上修改和查询。就是把树的剖分成一条一条链连起来,然后用线段树在链上进行操作,这条链上的节点的编号是根据先重儿子后轻儿子的原则,重链又由轻链连接,重儿子就是节点数最多的儿子。这个过程要两遍dfs。求两点之间的距离是由轻链跳到重链,重链跳到头,直到两点在一条重链上,因为重链上的序号是连续的,线段树可以算。
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cstring>
#define ll long long
#define INF 1000000000
using namespace std;
const int maxn=100010;
ll w[maxn];
int pre[maxn],other[maxn],last[maxn],dep[maxn],num,n,m,bl[maxn];
int top[maxn],fa[maxn],sz[maxn],dfn[maxn],son[maxn],belong[maxn];
char a[10];
struct tree{
int l,r;
ll sum,mx;
}t[maxn*4];
void add(int x,int y){
num++;
pre[num]=last[x];
last[x]=num;
other[num]=y;
}
void dfs1(int x){
sz[x]=1;
for(int i=last[x];i;i=pre[i]){
int v=other[i];
if(v!=fa[x]){
dep[v]=dep[x]+1;
fa[v]=x;
dfs1(v);
sz[x]+=sz[v];
if(!son[x]||sz[son[x]]<sz[v])
son[x]=v;
}
}
}
int cnt;
void dfs2(int x,int tp){
dfn[x]=++cnt;
top[x]=tp;
belong[cnt]=x;
if(son[x]) dfs2(son[x],tp);
for(int i=last[x];i;i=pre[i]){
int v=other[i];
if(v==fa[x]||v==son[x]) continue;
dfs2(v,v);
}
}
void build(int x,int l,int r){
t[x].l=l;t[x].r=r;
if(l==r){
t[x].sum=w[belong[l]];
t[x].mx=w[belong[l]];
return ;
}
int mid=(l+r)>>1;
build(x*2,l,mid);
build(x*2+1,mid+1,r);
t[x].sum=t[x*2].sum+t[x*2+1].sum;
t[x].mx=max(t[x*2].mx,t[x*2+1].mx);
}
void change(int x,int l,ll k){
if(t[x].l>l||t[x].r<l)
return ;
if(t[x].l==l&&t[x].r==l){
t[x].sum=k;
t[x].mx=k;
return ;
}
change(x*2,l,k);
change(x*2+1,l,k);
t[x].sum=t[x*2].sum+t[x*2+1].sum;
t[x].mx=max(t[x*2].mx,t[x*2+1].mx);
}
ll querymx(int x,int l,int r){
if(t[x].l>r||t[x].r<l)
return -INF;
if(t[x].l>=l&&t[x].r<=r)
return t[x].mx;
return max(querymx(x*2,l,r),querymx(x*2+1,l,r));
}
ll query(int x,int l,int r){
if(t[x].l>r||t[x].r<l)
return 0;
if(t[x].l>=l&&t[x].r<=r)
return t[x].sum;
return query(x*2,l,r)+query(x*2+1,l,r);
}
ll work(int l,int r){
int f1=top[l],f2=top[r];
ll ans=-INF;
while(f1!=f2){
if(dep[f1]<dep[f2])
swap(f1,f2),swap(l,r);
ans=max(ans,querymx(1,dfn[f1],dfn[l]));
l=fa[f1],f1=top[l];
}
if(dep[l]>dep[r]) swap(l,r);
ans=max(ans,querymx(1,dfn[l],dfn[r]));
return ans;
}
ll solve(int l,int r){
int f1=top[l],f2=top[r];
ll ans=0;
while(f1!=f2){
if(dep[f1]<dep[f2])
swap(f1,f2),swap(l,r);
ans+=query(1,dfn[f1],dfn[l]);
l=fa[f1],f1=top[l];
}
if(dep[l]>dep[r]) swap(l,r);
ans+=query(1,dfn[l],dfn[r]);
return ans;
}
int main(){
int x,y;
scanf("%d",&n);
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("%lld",&w[i]);
scanf("%d",&m);
dfs1(1);
dfs2(1,1);
build(1,1,cnt);
/*for(int i=1;i<=n;i++)
change(1,dfn[i],w[i]);*/
for(int i=1;i<=m;i++){
scanf("%s",a);
scanf("%d%d",&x,&y);
if(a[0]=='C'){
change(1,dfn[x],y);
}
else{
if(a[1]=='M'){
printf("%lld\n",work(x,y));
}
else{
printf("%lld\n",solve(x,y));
}
}
}
return 0;
}