洛谷P2912 [USACO08OCT]Pasture Walking G

最新推荐文章于 2022-05-21 15:40:14 发布

tylon2006

最新推荐文章于 2022-05-21 15:40:14 发布

阅读量178

点赞数

分类专栏： LCA

本文链接：https://blog.csdn.net/tylon2006/article/details/107958421

版权

LCA 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

Description

N头牛（2<=n<=1000）别人被标记为1到n，在同样被标记1到n的n块土地上吃草，第i头牛在第i块牧场吃草。这n块土地被n-1条边连接。奶牛可以在边上行走，第i条边连接第Ai，Bi块牧场，第i条边的长度是Li（1<=Li<=10000）。这些边被安排成任意两头奶牛都可以通过这些边到达的情况，所以说这是一棵树。这些奶牛是非常喜欢交际的，经常会去互相访问,他们想让你去帮助他们计算Q(1<=q<=1000)对奶牛之间的距离。

Input

第一行：两个被空格隔开的整数：N和Q

第二行到第n行：第i+1行有两个被空格隔开的整数：AI，BI，LI

第n+1行到n+Q行：每一行有两个空格隔开的整数：P1，P2，表示两头奶牛的编号。

Output

第1行到第Q行：每行输出一个数，表示那两头奶牛之间的距离。

Solution

？？？无脑lca
循环的时候注意建边初始编号

Code

#include<bits/stdc++.h>
using namespace std;
struct Edge{
    int v,s,nxt;
}e[400010];
int head[200010],tot;
int siz[200010],n,q;
int son[200010];
int dep[200010];
int top[200010];
int dis[200010];
int fa[200010];
int id[200010];
void addEdge(int u,int v,int s){
    ++ tot;
    e[tot].v = v;
    e[tot].s = s;
    e[tot].nxt = head[u];
    head[u] = tot;
}
inline void ad(int u,int v,int s){
    addEdge(u,v,s);
    addEdge(v,u,s);
}
void dfs1(int u,int father,int depth){
    siz[u]=1;
    fa[u]=father;
    dep[u]=depth;
    int maxn=-1;
    for(int i=head[u];i;i=e[i].nxt){
        int v=e[i].v,s=e[i].s;
        if(v==father) continue;
        dis[v]=dis[u]+s;
        dfs1(v,u,depth+1);
        siz[u]+=siz[v];
        //cout<<u<<" "<<v<<" "<<dis[u]<<" "<<s<<endl;
        if(siz[v]>maxn){
            maxn=siz[v];
            son[u]=v;
        }
    }
}
void dfs2(int u,int ltop){
    top[u]=ltop;
    if(!son[u]) return;
    dfs2(son[u],ltop);
    for(int i=head[u];i;i=e[i].nxt){
        int v=e[i].v;
        if(v==fa[u]||v==son[u])
        continue;
        dfs2(v,v);
    }
}
int lca(int x,int y){
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        x=fa[top[x]];
    }
    if(dep[x]>dep[y]) swap(x,y);
    return x;
} 
int main(){
	int x,y,z;
	scanf("%d%d",&n,&q);
	for(int i=1;i<n;i++){
		scanf("%d%d%d",&x,&y,&z);
		ad(x,y,z);
	}
	dfs1(1,-1,0);
	dfs2(1,1);
	//for(int i=1;i<=n;i++)
	//cout<<dis[i]<<endl;
	for(int i=1;i<=q;i++){
		scanf("%d%d",&x,&y);
		printf("%d\n",dis[x]+dis[y]-2*dis[lca(x,y)]);
	}
}