HDU 3966 树链剖分

HDU 3966
题目链接：
http://acm.hdu.edu.cn/showproblem.php?pid=3966
题意：
n个（<=1e5）兵营，以树的形式相互连接构成连通图。给出初始每个兵营的兵数量。
然后有修改操作，即u-v的路径上（包括u，v）经过的兵营兵的数量全减少或增加某一个值。
有查询操作，求一个兵营当前兵的数量。输出这个数量。
思路：
裸的树链剖分。
各种写错。
容易T的地方：
1）solve处的u=fa[tp1]写成u=fa[u]
2）Solve处各种写反
容易WA的地方：
1）solve处u=v则退出
2）建线段树时，tree[o].val = val[id[l]]而不是tree[o].val = val[tid[l]]（要用一个新的指针指回去，搞清楚线段树中这个点到底是哪一个点）
3）线段树更新时没有及时push_down。
源码：

#pragma comment(linker,"/STACK:100000000,100000000")
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
const int MAXN = 50000 + 5;
int tp[MAXN], fa[MAXN], num[MAXN], son[MAXN], dep[MAXN];
int id[MAXN], cnt, tid[MAXN];
vector<int>lin[MAXN];
void dfs1(int u, int f, int deep)
{
    num[u] = 1, fa[u] = f, son[u] = 0, dep[u] = deep;
    for(int i = 0 ; i < (int)lin[u].size() ; i++){
        int v = lin[u][i];
        if(v == f)  continue;
        dfs1(v, u, deep + 1);
        num[u] += num[v];
        if(num[son[u]] < num[v])    son[u] = v;
    }
}
void dfs2(int u, int f)
{
    id[u] = ++cnt;
    tid[cnt] = u;
    tp[u] = f;
    if(son[u])  dfs2(son[u], f);
    for(int i = 0 ; i < (int)lin[u].size() ; i++){
        int v = lin[u][i];
        if(v == fa[u] || v == son[u])   continue;
        dfs2(v, v);
    }
}
struct Tree
{
    int l, r, val;
    int add;
}tree[MAXN * 4];
int val[MAXN];
void build(int l, int r, int o)
{
    tree[o].l = l, tree[o].r = r, tree[o].add = 0, tree[o].val = 0;
    if(l == r){
        tree[o].val = val[tid[l]];
        return;
    }
    int mid = (l + r) / 2;
    if(mid >= l)    build(l, mid, o * 2);
    if(mid < r)     build(mid + 1, r, o * 2 + 1);
}
void push_down(int o)
{
    if(tree[o].add == 0)    return;
    tree[o * 2].add += tree[o].add;
    tree[o * 2 + 1].add += tree[o].add;
    tree[o].add = 0;
}
int query(int u, int o)
{
// printf("u = %d, o = %d\n", u, o);
// printf("tree[o].l = %d, tree[o].r = %d\n", tree[o].l, tree[o].r);
// system("pause");
    if(tree[o].l == tree[o].r && tree[o].l == u){
        return tree[o].add + tree[o].val;
    }
    int ans = 0;
    push_down(o);
    int mid = (tree[o].l + tree[o].r) / 2;
    if(mid >= u)    ans = query(u, o * 2);
    else    ans = query(u, o * 2 + 1);
    return ans;
}
void update(int l, int r, int val, int o)
{
// if(tree[o].r < l || tree[o].l > r) return;
// printf("l = %d, r = %d, val = %d, tree[o].l = %d, tree[o].r = %d, o = %d\n", l, r, val, tree[o].l, tree[o].r, o);
// system("pause");
    if(tree[o].l >= l && tree[o].r <= r){
        tree[o].add += val;
        return;
    }
    push_down(o);
    int mid = (tree[o].l + tree[o].r) / 2;
    if(mid < l) update(l, r, val, o * 2 + 1);
    else if(mid >= r)  update(l, r, val, o * 2);
    else{
        update(l, r, val, o * 2);
        update(l, r, val, o * 2 + 1);
    }
}
void solve(int u, int v, int val)
{
    int tp1 = tp[u], tp2 = tp[v];
    while(tp1 != tp2){
        if(dep[tp1] < dep[tp2]) swap(tp1, tp2), swap(u, v);
        update(id[tp1], id[u], val, 1);
        u = fa[tp1];
        tp1 = tp[u];
    }
    if(dep[u] > dep[v]) swap(u, v);
    update(id[u], id[v], val, 1);
}
char op[1000];
int main()
{
    int n, m, q;
    while(scanf("%d%d%d", &n, &m, &q) != EOF){
        for(int i = 0 ; i <= n ; i++)   lin[i].clear();
        int u, v, w;
        for(int i = 1 ; i <= n ; i++)   scanf("%d", &val[i]);
        for(int i = 0 ; i < m ; i++){
            scanf("%d%d", &u, &v);
            lin[u].push_back(v);
            lin[v].push_back(u);
        }
        cnt = 0;
        dfs1(1, 0, 1);
        dfs2(1, 1);
        build(1, n, 1);
        while(q--){
            scanf("%s", op);
            if(op[0] == 'Q'){
                scanf("%d", &u);
// printf("first = %d, second = %d\n", query(id[u], 1), val[id[u]]);
                printf("%d\n", query(id[u], 1));
            }
            else{
                scanf("%d%d%d", &u, &v, &w);
                if(op[0] == 'D')    solve(u, v, -w);
                else    solve(u, v, w);
            }
        }
    }
    return 0;
}