poj 2763 Housewife Wind

题目链接：http://poj.org/problem?id=2763

题意：

给出n个节点的一棵树，树上每个边表示走过这条边需要的时间 。有q个询问，人初始在s点。

有两种操作：

1. 0 u 表示人从当前点走到u点。 此时输出本次移动需要的时间。

2. 1 i w 表示把第i条路的权值变为w。

思路：

权值在边上的树链剖分+线段树单点更新+线段树成段询问

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
#define maxn 100010
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
int n, q, s, tot;
struct Node
{
    int to, next;
    Node(int t, int n)
    {
        to = t; next = n;
    }
    Node(){}
}mp[maxn<<2];
//vector <Node> mp[maxn]; //不能用vector建图
int head[maxn];
void AddEdge(int x, int y)
{
    mp[tot].to = y;
    mp[tot].next = head[x];
    head[x] = tot++;
}
struct Edge
{
    int from, to, val;
}e[maxn];
int pos;
int siz[maxn], dep[maxn], top[maxn], fa[maxn], son[maxn], w[maxn], fw[maxn];
long long cnt[maxn<<2];
void PushUp(int rt)
{
    cnt[rt] = cnt[rt<<1] + cnt[rt<<1|1];
}
void build(int l, int r, int rt)
{
    cnt[rt] = 0;
    if(l == r) return;
    int m = (l+r)>>1;
    build(lson);
    build(rson);
    PushUp(rt);
}
void update(int p, int val, int l, int r, int rt)
{
    if(l == r)
    {
        cnt[rt] = val; return;
    }
    int m = (l+r)>>1;
    if(p <= m) update(p, val, lson);
    else update(p, val, rson);
    PushUp(rt);
}
long long query(int L, int R, int l, int r, int rt)
{
    if(L <= l && R >= r)
    {
        return cnt[rt];
    }
    long long ret = 0;
    int m = (l+r)>>1;
    if(L <= m) ret += query(L, R, lson);
    if(R > m) ret += query(L, R, rson);
    return ret;
}
int dfs1(int u, int pre, int deep)
{
    siz[u] = 1; fa[u] = pre; dep[u] = deep;
    int mmax = 0;
   // for(int i = 0; i < mp[u].size(); i++)
    for(int i = head[u]; i != -1; i = mp[i].next)
    {
       // if(mp[u][i].to != pre)
        if(mp[i].to != pre)
        {
            int temp = dfs1(mp[i].to, u, deep+1);
            siz[u] += temp;
            if(son[u] == -1 || temp >= mmax) son[u] = mp[i].to;        
        }
    }
    return siz[u];
}
void dfs2(int u, int val)
{
    top[u] = val;
    if(son[u] != -1)
    {
        w[u] = pos++;
        fw[w[u]] = u;
        dfs2(son[u], val);
    }
    else if(son[u] == -1)
    {
        w[u] = pos++;
        fw[w[u]] = u;
        return;
    }
   // for(int i = 0; i < mp[u].size(); i++)
    for(int i = head[u]; i != -1; i = mp[i].next)
    {
        //if(mp[u][i].to != fa[u] && mp[u][i].to != son[u]) dfs2(mp[u][i].to, mp[u][i].to);
        if(mp[i].to != fa[u] && mp[i].to != son[u]) dfs2(mp[i].to, mp[i].to);
    }
}
long long find(int u, int v)
{
    int f1 = top[u], f2 = top[v];
    long long temp = 0;
    while(f1 != f2)
    {
        if(dep[f1] < dep[f2])
        {
            swap(f1, f2);
            swap(u, v);
        }
        temp += query(w[f1], w[u], 1, pos-1, 1);
        u = fa[f1]; f1 = top[u];
    }
    if(u == v) return temp;
    if(dep[u] > dep[v]) swap(u, v);
    temp += query(w[son[u]], w[v], 1, pos-1, 1);
    return temp;
}
int main() 
{
    //freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
    while(~scanf("%d%d%d", &n, &q, &s))
    {
       // for(int i = 1; i <= n; i++) mp[i].clear();
        pos = 0; tot = 0; memset(son, -1, sizeof(son));
        memset(head, -1, sizeof(head));
        
        for(int i = 1; i <= n-1; i++)
        {
            scanf("%d%d%d", &e[i].from, &e[i].to, &e[i].val);
            AddEdge(e[i].from, e[i].to);
            AddEdge(e[i].to, e[i].from);
           // mp[e[i].from].push_back(Node(e[i].to, e[i].val));
           // mp[e[i].to].push_back(Node(e[i].from, e[i].val));
        }
        dfs1(1, -1, 1);
        dfs2(1, 1);
        build(1, pos-1, 1);
        
        for(int i = 1; i <= n-1; i++)
        {
            if(dep[e[i].from] > dep[e[i].to])
            {
                swap(e[i].from, e[i].to);
            }
            update(w[e[i].to], e[i].val, 1, pos-1, 1);
        }
        while(q--)
        {
            int op; scanf("%d", &op);
            if(op == 0)
            {
                int u; scanf("%d", &u);
                printf("%lld\n", find(s, u));
                s = u;
            }
            else if(op == 1)
            {
                int i, ww; scanf("%d%d", &i, &ww);
                update(w[e[i].to], ww, 1, pos-1, 1);
            }
        }
    }
    return 0;
}