倍增LCA模板

#include<bits/stdc++.h>
#define N 100005
#define P pair<ll,int>
using namespace std;
typedef long long ll;
const int M=1e9+7;
const int inf=1e9+7;
ll d[N],b[N];
int deep[N],f[N][20],a[N],R[N];
bool ji[N],r[N];
vector<P>v[N];
void dfs(int p,int fa)
{
    if(r[p])R[p]=p;
    else R[p]=R[fa];
    d[p]=b[p]-b[R[p]];
    deep[p]=deep[fa]+1;
    f[p][0]=fa;
    for(int i=1;i<20;i++)
        f[p][i]=f[f[p][i-1]][i-1];
    for(int i=0;i<v[p].size();i++){
        int to=v[p][i].first;
        if(to==fa)continue;
        b[to]=b[p]+v[p][i].second;
        dfs(to,p);
    }
}
int LCA(int x,int y)
{
    if(deep[x]<deep[y])swap(x,y);
    for(int i=19;i>=0;i--)
        if(deep[f[x][i]]>=deep[y])x=f[x][i];
    if(x==y)return x;
    for(int i=19;i>=0;i--)
        if(f[x][i]!=f[y][i])x=f[x][i],y=f[y][i];
    return f[x][0];
}
bool cmp(int x,int y){
    return d[x]<d[y];
}
int main()
{
    int t,n,m,q,k;
    for(scanf("%d",&t);t;t--)
    {
        scanf("%d%d%d",&n,&m,&q);
        for(int i=1;i<=n;i++){
            d[i]=1e15;
            ji[i]=r[i]=0;
            v[i].clear();
        }
        while(m--){
            int x;
            scanf("%d",&x);
            r[x]=1;
        }
        for(int i=1;i<n;i++){
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            v[x].push_back({y,z});
            v[y].push_back({x,z});
        }
        dfs(1,0);
        while(q--){
            scanf("%d",&k);
            for(int i=1;i<=k;i++)
                scanf("%d",&a[i]);
            sort(a+1,a+k+1,cmp);
            ll ans=d[a[k-1]],mmax=0;
            int last=a[k];
            for(int i=k-1;i;i--){
                int now=LCA(last,a[i]);
                mmax=max(mmax+b[last]-b[now],min(b[a[i]]-b[now],d[a[i]]));
                ans=min(ans,max(mmax,d[a[i-1]]));
                last=now;
            }
            printf("%lld\n",ans);
        }
    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值