codevs 2370 小机房的树

转载

mb5fd86caa0a310 2017-08-20 22:00:00

文章标签 lca #include i++ 数据 ios 文章分类 代码人生

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

1 0 1

2 0 1

1 0

2 0

1 2

样例输出 Sample Output

数据范围及提示 Data Size & Hint

1<=n<=50000， 1<=m<=75000， 0<=c<=1000

分类标签 Tags 点此展开

思路：lca水题

#include<iostream>
#include<map>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define MAXN 700000+15
using namespace std;
int T,n,m,tot;
int dad[MAXN*4],deep[MAXN*4],size[MAXN*4],top[MAXN*4],length[MAXN*4];
int to[MAXN*5],from[MAXN*5],net[MAXN*5],cap[MAXN*5];
void add(int u,int v,int w){
    to[++tot]=v;net[tot]=from[u];from[u]=tot;cap[tot]=w;
    to[++tot]=u;net[tot]=from[v];from[v]=tot;cap[tot]=w;
}
void dfs(int x){
    size[x]=1;
    deep[x]=deep[dad[x]]+1;
    for(int i=from[x];i;i=net[i])
        if(dad[x]!=to[i]){
            dad[to[i]]=x;
            length[to[i]]=length[x]+cap[i];
            dfs(to[i]);
            size[x]+=size[to[i]];
        }
}
void dfs1(int x){
    int t=0;
    if(!top[x])    top[x]=x;
    for(int i=from[x];i;i=net[i])
        if(dad[x]!=to[i]&&size[t]<size[to[i]])
            t=to[i];
    if(t){
        top[t]=top[x];
        dfs1(t);
    }
    for(int i=from[x];i;i=net[i])
        if(dad[x]!=to[i]&&t!=to[i])
            dfs1(to[i]);
}
int lca(int x,int y){
    for(;top[x]!=top[y];){
        if(deep[top[x]]<deep[top[y]])
            swap(x,y);
        x=dad[top[x]];
    }
    if(deep[x]>deep[y])
        swap(x,y);
    return x;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        int u,v,z;
        scanf("%d%d%d",&u,&v,&z);
        add(u+1,v+1,z);
    }
    dfs(1);
    dfs1(1);
    cin>>m;
    for(int i=1;i<=m;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        u+=1;v+=1;
        int l=lca(u,v);
        cout<<length[u]+length[v]-2*length[l]<<endl;
    }
}