codevs 2370 小机房的树

2370 小机房的树
时间限制: 1 s
空间限制: 256000 KB
题目等级 : 钻石 Diamond
题解
题目描述 Description
小机房有棵焕狗种的树，树上有N个节点，节点标号为0到N-1，有两只虫子名叫飘狗和大吉狗，分居在两个不同的节点上。有一天，他们想爬到一个节点上去搞基，但是作为两只虫子，他们不想花费太多精力。已知从某个节点爬到其父亲节点要花费 c 的能量（从父亲节点爬到此节点也相同），他们想找出一条花费精力最短的路，以使得搞基的时候精力旺盛，他们找到你要你设计一个程序来找到这条路，要求你告诉他们最少需要花费多少精力

输入描述 Input Description
第一行一个n，接下来n-1行每一行有三个整数u，v, c 。表示节点 u 爬到节点 v 需要花费 c 的精力。
第n+1行有一个整数m表示有m次询问。接下来m行每一行有两个整数 u ，v 表示两只虫子所在的节点
输出描述 Output Description
一共有m行，每一行一个整数，表示对于该次询问所得出的最短距离。

样例输入 Sample Input
3

1 0 1

2 0 1

3

1 0

2 0

1 2

样例输出 Sample Output
1

1

2

数据范围及提示 Data Size & Hint
1<=n<=50000， 1<=m<=75000， 0<=c<=1000

思路： 裸lca
倍增or树剖

//倍增版 vector
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<cmath>
using namespace std;
const int maxn=50000+50;
vector<int>map[maxn],cost[maxn];
int deep[maxn],rank[maxn],fa[maxn][25];
int n,m,z,a,b;
void build(int f,int t,int v){
    map[f].push_back(t);
    cost[f].push_back(v);
    return ;
}
void done(int f,int t,int v){
    deep[t]=deep[f]+1;
    fa[t][0]=f;rank[t]=v;
    for(int i=0;i<map[t].size();i++){
        if(map[t][i]!=f)
        done(t,map[t][i],v+cost[t][i]);
    }
    return ;
}
void make_lca(){
    for(int i=1;i<=log2(n);i++)
    for(int j=1;j<=n;j++)
    fa[j][i]=fa[fa[j][i-1]][i-1];
}
int lca(int x,int y){

    if(deep[x]<deep[y])     swap(x,y);
    for(int i=log2(n);i>=0;i--){
        if(deep[fa[x][i]]>=deep[y]) {
            x=fa[x][i];
        }
    }
    if(x==y) return x;
    for(int j=log2(n);j>=0;j--){
        if(fa[x][j]!=fa[y][j]){
            x=fa[x][j];y=fa[y][j];
        }
    }
    return fa[x][0];
}
int main(){
    scanf("%d",&n);
    int p1,p2,p3;
    for(int i=1;i<n;i++){
        scanf("%d%d%d",&p1,&p2,&p3);
        build(p1,p2,p3);
        build(p2,p1,p3);
    }

    done(0,0,0);make_lca();
    /*for(int i=0;i<n;i++){
        printf("\ni %d deep[i] %d rank[i] %d ",i,deep[i],rank[i]);
        for(int j=0;j<=log2(n);j++)
        printf(" fa[%d][%d]=%d ",i,j,fa[i][j]);
    }*/

    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        scanf("%d%d",&a,&b);
        printf("%d\n",rank[a]+rank[b]-2*rank[lca(a,b)]);

    }
    return 0;
}

//tarjian版 vector
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=100000;
vector<int> map[maxn],cost[maxn],ch[maxn],sh[maxn];
int fa[maxn],ans[maxn],rank[maxn],n,m,a,b,tot,ans2[maxn];
bool vis[maxn];

