hdu2586How far away 最近公共祖先Lca tarjan算法

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2586
第一次用tarjan算法,留个模板。

#include<stdio.h>
#include<string.h>
#define spot 40010
#define edge 80010
#define query 410
struct stu
{
    int to,value,next;
} a[edge];
int head[spot],x[query],y[query],z[query],sizes,dis[spot],fa[spot],n,m;  //fa祖节点
bool vis[spot]; 
void init()
{
    sizes=0;
    memset(head,-1,sizeof(head));
    memset(vis,0,sizeof(vis));
    memset(dis,0,sizeof(dis));
    for(int i=1; i<=n; i++)
        fa[i]=i;
}
void add_edge(int from,int to,int value)
{
    a[sizes].to=to;
    a[sizes].value=value;
    a[sizes].next=head[from];
    head[from]=sizes++;
}
int finds(int x)
{
    while(x!=fa[x])
        return fa[x]=finds(fa[x]);
    return x;
}
void tarjan(int u)
{
    int k,v,i;
    vis[u]=1;
    for(i=1; i<=m; i++)   
    {
        if(u==x[i]&&vis[y[i]]) z[i]=fa[y[i]];
        if(u==y[i]&&vis[x[i]]) z[i]=fa[x[i]];
    }                         
    for(k=head[u]; k+1; k=a[k].next)
    {
        v=a[k].to;
        if(!vis[v])
        {
            dis[v]=dis[u]+a[k].value;
            tarjan(v);
            fa[v]=u;
        }
    }
}
int main()
{
    int nn;
    scanf("%d",&nn);
    while(nn--)
    {
        int from,to,value,i;
        scanf("%d%d",&n,&m);
        init();
        for(i=1; i<n; i++)
            scanf("%d%d%d",&from,&to,&value),add_edge(from,to,value),add_edge(to,from,value);
        for(i=1; i<=m; i++)
            scanf("%d%d",&x[i],&y[i]);
        tarjan(1);
        for(i=1; i<=m; i++)
            printf("%d\n",dis[x[i]]+dis[y[i]]-2*dis[z[i]]);
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值