ssl1746-商务旅行【tarjan,LCA】

正题


题目大意

一个n-1个点的有向无环图,给出若干个点,要求依次到达的最少时间。


解题思路

有向无环图我们可以把其看做一颗树,然后每次用LCA求两个点之间的距离,然后把所有距离统计一下就是结果。


代码

#include<cstdio>
#define min(a,b) a<b?a:b
using namespace std;
struct line{
    int next,to;
}a[60001],num[60001];
int father[30001],n,m,q,p,v[30001],tot,ls[30001],t,x,y,in[30001];
int last,ls1[30001],tot2,nber,sum,w[30001],d[30001],ans[30001];
void addl(int x,int y)//加边
{
    a[++tot].to=y;
    a[tot].next=ls[x];
    ls[x]=tot;
}
void addn(int x,int y)//表示要求这两个点的距离
{
    num[++tot2].to=y;
    num[tot2].next=ls1[x];
    ls1[x]=tot2;
}
int find(int x)//并查集
{
    return x==father[x]?x:father[x]=find(father[x]);
}
void tarjan(int x)
{
    v[x]=1;
    for (int i=ls[x];i;i=a[i].next)
    {
        int y=a[i].to;
        if (v[y]) continue;
        d[y]=d[x]+1;//计算深度
        tarjan(y);//tarjan子节点
        father[y]=x;
    }
    for (int i=ls1[x];i;i=num[i].next)
    {
        int y=num[i].to,id=w[i];
        if (v[y]==2)
          ans[id]=min(ans[id],d[x]+d[y]-d[find(y)]*2);//计算距离
    }
    v[x]=2;
}
int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
      father[i]=i;
    for (int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        addl(x,y);addl(y,x);
    }
    scanf("%d",&m);
    last=1;
    for (int i=1;i<=m;i++)
    {
        scanf("%d",&x);
        ans[i]=1<<30;
        addn(x,last);
        w[tot2]=i;
        addn(last,x);
        w[tot2]=i;
        ans[i]=last==x?0:ans[i];
        last=x;
    }
    tarjan(1);
    for (int i=1;i<=m;i++)
      sum+=ans[i];
    printf("%d",sum);
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值