void build(int f,int t,int v){
    map[f].push_back(t);
    cost[f].push_back(v);
    return ;
}
int find(int w){
    if(fa[w]==w) return w;
    if(fa[w]!=w){
        return fa[w]=find(fa[w]);
    }
}
void dfs(int l,int now){
//  printf("l %d now %d vis[now] %d \n",l,now,vis[now]);
//  for(int i=0;i<n;i++)
//  printf("%d  ",fa[i]);
//  printf("\n");
    tot++;
    if(map[now].size()){
        for(int i=0;i<map[now].size();i++){
        if(vis[map[now][i]]==0&&map[now][i]!=l){
            dfs(now,map[now][i]);
        }
    }
    }


   // printf("now %d  fa %d\n",now,fa[now]);
    for(int j=0;j<ch[now].size();j++){
        if(vis[ch[now][j]]==1&&ans[sh[now][j]]==0){
            ans2[sh[now][j]]=find(ch[now][j]);
            ans[sh[now][j]]=rank[now]+rank[ch[now][j]]-2*rank[find(ch[now][j])];

        }
    }
    vis[now]=1;
    fa[now]=l;
    return ;
}
void done(int f,int t,int v){
    //deep[t]=deep[f]+1;
    rank[t]=v;fa[t]=t;
    for(int i=0;i<map[t].size();i++){
        if(map[t][i]!=f)
        done(t,map[t][i],v+cost[t][i]);
    }
    return ;
}
void jian(){
    for(int i=0;i<n;i++){
        for(int j=0;j<map[i].size();j++)
        printf("%d",map[i][j]);
        printf("\n");
    }

    return ;
}
int main(){
    freopen("data2.txt","w",stdout);
    scanf("%d",&n);
    int p1,p2,p3;
    for(int i=1;i<n;i++){
        scanf("%d%d%d",&p1,&p2,&p3);
        build(p1,p2,p3);
        build(p2,p1,p3);
    }
    done(0,0,0);
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        scanf("%d%d",&a,&b);
        ch[a].push_back(b);
        sh[a].push_back(i);
        ch[b].push_back(a);
        sh[b].push_back(i);
    }
    dfs(0,0);
  //    jian();
    for(int i=1;i<=m;i++)
    printf("%d\n",ans[i]);

    return 0;
}

//树剖lca 
#include<iostream>
#include<cstdio>
using namespace std;
const int N=500000+60,M=750000+60;
int n,m,a,b,c,tot;
struct edge{
    int t,v;
}e[M];

int first[N],next[M],fa[N],deep[N],top[N],sz[N],son[N],dis[N];


void build(int ff,int tt,int vv){
    e[++tot]=(edge){tt,vv};
    next[tot]=first[ff];
    first[ff]=tot;
    return ;
}
void dfs1(int x,int fx){
    fa[x]=fx;deep[x]=deep[fx]+1;
    sz[x]=1;
    for(int i=first[x];i;i=next[i]){
        if(e[i].t==fx) continue;
        dis[e[i].t]=dis[x]+e[i].v;
        dfs1(e[i].t,x);
        sz[x]+=sz[e[i].t];
        if(!son[x]||sz[son[x]]<sz[e[i].t]) 
        son[x]=e[i].t;   
    }
    return ;
}
void dfs2(int t,int tp){
    top[t]=tp;
    if(!son[t]) return ;
    dfs2(son[t],tp);
    for(int i=first[t];i;i=next[i]){
        int v=e[i].t;
        if(v==fa[t]||v==son[t]) continue;
        dfs2(v,v);
    }
    return ;
}
int find(int x,int y){
    while(top[x]!=top[y]){
        if(deep[top[y]]>deep[top[x]])
        swap(x,y);
        x=fa[top[x]];
    }
    if(deep[x]>deep[y]) return y;
    else return x;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        scanf("%d%d%d",&a,&b,&c);
        build(a,b,c);build(b,a,c);
    }
    dfs1(0,0);
    dfs2(0,0);
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        int lca=find(x,y);
        printf("%d\n",dis[x]+dis[y]-2*dis[lca]);
    }
    return 0;
}