Codevs_P2370 小机房的树(LCA)

题目描述 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,但比较麻烦,先DFS计算深度,在处理它2^i的数据,需要两个数组存跳2^i步的位置及权值,存算跳2^i步的时候忘记加上前面走过的了,调了半天qwq
PS:出题人脑洞有点大23333333

#include
#include
#include
#include
using namespace std;
#define N 50005
int n,m;
int deep[N];
int f[N][31],t[N][31];
struct g{
    vector<int> q;
    vector<int> v;
}w[N];
void dfs(int x,int d){
    deep[x]=d;
    for(int i=0;iif(!deep[w[x].q[i]]){
            dfs(w[x].q[i],d+1);
            f[w[x].q[i]][0]=x;
            t[w[x].q[i]][0]=w[x].v[i];
        }
}
void work(){
    for(int i=1;i<=30;i++)
        for(int j=0;j1]][i-1];
            t[j][i]=t[j][i-1]+t[f[j][i-1]][i-1];
        }
}
int lca(int u,int v){
    if(deep[u]int l=deep[u]-deep[v],vv=0;
    if(l)
        for(int i=0;i<=30;i++){
            if((1<if(u==v) return vv;
    for(int i=30;i>=0;i--){
        if(f[u][i]!=f[v][i]){
            vv+=t[u][i]+t[v][i];
            u=f[u][i];v=f[v][i];
        }
    }
    vv+=t[u][0]+t[v][0];
    return vv;
}
int main(){
    scanf("%d",&n);int u,v,c;
    for(int i=1;iscanf("%d%d%d",&u,&v,&c);
        w[u].q.push_back(v);w[u].v.push_back(c);
        w[v].q.push_back(u);w[v].v.push_back(c);
    }
    dfs(0,1);work();
    scanf("%d",&m);
    while(m--){
        scanf("%d%d",&u,&v);
        printf("%d\n",lca(u,v));
    }
}

你可能感兴趣的:(Codevs,图论,LCA)