比赛 寒假集训5 评测结果 AAAAAAAAAA
题目名称 白色相簿的季节 最终得分 100
用户昵称 小福鑫 运行时间 2.188 s
代码语言 C++ 内存使用 23.64 MiB
提交时间 2026-03-01 12:58:17
显示代码纯文本
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int INF=1e18;
int n,q,k,s,t,h[200005],to[400005],val[400005],nxt[400005],cnt,ky[200005],fa[200005][20],dep[200005],dis[200005],g[200005],mn[200005][20];
void add(int a,int b,int c) {
	to[++cnt]=b;
	val[cnt]=c;
	nxt[cnt]=h[a];
	h[a]=cnt;
}
void dijkstra(){
	priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>>pq;
	for(int i=0;i<=n;i++)g[i]=INF;
	for(int i=1;i<=k;i++) {
		g[ky[i]]=0;
		pq.push({0,ky[i]});
	}
	while(!pq.empty()) {
		int d=pq.top().first;
		int u=pq.top().second;
		pq.pop();
		if(d>g[u])continue;
		for(int i=h[u];i;i=nxt[i]){
			int v=to[i];
			if(g[v]>g[u]+val[i]){
				g[v]=g[u]+val[i];
				pq.push({g[v],v});
			}
		}
	}
}
void dfs(int u,int f){
	dep[u]=dep[f]+1;
	fa[u][0]=f;
	mn[u][0]=min(g[u],g[f]);
	for(int i=1;i<20;i++){
		fa[u][i]=fa[fa[u][i-1]][i-1];
		mn[u][i]=min(mn[u][i-1],mn[fa[u][i-1]][i-1]);
	}
	for(int i=h[u];i;i=nxt[i]){
		int v=to[i];
		if(v==f)continue;
		dis[v]=dis[u]+val[i];
		dfs(v,u);
	}
}
int lca(int u,int v) {
	if(dep[u]<dep[v])swap(u,v);
	for(int i=19;i>=0;i--) {
		if(dep[fa[u][i]]>=dep[v]) {
			u=fa[u][i];
		}
	}
	if(u==v)return u;
	for(int i=19;i>=0;i--){
		if(fa[u][i]!=fa[v][i]){
			u=fa[u][i],v=fa[v][i];
		}
	}
	return fa[u][0];
}
int get_min(int u,int v){
	int res=g[u];
	for(int i=19; i>=0; i--){
		if(dep[fa[u][i]]>=dep[v]){
			res=min(res,mn[u][i]);
			u=fa[u][i];
		}
	}
	return res;
}
signed main() {
    freopen("wa.in","r",stdin);
    freopen("wa.out","w",stdout);
	cin>>n>>q>>k;
	for(int i=1,u,v,w;i<n;i++){
		cin>>u>>v>>w;
		add(u,v,w);
		add(v,u,w);
	}
	for(int i=1;i<=k;i++){
		cin>>ky[i];
	}
	dijkstra();
	for(int i=0;i<20;i++)mn[0][i]=INF;
	dfs(1,0);
	while(q--){
		cin>>s>>t;
		int l=lca(s,t);
		int pd=dis[s]+dis[t]-2*dis[l];
		int md=min(get_min(s,l),get_min(t,l));
		cout<<pd+2*md<<"\n";
	}
}