1036 商务旅行(LCA例题)

http://codevs.cn/problem/1036/
题目描述 Description
某首都城市的商人要经常到各城镇去做生意,他们按自己的路线去做,目的是为了更好的节约时间。

假设有N个城镇,首都编号为1,商人从首都出发,其他各城镇之间都有道路连接,任意两个城镇之间如果有直连道路,在他们之间行驶需要花费单位时间。该国公路网络发达,从首都出发能到达任意一个城镇,并且公路网络不会存在环。

你的任务是帮助该商人计算一下他的最短旅行时间。

输入描述 Input Description
输入文件中的第一行有一个整数N,1<=n<=30 000,为城镇的数目。下面N-1行,每行由两个整数a 和b (1<=a, b<=n; a<>b)组成,表示城镇a和城镇b有公路连接。在第N+1行为一个整数M,下面的M行,每行有该商人需要顺次经过的各城镇编号。

输出描述 Output Description
在输出文件中输出该商人旅行的最短时间。

样例输入 Sample Input
5
1 2
1 5
3 5
4 5
4
1
3
2
5
样例输出 Sample Output
7

#include 
#include 
#include 
#include 
#include 
using namespace std;
const int maxn=30000+10;
int pa[maxn],dist[maxn],a[maxn];
vector<int>tree[maxn];
int n,m;
int dfs(int now,int deep)
{
    dist[now]=deep;
    for(int i=0;iif(dist[tree[now][i]]<0){
            dfs(tree[now][i],deep+1);
        }
    }
}
int camp(int u,int v)
{
    while(dist[u]>dist[v]){
        for(int i=0;iif(dist[tree[u][i]]break;
            }
        }
    }
    return u;
}
int LCA(int u,int v)
{
    int tu=u,tv=v;
    if(dist[u]>dist[v])tu=camp(u,v);
    if(dist[u]while(tu!=tv){
        if(tu!=1)
        for(int i=0;iif(dist[tree[tu][i]]break;
            }
        }
        if(tv!=1)
        for(int i=0;iif(dist[tree[tv][i]]break;
            }
        }
    }
    return dist[u]+dist[v]-2*dist[tu];
}
int main()
{
    scanf("%d",&n);
    int u,v;
    for(int i=0;i1;i++){
        scanf("%d%d",&u,&v);
        tree[u].push_back(v);
        tree[v].push_back(u);
    }
    scanf("%d",&m);
    for(int i=0;iscanf("%d",&a[i]);
    }
    memset(dist,-1,sizeof(dist));
    dfs(1,0);
    int ans=LCA(1,a[0]);
    for(int i=0;i1;i++)
        ans+=LCA(a[i],a[i+1]);
    printf("%d\n",ans);
    return 0;
}

你可能感兴趣的:(C++,图论)