CodeVS 2370 LCA 解题报告

2370 小机房的树

题目描述 Description

小机房有棵焕狗种的树，树上有N个节点，节点标号为0到N-1，有两只虫子名叫飘狗和大吉狗，分居在两个不同的节点上。有一天，他们想爬到一个节点上去搞基，但是作为两只虫子，他们不想花费太多精力。已知从某个节点爬到其父亲节点要花费 c 的能量（从父亲节点爬到此节点也相同），他们想找出一条花费精力最短的路，以使得搞基的时候精力旺盛，他们找到你要你设计一个程序来找到这条路，要求你告诉他们最少需要花费多少精力

输入描述 Input Description

第一行一个n，接下来n-1行每一行有三个整数u，v, c 。表示节点 u 爬到节点 v 需要花费 c 的精力。
第n+1行有一个整数m表示有m次询问。接下来m行每一行有两个整数 u ，v 表示两只虫子所在的节点
输出描述 Output Description
一共有m行，每一行一个整数，表示对于该次询问所得出的最短距离。

样例输入 Sample Input

3
1 0 1
2 0 1
3
1 0
2 0
1 2

样例输出 Sample Output

1
1
2

数据范围及提示 Data Size & Hint

1<=n<=50000， 1<=m<=75000， 0<=c<=1000

【解题报告】
倍增求LCA裸题。
莫名写挂，调了半天。。。。。

代码如下：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std; 
#define N 50010
#define M 75010

int n,m,cnt=0;
int dis[N],deep[N],head[M],fa[N][30];
struct Edge
{
    int to,nxt,w;
}e[M<<1];

void adde(int u,int v,int w)
{
    e[++cnt].w=w;
    e[cnt].to=v;
    e[cnt].nxt=head[u];
    head[u]=cnt;
}
void dfs(int u,int x)
{
    deep[u]=x;
    for(int i=head[u];~i;i=e[i].nxt)
    {
        int v=e[i].to;
        if(!deep[v]&&v)
        {
            fa[v][0]=u;
            dis[v]=dis[u]+e[i].w;
            dfs(v,x+1);
        }
    }
}
void getfa()
{
    for(int i=0;i<n;++i)
    for(int j=1;j<=22;++j)
    fa[i][j]=fa[fa[i][j-1]][j-1];
}
int get_same(int u,int v)
{
    for(int i=0;i<=22;i++)
    if((1<<i)&v) u=fa[u][i];
    return u;
}
int lca(int u,int v)
{
    if(deep[u]<deep[v]) swap(u,v);
    u=get_same(u,deep[u]-deep[v]);
    for(int i=22;i>=0;i--)
    if(fa[u][i]!=fa[v][i])
    {
        u=fa[u][i];
        v=fa[v][i];
    }
    if(u==v) return u;
    return fa[u][0];
}
int main()
{
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    for(int i=1;i<n;++i)
    {
        int u,v,w;scanf("%d%d%d",&u,&v,&w);
        adde(u,v,w);
        adde(v,u,w);
    }
    dfs(0,0);
    getfa();
    scanf("%d",&m);
    for(int i=1;i<=m;++i)
    {
        int u,v;scanf("%d%d",&u,&v);
        int ans=lca(u,v);
        if(ans==u) printf("%d\n",dis[v]-dis[u]);
        else printf("%d\n",dis[u]+dis[v]-2*dis[ans]);
    }
    return 0;
}

板子：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 50010

int t,n,m,cnt=-1,head[N];
struct Edge{int to,nxt,w;}e[N<<1];
int dis[N],dep[N],fa[N],anc[N][22];

void adde(int u,int v,int w)
{
    e[++cnt].to=v;e[cnt].w=w;
    e[cnt].nxt=head[u];head[u]=cnt;
    e[++cnt].to=u;e[cnt].w=w;
    e[cnt].nxt=head[v];head[v]=cnt;
}
void dfs(int u)
{
    anc[u][0]=fa[u];
    for(int i=1;i<=20;++i) anc[u][i]=anc[anc[u][i-1]][i-1];
    for(int i=head[u];~i;i=e[i].nxt)
    {
        int v=e[i].to;
        if(v==fa[u]) continue;
        fa[v]=u;
        dep[v]=dep[u]+1;
        dis[v]=dis[u]+e[i].w;
        dfs(v);
    }
}
int lca(int u,int v)
{
    if(dep[u]<dep[v]) swap(u,v);
    for(int i=20;i>=0;--i)
    {
        if(dep[anc[u][i]]>=dep[v])
        {
            u=anc[u][i];
        }
    }
    if(u==v) return u;
    for(int i=20;i>=0;--i)
    {
        if(anc[u][i]!=anc[v][i])
        {
            u=anc[u][i];
            v=anc[v][i];
        }
    }
    return anc[u][0];
}
int main()
{
    for(scanf("%d",&t);t;--t)
    {
        cnt=-1;
        memset(head,-1,sizeof(head));
        memset(fa,0,sizeof(fa));
        memset(anc,0,sizeof(anc));
        memset(dep,0,sizeof(dep));
        memset(dis,0,sizeof(dis));
        scanf("%d%d",&n,&m);
        for(int i=1;i<n;++i)
        {
            int u,v,w;scanf("%d%d%d",&u,&v,&w);
            adde(u,v,w);
        }
        dfs(1);
//      for(int i=1;i<=n;++i) printf("%d ",dep[i]);puts("");
//      for(int i=1;i<=n;++i) printf("%d ",dis[i]);puts("");
        for(int i=1;i<=m;++i)
        {
            int u,v;scanf("%d%d",&u,&v);
            printf("%d\n",dis[u]+dis[v]-2*dis[lca(u,v)]);
        }
    }   
    return 0;
}
/*
1
5 1
1 2 10
1 3 10
3 4 10
3 5 10
4 2
*/