LOJ145 DFS序2（dfs序模板）

题目：LOJ145.

这只是个用来屯板子的blog，原文.

代码如下：

#include<bits/stdc++.h> 
  using namespace std;
  
#define Abigail inline void
typedef long long LL;

const int N=1000000;

int Ri(){
  int x=0,y=1;
  char c=getchar();
  for (;c<'0'||c>'9';c=getchar()) if (c=='-') y=-1;
  for (;c<='9'&&c>='0';c=getchar()) x=x*10+c-'0';
  return x*y;
}

struct side{
  int y,next;
}e[N*2+9];
int lin[N+9],top,n,m,rot;

void Ins(int x,int y){
  e[++top].y=y;
  e[top].next=lin[x];
  lin[x]=top;
}

int ld[N+9],rd[N+9],ord[N+9],co;
LL v[N+9],c[2][N+9];

void Dfs(int k,int fa){
  ord[++co]=k;ld[k]=co;
  for (int i=lin[k];i;i=e[i].next)
    if (e[i].y^fa) Dfs(e[i].y,k);
  rd[k]=co;
}

void Bit_add(int t,int x,LL v){for (;x<=n;x+=x&-x) c[t][x]+=v;}
LL Bit_query(int t,int x){LL sum=0;for (;x;x-=x&-x) sum+=c[t][x];return sum;}
LL Bit_query(int t,int l,int r){return Bit_query(t,r)-Bit_query(t,l-1);}
void Add_seg(int l,int r,LL v){Bit_add(0,l,v);Bit_add(1,l,v*(l-1));Bit_add(0,r+1,-v);Bit_add(1,r+1,-v*r);}
LL Query_seg(int l,int r){return Bit_query(0,r)*r-Bit_query(1,r)-Bit_query(0,l-1)*(l-1)+Bit_query(1,l-1);}
void Add_tree(int x,LL v){Add_seg(ld[x],rd[x],v);}
LL Query_tree(int x){return Query_seg(ld[x],rd[x]);}
void Build(){for (int i=1;i<=n;++i) Add_seg(ld[i],ld[i],v[i]);}

Abigail into(){
  n=Ri();m=Ri();rot=Ri();
  int x,y;
  for (int i=1;i<=n;++i) v[i]=(LL)Ri();
  for (int i=1;i<n;++i){
  	x=Ri();y=Ri();
  	Ins(x,y);Ins(y,x);
  }
}

Abigail work(){
  Dfs(rot,0);
  Build();
}

Abigail getans(){
  int opt,a;
  while (m--){
  	opt=Ri();a=Ri();
  	if (opt==2) printf("%lld\n",Query_tree(a));
  	else Add_tree(a,(LL)Ri());
  }
}

int main(){
  into();
  work();
  getans();
  return 0;
}