关闭

wikioi 1036 商务旅行

444人阅读 评论(0) 收藏 举报

http://wikioi.com/problem/1036/

由于n-1边连接n个点,所以不存在路径选择的问题,只能有一条路

而找2个点的距离就找到公共祖先的距离之和

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#include<cstring>
using namespace std;
int n,m;
struct edge
{
	int v,next,w;
}e[30000*2+10];
int k=1;
int head[30000+10];
void add(int u,int v)
{
	e[k].v=v;
	e[k].w=1;
	e[k].next=head[u];
	head[u]=k++;
}
int vis[30000+10]={0};
int dep[30000+10];
int fa[30000+10]={0};
int s=0;
void dfs(int f,int x)//fa节点,x当前节点
{
	if(vis[x])return;
	dep[x]=dep[f]+1;
	s=max(dep[x],s);
	vis[x]=1;
	for(int i=head[x];i!=-1;i=e[i].next)
	{	
		int v=e[i].v;
		if(!vis[v])
		{
			fa[v]=x;
			dfs(x,v);
		}
	}
}//树的形态是自己确定的
int lca(int u,int v)
{
	int nu=u,nv=v;
	if(dep[v]<dep[u])swap(u,v);//深度不同要走到相同深度
	if(dep[v]>dep[u])
	{
		while(dep[v]>dep[u])//这个不是正规的,应该多开一维记录第2^i个祖先
		{
			v=fa[v];
		}
	}
	while(u!=v)//同时走
	{
		u=fa[u];
		v=fa[v];
	}
	int final=v;
	return dep[nu]+dep[nv]-2*dep[final];
}
int main()
{
	memset(head,-1,sizeof(head));
	cin>>n;
	for(int i=1;i<n;i++)
	{
		int a,b;
		scanf("%d%d",&a,&b);
		add(a,b);
		add(b,a);
	}
	dep[0]=0;
	fa[1]=1;
	dfs(0,1);//建树
	cin>>m;
	int last=1;	
	int ans=0;
	for(int i=1;i<=m;i++)
	{
		int a;
		scanf("%d",&a);
		ans+=lca(last,a);
		last=a;
	}
	cout<<ans<<endl;
}


0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:18146次
    • 积分:585
    • 等级:
    • 排名:千里之外
    • 原创:42篇
    • 转载:0篇
    • 译文:0篇
    • 评论:1条
    文章存档
    最新评论