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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章