zzuliOJ1916：树（DFS序 + 树状数组）

1916: 树

Time Limit: 1 Sec  Memory Limit: 128 MB
Submit: 226  Solved: 19

SubmitStatusWeb Board

Description

给一颗树,有n个结点,编号为1到n，1为根节点,有两种操作，1 x y把x结点权值加y,2 x查询x到根节点所有结点的权值和.
每个结点权值初始化为0。

Input

第一行输入一个整数t，代表有t组测试数据。
每组数据第一行为两个整数n,m代表结点个数和操作次数。
接下来n-1行，每行两个整数a,b，表示a结点和b结点有一条边。
接下来m行每行一个操作格式如上。
0<=n<=50000 ,0<=m<=50000，0<=y<=50000

Output

对于每组查询输出一个整数表示结果

Sample Input

1

5 5

1 2

1 3

3 4

3 5

2 5

1 1 2

1 3 1

2 4

2 1

Sample Output

0

3

2

HINT

Source

haut

思路：因为涉及到多次修改权值，考虑差分的思想，先标记一下，查询的时候再计算前缀和，那么需要用dfs序将树重新编号，in和out分别表示进入节点i和离开节点i的时间戳，那么in[i]和out[i]之间的都是i的子树节点，in标记为正，out标记位负，查询时计算下前缀和即可，这里计算前缀和用树状数组加速。

# include <bits/stdc++.h>using namespace std;typedef long long LL;const int maxn = 5e4+3;int in[maxn], out[maxn], Next[maxn], cnt, cnt2;int t, n, m, a, b, c;LL sum[maxn<<1];struct node{    int e, next, w;}edge[maxn<<1];void add(int u, int v, int e){    edge[cnt].e = v;    edge[cnt].next = Next[u];    Next[u] = cnt++;}void dfs(int u, int pre){    in[u] = ++cnt2;    for(int i=Next[u]; i!=-1; i=edge[i].next)    {        int v = edge[i].e;        if(v == pre) continue;        dfs(v, u);    }    out[u] = ++cnt2;}int lowbit(int x){    return x&(-x);}void update(int x, int num){    for(int i=x; i<=(n<<1); i+=lowbit(i))        sum[i] += num;}LL query(int x){    LL ans = 0;    for(int i=x; i>=1; i-=lowbit(i))        ans += sum[i];    return ans;}int main(){    scanf("%d",&t);    while(t--)    {        cnt = cnt2 = 0;        memset(Next, -1, sizeof(Next));        memset(sum , 0, sizeof(sum));        scanf("%d%d",&n,&m);        for(int i=0; i<n-1; ++i)        {            scanf("%d%d",&a,&b);            add(a, b, 0);            add(b, a, 0);        }        dfs(1, -1);        while(m--)        {            scanf("%d",&c);            if(c == 1)            {                scanf("%d%d",&a,&b);                update(in[a], b);                update(out[a], -b);            }            else            {                scanf("%d",&a);                printf("%lld\n",query(in[a]));            }        }    }    return 0;}