BZOJ4034 T2

Description

有一棵点数为 N 的树，以点 1 为根，且树点有边权。然后有 M 个操作，分为三种：

操作 1 ：把某个节点 x 的点权增加 a 。

操作 2 ：把某个节点 x 为根的子树中所有点的点权都增加 a 。

操作 3 ：询问某个节点 x 到根的路径中所有点的点权和。

 

Input

第一行包含两个整数 N, M 。表示点数和操作数。

接下来一行 N 个整数，表示树中节点的初始权值。

接下来 N-1 行每行三个正整数 fr, to ， 表示该树中存在一条边 (fr, to) 。

再接下来 M 行，每行分别表示一次操作。其中第一个数表示该操

作的种类（ 1-3 ） ，之后接这个操作的参数（ x 或者 x a ） 。

 

Output

对于每个询问操作，输出该询问的答案。答案之间用换行隔开。

 

Sample Input

5 5
1 2 3 4 5
1 2
1 4
2 3
2 5
3 3
1 2 1
3 5
2 1 2
3 3

Sample Output

6
9
13

HINT

对于 100% 的数据， N,M<=100000 ，且所有输入数据的绝对值都不

会超过 10^6 。

 

 

正解：树链剖分

解题报告：

　　链剖裸题。

　　但是我居然调试了很久！！！先是迷之RE，结果是细节问题。。。然后是迷之WA，结果是中间变量没开long long

　　多么痛的领悟。。。

 

 

 //It is made by jump~
 #include <iostream>
 #include <cstdlib>
 #include <cstring>
 #include <cstdio>
 #include <cmath>
 #include <algorithm>
 #include <ctime>
 #include <vector>
 #include <queue>
 #include <map>
 #include <set>
 #ifdef WIN32
 #define OT "%I64d"
 #else
 #define OT "%lld"
 #endif
 using namespace std;
 typedef long long LL;
 const int MAXN = ;
 const int MAXM = ;
 int n,m;
 int ecnt,cnt;
 int first[MAXN],to[MAXM],next[MAXM],id[MAXN],pre[MAXN],last[MAXN],deep[MAXN],father[MAXN],size[MAXN],son[MAXN],top[MAXN];
 int num[MAXN];
 LL qx,qy;
 int ql,qr;
 LL ans;

 struct node{
     LL add;
     LL val;
 }t[MAXN*];

 inline int getint()
 {
        int w=,q=;
        char c=getchar();
        while((c<'' || c>'') && c!='-') c=getchar();
        if (c=='-')  q=, c=getchar();
        while (c>='' && c<='') w=w*+c-'', c=getchar();
        return q ? -w : w;
 }

 inline void dfs1(int x,int fa) {
     size[x] = ;
     for(int i = first[x];i;i = next[i]) {
     int v = to[i];
     if(v != fa){
         father[v] = x; deep[v] = deep[x]+;
         dfs1(v,x);
         size[x] += size[v]; if(size[v] > size[son[x]]) son[x] = v;
     }
     }
 }

 inline void dfs2(int x,int fa){
     id[x] = ++cnt; pre[cnt]=x; if(son[x]!=) top[son[x]]=top[x],dfs2(son[x],x);
     for(int i = first[x];i;i = next[i]){
     int v = to[i];
     if(v != fa && v != son[x]) {
         top[v]=v;
         dfs2(v,x);
     }
     }
     last[x] = cnt;
 }

 inline void build(int root,int l,int r){
     if(l == r) { t[root].val = num[pre[l]]; return ; }
     int mid = (l + r)/; int lc = root*,rc = lc+;
     build(lc,l,mid); build(rc,mid+,r);
     t[root].val = t[lc].val + t[rc].val;
 }

 inline void update(int root,int l,int r){
     if(ql <= l && qr >= r) { t[root].add += qy; t[root].val += qy*(r-l+); }
     else{
     int mid = (l + r)/;
     int lc = root*,rc = lc + ;
     if(ql <= mid) update(lc,l,mid); if(qr > mid) update(rc,mid+,r);
     t[root].val = t[lc].val + t[rc].val;
     t[root].val += t[root].add*(r-l+);
     }
 }

 inline void query(int root,int l,int r,LL lei){
     if(ql <= l && qr >= r) {
     ans += t[root].val;
     ans += (LL)lei*(LL)(r-l+);
     return ;
     }
     int mid = (l+r)/; int lc = root*,rc = lc+;
     if(ql <= mid) query(lc,l,mid,lei+t[root].add); if(qr > mid) query(rc,mid+,r,lei+t[root].add);
 }

 inline void work(int x){
     ans=;
     int f1 = top[x];
     while(f1!=) {
     ql=id[f1]; qr=id[x];
     query(,,n,);
     x=father[f1]; f1=top[x];
     }
     ql=; qr=id[x]; query(,,n,);
     printf(OT"\n",ans);
 }

 inline void solve(){
     n = getint(); m = getint();
     for(int i=;i<=n;i++) num[i] = getint();
     int x,y;
     for(int i = ;i < n;i++) {
     x = getint(); y = getint();
     next[++ecnt] = first[x]; first[x] = ecnt; to[ecnt] = y;
     next[++ecnt] = first[y]; first[y] = ecnt; to[ecnt] = x;
     }

     deep[]=; dfs1(,);
     top[]=; dfs2(,);
     build(,,n);
     int ljh;
     for(int i = ;i <= m;i++) {
     ljh = getint();
     if(ljh == ){
         qx = id[getint()]; qy = getint();
         ql=qx; qr=qx;
         update(,,n);
     }
     else if(ljh == ){
         x = getint(); qy = getint();
         qr = last[x];  ql = id[x];
         update(,,n);
     }
     else{
         qx=getint(); work(qx);
     }
     }
 }

 int main()
 {
   solve();
   return ;
 }