比赛 寒假集训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;
}