FZU Problem 2082 过路费

Problem 2082 过路费

Accept: 875    Submit: 2839
Time Limit: 1000 mSec    Memory Limit : 32768
KB

Problem Description

有n座城市，由n-1条路相连通，使得任意两座城市之间可达。每条路有过路费，要交过路费才能通过。每条路的过路费经常会更新，现问你，当前情况下，从城市a到城市b最少要花多少过路费。

Input

有多组样例，每组样例第一行输入两个正整数n,m(2 <= n<=50000，1<=m <=
50000),接下来n-1行，每行3个正整数a b c，(1 <= a,b <= n , a != b , 1 <= c <=
1000000000).数据保证给的路使得任意两座城市互相可达。接下来输入m行，表示m个操作，操作有两种：一. 0 a b，表示更新第a条路的过路费为b，1
<= a <= n-1 ； 二. 1 a b ， 表示询问a到b最少要花多少过路费。

Output

对于每个询问，输出一行，表示最少要花的过路费。

Sample Input

2 3
1 2 1
1 1 2
0 1 2
1 2 1

Sample Output

1
2

Source

FOJ有奖月赛-2012年4月（校赛热身赛）

 

思路：树链剖分

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<set>
#include<map>
#include<vector>
#include<stack>
#include<cmath>
#include<string>
#include<algorithm>
using namespace std;
typedef long long ll;
const int sigma_size=;
const int N=+;
const int MAXN=+;
const int inf=0x3fffffff;
const double eps=1e-;
const int mod=+;
#define L(x) (x<<1)
#define R(x) (x<<1|1)
#define PII pair<int, int>
#define mk(x,y) make_pair((x),(y))
int n,m,edge_cnt,cnt;
int head[MAXN],sz[MAXN],son[MAXN],fa[MAXN],top[MAXN],dep[MAXN],pos[MAXN];
struct Edge{
    int u,v,w,next;
}edge[MAXN<<];
struct node{
    int l,r;
    ll sum;
}segtree[MAXN<<];
void init(){
    edge_cnt=cnt=;
    memset(head,-,sizeof(head));
}
void addedge(int u,int v,int w){
    edge[edge_cnt].u=u;
    edge[edge_cnt].v=v;
    edge[edge_cnt].w=w;
    edge[edge_cnt].next=head[u];
    head[u]=edge_cnt++;
}
void dfs1(int u,int pre,int depth){
    sz[u]=;
    son[u]=;
    dep[u]=depth;
    fa[u]=pre;
    for(int i=head[u];i!=-;i=edge[i].next){
        int v=edge[i].v;
        if(v == pre)
            continue;
        dfs1(v,u,depth+);
        sz[u]+=sz[v];
        if(sz[son[u]]<sz[v])
            son[u]=v;
    }
}
void dfs2(int u,int tp){
    pos[u]=cnt++;
    top[u]=tp;
    if(son[u]!=)
        dfs2(son[u],top[u]);
    for(int i=head[u];i!=-;i=edge[i].next){
        int v=edge[i].v;
        if(v == fa[u] || v == son[u])
            continue;
        dfs2(v,v);
    }
}
void build(int rt,int l,int r){
    segtree[rt].l=l;
    segtree[rt].r=r;
    segtree[rt].sum=;
    if(l == r)
        return ;
    int mid=(l+r)>>;
    build(L(rt),l,mid);
    build(R(rt),mid+,r);
}
void push_up(int rt){
    segtree[rt].sum=segtree[L(rt)].sum+segtree[R(rt)].sum;
}
void update(int rt,int p,int x){
    if(segtree[rt].l == segtree[rt].r){
        segtree[rt].sum=x;
        return ;
    }
    int mid=(segtree[rt].l+segtree[rt].r)>>;
    if(p<=mid)
        update(L(rt),p,x);
    else
        update(R(rt),p,x);
    push_up(rt);
}
ll query(int rt,int l,int r){
    if(segtree[rt].l == l && segtree[rt].r == r)
        return segtree[rt].sum;
    int mid=(segtree[rt].l+segtree[rt].r)>>;
    if(r<=mid)
        return query(L(rt),l,r);
    else if(l>mid)
        return query(R(rt),l,r);
    else
        return query(L(rt),l,mid)+query(R(rt),mid+,r);
}
ll solve(int u,int v){
    ll ans=;
    while(top[u]!=top[v]){
        if(dep[top[u]]<dep[top[v]])
            swap(u,v);
        ans+=query(,pos[top[u]],pos[u]);
        u=fa[top[u]];
    }
    if(dep[u]>dep[v])
        swap(u,v);
    if(u!=v)
        ans+=query(,pos[u]+,pos[v]);
    return ans;
}
int main(){
    while(~scanf("%d%d",&n,&m)){
        init();
        for(int i=;i<n;i++){
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            addedge(u,v,w);
            addedge(v,u,w);
        }
        dfs1(,,);
        dfs2(,);
        build(,,n-);
        for(int i=;i<edge_cnt;i+=){
            int u=edge[i].u,v=edge[i].v,w=edge[i].w;
            if(dep[u]<dep[v])
                swap(u,v);
            update(,pos[u],w);
        }
        for(int i=;i<m;i++){
            int id,u,v;
            scanf("%d",&id);
            if(id == ){
                int x,w;
                scanf("%d%d",&x,&w);
                x=(x-)*;
                u=edge[x].u;
                v=edge[x].v;
                if(dep[u]<dep[v])
                    swap(u,v);
                update(,pos[u],w);
            }
            else{
                scanf("%d%d",&u,&v);
                printf("%lld\n",solve(u,v));
            }
        }
    }
}