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
Sample Output
4 1 2 2 10 6 5 6 5 16
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <queue>
#include <iostream>
#include <algorithm>
#define lson l,mid,o<<1
#define rson mid+1,r,o<<1|1
using namespace std;
const int maxn = 30005;
int n;
int son[maxn]; //表示该点重链的子节点
int num[maxn]; //表示以该点为根的子树的节点数量
int dep[maxn]; //表示该节点的深度
int fa[maxn]; // 表示该节点的父亲节点
int vis[maxn]; // 表示该节点在线段树中的编号
int top[maxn]; // 表示这条重链的顶端节点
int now;
int head[maxn],cnt,A,B,C;
struct Edge
{
int v,next;
}e[maxn<<1];
void add(int a,int b)
{
e[cnt].v = b;
e[cnt].next = head[a];
head[a] = cnt++;
}
void init()
{
memset(head,-1,sizeof(head));
memset(son,-1,sizeof(son));
now = cnt = 0;
}
void dfs1(int u,int v,int step) //第一遍dfs求出son,fa,num,dep
{
num[v] = 1;dep[v] = step;fa[v] = u;
int i;
for(i = head[v]; i != -1; i = e[i].next)
{
int end = e[i].v;
if(end == u)
continue;
dfs1(v,end,step+1);
if(son[v] == -1 || num[son[v]] < num[end])// 取子节点数量最多的点
{
son[v] = end;
}
num[v] += num[end];
}
}
void dfs2(int u,int v)// 第二遍dfs求出top和vis
{
top[v] = u;
vis[v] = ++now; // 一个节点在线段树里面代表一个点
if(son[v] != -1) //将重链在线段树里连续编号
{
dfs2(u,son[v]);
}
else
return;
int i;
for(i = head[v] ; i != -1; i = e[i].next)
{
int end = e[i].v;
if(end == fa[v] || end == son[v])//等于父亲节点或者等于重链的子节点跳过,这里的u不是父亲节点,是重链的顶端
continue;
dfs2(end,end); // 所有轻链重新为其节点的重链编号
}
}
int sum[maxn<<2],ma[maxn<<2];
void update(int l,int r,int o)
{
if(l == r)
{
sum[o] = ma[o] = B;
return;
}
int mid = (l + r) >> 1;
if(A <= mid) update(lson);
else update(rson);
sum[o] = sum[o<<1] + sum[o << 1 | 1];
ma[o] = max(ma[o<<1],ma[o<<1|1]);
}
int ansma,anss;
void qurry(int l,int r,int o)
{
if(A <= l && r <= B)
{
anss += sum[o];
ansma = max(ansma,ma[o]);
return;
}
int mid = (l + r) >> 1;
if(A <= mid) qurry(lson);
if(mid < B) qurry(rson);
}
//更新的时候记得用vis取得线段树下标,不要用原来树中的编号。
void solve(int a,int b,int c)
{
if(c == 0)
{
A = vis[a];
B = b;
update(1,n,1);
}
else
{
int q = top[a],w = top[b];
ansma = -1000005,anss = 0;
while(q != w)
{
if(dep[q] < dep[w])
{
swap(q,w);
swap(a,b);
}
A = vis[q] , B = vis[a];
qurry(1,n,1);
//记得更新a跟q顺序不能反
a = fa[q];
q = top[a];
}
if(dep[a] > dep[b])
{
swap(a,b);
}
A = vis[a],B = vis[b];
qurry(1,n,1);
if(c == 1)
printf("%d\n",ansma);
else
printf("%d\n",anss);
}
}
int main()
{
while(scanf("%d",&n) != EOF)
{
init();
int i,j,a1,b1;
for(i = 1; i < n; i++)
{
scanf("%d %d",&a1,&b1);
add(a1,b1);
add(b1,a1);
}
dfs1(1,1,1);
dfs2(1,1);
for(i = 1; i <= n; i++)
{
scanf("%d",&a1);
A = vis[i];B = a1;
update(1,n,1);
}
int m;
scanf("%d",&m);
char str[10];
while(m--)
{
scanf("%s %d %d",str,&a1,&b1);
int now = 0;
if(str[0] == 'Q' && str[1] == 'M')
now = 1;
else if(str[0] == 'Q' && str[1] == 'S')
now = 2;
solve(a1,b1,now);
}
}
return 0;
}