HYSBZ 1036 树的统计Count（树链剖分）

思路：树链剖分的又一个模板题，不过坑了好久...不知道为什么输入询问的时候用string会RE...

#include<iostream>#include<cmath>#include<cstdio>#include<vector>#include<cstring>#include<string>using namespace std;const int maxn = 30005;#define lson i<<1,l,m#define rson i<<1|1,m+1,rint n,tot;int dep[maxn],son[maxn],siz[maxn],fa[maxn],top[maxn],id[maxn];vector<int>e[maxn];int val[maxn];int maxv[maxn*4];int sum[maxn*4];int temp[maxn];void push_up(int i){maxv[i] = max(maxv[i<<1],maxv[i<<1|1]);sum[i] = sum[i<<1]+sum[i<<1|1];}void build(int i,int l,int r){maxv[i]=-0x3f3f3f3f;if(l==r){maxv[i]=val[l];sum[i]=val[l];return;}int m = (l+r)>>1;build(lson);build(rson);push_up(i);}void update(int v,int vall,int i,int l,int r){    if(l==r){sum[i] = vall;maxv[i]=vall;return;}int m = (l+r)>>1;if(v <= m)update(v,vall,lson);elseupdate(v,vall,rson);push_up(i);}int query1(int ql,int qr,int i,int l,int r){if(ql<=l && qr>=r)return sum[i];int m = (l+r)>>1;int ans = 0;if(ql<=m)ans+=query1(ql,qr,lson);if(qr>m)ans+=query1(ql,qr,rson);return ans;}int query2(int ql,int qr,int i,int l,int r){if(ql<=l && qr>=r)return maxv[i];int m = (l+r)>>1;int ans = -1e9;if(ql<=m)ans = max(query2(ql,qr,lson),ans);if(qr>m)ans=max(query2(ql,qr,rson),ans);return ans;}void dfs1(int u,int f,int d){dep[u]=d;fa[u]=f;son[u]=0;siz[u]=1;for(int i = 0;i<e[u].size();i++){int v = e[u][i];if(v==f)continue;dfs1(v,u,d+1);siz[u]+=siz[v];if (siz[son[u]] < siz[v])son[u]=v;}}void dfs2(int u,int tp){top[u]=tp;id[u]=++tot;if(son[u])dfs2(son[u],tp);for(int i = 0;i<e[u].size();i++){int v = e[u][i];if(v==fa[u]|| v==son[u])continue;dfs2(v,v);}}int Yougth2(int u,int v){int tp1 = top[u],tp2 = top[v];int ans = -1e9;while(tp1!=tp2){if(dep[tp1] < dep[tp2]){swap(tp1,tp2);swap(u,v);}ans = max(query2(id[tp1],id[u],1,1,tot),ans);u = fa[tp1];tp1 = top[u];}if(dep[u] > dep[v])swap(u,v);ans=max(query2(id[u],id[v],1,1,tot),ans);return ans;}int Yougth1(int u,int v){int tp1 = top[u],tp2 = top[v];int ans = 0;while(tp1!=tp2){if(dep[tp1] < dep[tp2]){swap(tp1,tp2);swap(u,v);}ans+=query1(id[tp1],id[u],1,1,tot);u = fa[tp1];tp1 = top[u];}if(dep[u] > dep[v])swap(u,v);ans+=query1(id[u],id[v],1,1,tot);return ans;}int main(){    while(scanf("%d",&n)!=EOF){tot = 0;memset(val,0,sizeof(val));memset(son,0,sizeof(son));for(int i = 0;i<=n;i++)e[i].clear();for(int i =1;i<n;i++){int u,v;scanf("%d%d",&u,&v);e[u].push_back(v);e[v].push_back(u);}for(int i = 1;i<=n;i++)scanf("%d",&temp[i]);dfs1(1,0,1);dfs2(1,1);for(int i = 1;i<=n;i++)val[id[i]]=temp[i];build(1,1,tot);int q;scanf("%d",&q);while(q--){    char s[105];int a,b;scanf("%s%d%d",s,&a,&b);if(s[1] == 'M'){                  printf("%d\n",Yougth2(a,b));}if(s[1]=='S'){                  printf("%d\n",Yougth1(a,b));}if (s[1]=='H'){                 update(id[a],b,1,1,tot);}}}}

Description

　　一棵树上有n个节点，编号分别为1到n，每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成
一些操作： I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 I
II. 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

41 22 34 14 2 1 312QMAX 3 4QMAX 3 3QMAX 3 2QMAX 2 3QSUM 3 4QSUM 2 1CHANGE 1 5QMAX 3 4CHANGE 3 6QMAX 3 4QMAX 2 4QSUM 3 4

Sample Output

412210656516