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”的操作，每行输出一个整数表示要求输出的结果。

分析：

　　裸的树链剖分。注意一下，输入的节点权值的下标要改为在线段树上的节点下标。

  1 #include <set>

  2 #include <map>

  3 #include <list>

  4 #include <cmath>

  5 #include <queue>

  6 #include <stack>

  7 #include <string>

  8 #include <vector>

  9 #include <cstdio>

 10 #include <cstring>

 11 #include <iostream>

 12 #include <algorithm>

 13 

 14 using namespace std;

 15 

 16 typedef long long ll;

 17 typedef unsigned long long ull;

 18 

 19 #define debug puts("here")

 20 #define rep(i,n) for(int i=0;i<n;i++)

 21 #define rep1(i,n) for(int i=1;i<=n;i++)

 22 #define REP(i,a,b) for(int i=a;i<=b;i++)

 23 #define foreach(i,vec) for(unsigned i=0;i<vec.size();i++)

 24 #define pb push_back

 25 #define RD(n) scanf("%d",&n)

 26 #define RD2(x,y) scanf("%d%d",&x,&y)

 27 #define RD3(x,y,z) scanf("%d%d%d",&x,&y,&z)

 28 #define RD4(x,y,z,w) scanf("%d%d%d%d",&x,&y,&z,&w)

 29 #define All(vec) vec.begin(),vec.end()

 30 #define MP make_pair

 31 #define PII pair<int,int>

 32 #define PQ priority_queue

 33 #define cmax(x,y) x = max(x,y)

 34 #define cmin(x,y) x = min(x,y)

 35 #define Clear(x) memset(x,0,sizeof(x))

 36 /*

 37 

 38 #pragma comment(linker, "/STACK:1024000000,1024000000")

 39 

 40 int size = 256 << 20; // 256MB

 41 char *p = (char*)malloc(size) + size;

 42 __asm__("movl %0, %%esp\n" :: "r"(p) );

 43 

 44 */

 45 

 46 /******** program ********************/

 47 

 48 const int MAXN = 100005;

 49 const int INF = 1e9;

 50 

 51 int val[MAXN];

 52 int po[MAXN],tol;

 53 int sz[MAXN],dep[MAXN],fa[MAXN],son[MAXN],tid[MAXN],top[MAXN];

 54 bool use[MAXN];

 55 int tot;

 56 int mx,msum;

 57 

 58 struct node{

 59     int y,next;

 60 }edge[MAXN];

 61 

 62 struct Tree{

 63     int l,r,mx,sum;

 64     inline int mid(){

 65         return (l+r)>>1;

 66     }

 67 }tree[MAXN<<2];

 68 

 69 inline void add(int x,int y){

 70     edge[++tol].y = y;

 71     edge[tol].next = po[x];

 72     po[x] = tol;

 73 }

 74 

 75 void dfsFind(int x,int pa,int depth){

 76     use[x] = true;

 77     dep[x] = depth;

 78     fa[x] = pa;

 79     sz[x] = 1;

 80     son[x] = 0;

 81 

 82     for(int i=po[x];i;i=edge[i].next){

 83         int y = edge[i].y;

 84         if(use[y])continue;

 85         dfsFind(y,x,depth+1);

 86         sz[x] += sz[y];

 87         if(sz[y]>sz[ son[x] ])

 88             son[x] = y;

 89     }

 90 }

 91 

 92 void dfsCon(int x,int pa){

 93     use[x] = true;

 94     tid[x] = ++ tot;

 95     top[x] = pa;

 96     if(son[x])

 97         dfsCon(son[x],pa);

 98     for(int i=po[x];i;i=edge[i].next){

 99         int y = edge[i].y;

100         if(use[y])continue;

101         dfsCon(y,y);

102     }

103 }

104 

105 inline void update(int rt){

106     tree[rt].sum = tree[rt<<1].sum+tree[rt<<1|1].sum;

107     tree[rt].mx = max( tree[rt<<1].mx , tree[rt<<1|1].mx );

108 }

109 

110 void build(int l,int r,int rt){

111     tree[rt].l = l;

112     tree[rt].r = r;

113     if(l==r){

114         tree[rt].sum = tree[rt].mx = val[l];

115         return;

116     }

117     int mid = tree[rt].mid();

118     build(l,mid,rt<<1);

119     build(mid+1,r,rt<<1|1);

120 

121     update(rt);

122 }

123 

124 void modify(int pos,int c,int rt){

125     if(tree[rt].l==tree[rt].r){

126         tree[rt].sum = tree[rt].mx = c;

127         return;

128     }

129     int mid = tree[rt].mid();

130     if(pos<=mid)

131         modify(pos,c,rt<<1);

132     else

133         modify(pos,c,rt<<1|1);

134 

135     update(rt);

136 }

137 

138 void ask(int l,int r,int rt){

139     if(tree[rt].l==l&&tree[rt].r==r){

140         mx = max(mx,tree[rt].mx);

141         //cmax( mx , tree[rt].mx );

142         msum += tree[rt].sum;

143         return;

144     }

145     int mid = tree[rt].mid();

146     if(r<=mid)

147         ask(l,r,rt<<1);

148     else if(l>mid)

149         ask(l,r,rt<<1|1);

150     else{

151         ask(l,mid,rt<<1);

152         ask(mid+1,r,rt<<1|1);

153     }

154 }

155 

156 int main(){

157 

158 #ifndef ONLINE_JUDGE

159     freopen("sum.in","r",stdin);

160     //freopen("sum.out","w",stdout);

161 #endif

162 

163     int n,m,x,y;

164     while(cin>>n){

165         Clear(po);

166         tol = 1;

167 

168         REP(i,2,n){

169             RD2(x,y);

170             add(x,y);

171             add(y,x);

172         }

173 

174         Clear(use);

175         dfsFind(1,1,1);

176 

177         Clear(use);

178         tot = 0;

179         dfsCon(1,1);

180 

181         rep1(i,n){

182             RD(y);

183             val[ tid[i] ] = y;

184         }

185 

186         build(1,n,1);

187 

188         RD(m);

189         char op[10];

190         while(m--){

191             scanf("%s%d%d",op,&x,&y);

192             if(op[1]=='H'){

193                 modify(tid[x],y,1);

194             }else{

195                 mx = -INF;

196                 msum = 0;

197 

198                 while( top[x] != top[y] ){

199                     if( dep[ top[x] ] < dep[ top[y] ] )

200                         swap(x,y);

201                     ask( tid[ top[x] ] , tid[x] , 1 );

202                     x = fa[ top[x] ];

203                 }

204                 if(dep[x]>dep[y])

205                     swap(x,y);

206                 ask(tid[x],tid[y],1);

207 

208                 printf("%d\n",op[1]=='M'?mx:msum);

209             }

210         }

211     }

212 

213     return 0;

214 }