HDU 6115 Factory（在线倍增LCA）

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\le$ 边权$\le 100$

$1\le N,M,Q,$工厂总数$\le 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

Solution

对于一组查询$A$和$B$，暴力枚举$x\in A$和$y\in B$，求出$dis(x)+dis(y)-2\cdot dis(lca(x,y))$的最小值即可，其中$dis(x)$表示$x$到根节点的距离，$LCA$在线倍增求即可

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
#define maxn 100005
struct edge
{
    int v,w,next;
}g[2*maxn];
int tot,head[maxn];
void init()
{
    tot=0;
    memset(head,-1,sizeof(head));
}
void add(int u,int v,int w)
{
    g[tot].v=v,g[tot].w=w,g[tot].next=head[u],head[u]=tot++;
}
int Fa[maxn][21],Deep[maxn],Dis[maxn];
void dfs(int u,int fa)
{
    Fa[u][0]=fa;
    for(int i=1;i<=20;i++)Fa[u][i]=Fa[Fa[u][i-1]][i-1];
    for(int i=head[u];~i;i=g[i].next)
    {
        int v=g[i].v,w=g[i].w;
        if(v==fa)continue;
        Dis[v]=Dis[u]+w,Deep[v]=Deep[u]+1;
        dfs(v,u);
    }
}
int Lca(int u,int v)
{
    if(Deep[u]<Deep[v])swap(u,v);
    for(int i=20;i>=0;i--)
        if(Deep[Fa[u][i]]>=Deep[v])
            u=Fa[u][i];
    if(u==v)return u;
    for(int i=20;i>=0;i--)
        if(Fa[u][i]!=Fa[v][i])
            u=Fa[u][i],v=Fa[v][i];
    return Fa[u][0];
}
int get_dis(int u,int v)
{
    int lca=Lca(u,v);
    return Dis[u]+Dis[v]-2*Dis[lca];
}
int T,n,m,q;
vector<int>v[maxn];
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        init();
        for(int i=1;i<n;i++)
        {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w),add(v,u,w);
        }
        dfs(1,1);
        for(int i=1;i<=m;i++)
        {
            v[i].clear();
            int num,temp;
            scanf("%d",&num);
            while(num--)
            {
                scanf("%d",&temp);
                v[i].push_back(temp);
            }
        }
        scanf("%d",&q);
        while(q--)
        {
            int x,y,ans;
            scanf("%d%d",&x,&y);
            ans=get_dis(v[x][0],v[y][0]);
            for(int i=0;i<v[x].size();i++)
            {
                if(!ans)break;
                for(int j=0;j<v[y].size();j++)
                {
                    ans=min(ans,get_dis(v[x][i],v[y][j]));
                    if(!ans)break;
                }
            }
            printf("%d\n",ans); 
        }
    }
    return 0;
}