Problem Description
我们将A省简化为由N个城市组成,某些城市之间存在双向道路,而且A省的交通有一个特点就是任意两个城市之间都能通过道路相互到达,且在不重复经过城市的情况下任意两个城市之间的到达方案都是唯一的。聪明的你一定已经发现,这些城市构成了树这样一个结构。
现在百度陆续开了许许多多的子公司。每家子公司又会在各城市中不断兴建属于该子公司的办公室。
由于各个子公司之间经常有资源的流动,所以公司员工常常想知道,两家子公司间的最小距离。
我们可以把子公司看成一个由办公室组成的集合。那么两个子公司A和B的最小距离定义为min(dist(x,y))(x∈A,y∈B)。其中dist(x,y)表示两个办公室之间的最短路径长度。现在共有Q个询问,每次询问分别在两个子公司间的最小距离。
Input
第一行一个正整数T,表示数据组数。
对于每组数据:
第一行两个正整数N和M。城市编号为1至N,子公司编号为1至M。
接下来N-1行给定所有道路的两端城市编号和道路长度。
接下来M行,依次按编号顺序给出各子公司办公室所在位置,每行第一个整数G,表示办公室数,接下来G个数为办公室所在位置。
接下来一个整数Q,表示询问数。
接下来Q行,每行两个正整数a,b(a不等于b),表示询问的两个子公司。
【数据范围】
0<=边权<=100
1<=N,M,Q,工厂总数<=100000
Output
对于每个询问,输出一行,表示答案。
Sample Input
1
3 3
1 2 1
2 3 1
2 1 1
2 2 3
2 1 3
3
1 2
2 3
1 3
Sample Output
1
0
0
思路
中文题面,题意就不解释了,先利用dis
数组处理出每个节点到根的距离,然后用vector
存一下子公司的信息,然后暴力枚举,刷新ans最小值就可以了,用在线倍增来做
代码
#include <cstdio>
#include <cstring>
#include <cctype>
#include <stdlib.h>
#include <string>
#include <map>
#include <iostream>
#include <stack>
#include <cmath>
#include <queue>
#include <vector>
#include <algorithm>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define mem(a,b) memset(a,b,sizeof(a))
const int N=100000+7;
int n,m,s;
int tot;
int first[N],d[N],p[N][21];
int dis[N];
vector<int>G[N];
struct edge
{
int v,next,w;
} e[2*N];
void add_edge(int u,int v,int w)
{
e[tot].w=w;
e[tot].v=v;
e[tot].next=first[u];
first[u]=tot++;
}
void dfs(int u,int fa,int len)
{
d[u]=d[fa]+1;//dfs处理深度
dis[u]=len;
p[u][0]=fa;//u的第一号祖先(父亲节点)
for(int i=1; (1<<i)<=d[u]; i++)
p[u][i]=p[p[u][i-1]][i-1];
for(int i=first[u]; ~i; i=e[i].next)
{
int v=e[i].v,w=e[i].w;
if(v!=fa)
dfs(v,u,len+w);
}
}
int lca(int a,int b)
{
if(d[a]>d[b])
swap(a,b);//保证a在b点的上方
for(int i=20; i>=0; i--)
if(d[a]<=d[b]-(1<<i))
b=p[b][i]; //把b移到和a同一个深度
if(a==b)
return a;
for(int i=20; i>=0; i--)
{
if(p[a][i]==p[b][i])
continue;
else
a=p[a][i],b=p[b][i];//一起向上跳跃
}
return p[a][0];
}
void init()
{
tot=0;
mem(first,-1);
mem(d,0);
mem(p,0);
mem(G,0);
mem(dis,0);
}
int main()
{
int t,u,v,w,q;
scanf("%d",&t);
while(t--)
{
init();
scanf("%d%d",&n,&m);
for(int i=1; i<n; i++)
{
scanf("%d%d%d",&u,&v,&w);
add_edge(u,v,w);
add_edge(v,u,w);
}
int num,x;
for(int i=1; i<=m; i++)
{
scanf("%d",&num);
while(num--)
{
scanf("%d",&x);
G[i].push_back(x);
}
}
dfs(1,0,0);
scanf("%d",&q);
while(q--)
{
int ans=inf;
scanf("%d%d",&u,&v);
for(int i=0; i<G[u].size(); i++)
for(int j=0; j<G[v].size(); j++)
{
int a=G[u][i];
int b=G[v][j];
ans=min(ans,dis[a]+dis[b]-2*dis[lca(a,b)]);
}
printf("%d\n",ans);
}
}
return 0;
}