| 比赛 |
寒假集训5 |
评测结果 |
AAAAAWEEEE |
| 题目名称 |
白色相簿的季节 |
最终得分 |
50 |
| 用户昵称 |
rzzakioi |
运行时间 |
1.435 s |
| 代码语言 |
C++ |
内存使用 |
8.51 MiB |
| 提交时间 |
2026-03-01 09:10:42 |
显示代码纯文本
#include<cstdio>
#include<queue>
#include<cstring>
#define pii pair<int,int>
using namespace std;
int n,q,k,to[200005],val[200005],nxt[200005],h[200005];
int cnt,imp[100005],dis[1005][1005];
int f[100005][25],d[100005][25],dep[100005];
bool vis[1005];
void add(int u,int v,int w){
to[++cnt]=v;
val[cnt]=w;
nxt[cnt]=h[u];
h[u]=cnt;
}
priority_queue<pii,vector<pii>,greater<pii> >Q;
void dijkstra(int st){
memset(vis,0,sizeof(vis));
dis[imp[st]][imp[st]]=0;
Q.push(make_pair(0,imp[st]));
while(!Q.empty()){
int u=Q.top().second;
Q.pop();
if(vis[u])continue;
vis[u]=1;
for(int i=h[u];i;i=nxt[i]){
int v=to[i];
if(dis[imp[st]][v]>dis[imp[st]][u]+val[i]){
dis[imp[st]][v]=dis[imp[st]][u]+val[i];
Q.push(make_pair(dis[imp[st]][v],v));
}
}
}
}
void dfs(int u,int fa,int w){
f[u][0]=fa;
dep[u]=dep[fa]+1;
d[u][0]=w;
for(int i=h[u];i;i=nxt[i]){
int v=to[i];
if(v==fa)continue;
dfs(v,u,val[i]);
}
}
int solve(int u,int v){
int ans=0;
if(dep[u]<dep[v])swap(u,v);
for(int i=20;i>=0;i--){
if(dep[f[u][i]]>=dep[v]){
ans+=d[u][i];
u=f[u][i];
}
}
if(u==v)return ans;
for(int i=20;i>=0;i--){
if(f[u][i]!=f[v][i]){
ans+=d[u][i];
u=f[u][i];
ans+=d[v][i];
v=f[v][i];
}
}
return ans+d[u][0]+d[v][0];
}
int main(){
freopen("wa.in","r",stdin);
freopen("wa.out","w",stdout);
scanf("%d%d%d",&n,&q,&k);
for(int i=1;i<n;i++){
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
add(u,v,w);
add(v,u,w);
}
for(int i=1;i<=k;i++){
scanf("%d",&imp[i]);
}
if(k==n){
dfs(1,0,0);
for(int j=1;(1<<j)<=n;j++){
for(int i=1;i<=n;i++){
f[i][j]=f[f[i][j-1]][j-1];
d[i][j]=d[i][j-1]+d[f[i][j-1]][j-1];
}
}
for(int i=1;i<=q;i++){
int u,v;
scanf("%d%d",&u,&v);
printf("%d\n",solve(u,v));
}
return 0;
}
memset(dis,0x3f,sizeof(dis));
for(int i=1;i<=k;i++){
dijkstra(i);
}
for(int i=1;i<=q;i++){
int u,v;
scanf("%d%d",&u,&v);
int ans=0x3f3f3f3f;
for(int j=1;j<=k;j++){
ans=min(ans,dis[imp[j]][u]+dis[imp[j]][v]);
}
printf("%d\n",ans);
}
return 0;
}