2017"百度之星"程序设计大赛 - 初赛（B）(HDU6115) 1002 Factory（LCA）

最新推荐文章于 2020-07-20 10:09:46 发布

原创最新推荐文章于 2020-07-20 10:09:46 发布 · 1.0k 阅读

0 ·

CC 4.0 BY-SA版权

LCA 专栏收录该内容

1 篇文章

订阅专栏

我们将A省简化为由N个城市组成，某些城市之间存在双向道路，而且A省的交通有一个特点就是任意两个城市之间都能通过道路相互到达，且在不重复经过城市的情况下任意两个城市之间的到达方案都是唯一的。聪明的你一定已经发现，这些城市构成了树这样一个结构。

现在百度陆续开了许许多多的子公司。每家子公司又会在各城市中不断兴建属于该子公司的办公室。

由于各个子公司之间经常有资源的流动，所以公司员工常常想知道，两家子公司间的最小距离。我们可以把子公司看成一个由办公室组成的集合。那么两个子公司A和B的最小距离定义为min(dist(x，y))（x∈A，y∈B）。其中dist(x，y)表示两个办公室之间的最短路径长度。

现在共有Q个询问，每次询问分别在两个子公司间的最小距离。

第一行一个正整数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

中文题，题意不说了。这题应该是数据弱，直接大力LCA就行。。。用lca求出两个集合中距离最小的两个点的距离，就是答案。

#include<cmath>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define LL long long

using namespace std;

const int inf = 1e9;

struct edge
{
    int to,cost;
    edge(int _to,int _cost)
    {
        to = _to;
        cost = _cost;
    }
};
vector<edge> e[100010];
int n,m;
int fa[100010][20];
int deep[100010];
int dis[100010];

void dfs(int x,int pre,int d,int len)
{
    fa[x][0] = pre;
    deep[x] = d;
    dis[x] = len;
    for(int i=0;i<e[x].size();i++)
    {
        int xx = e[x][i].to;
        if(xx == pre)
            continue;
        dfs(xx,x,d+1,len+e[x][i].cost);
    }
}
void init()
{
    memset(deep,-1,sizeof(deep));
    memset(dis,-1,sizeof(dis));
    dfs(1,0,0,0);
    for(int j=1;(1<<j)<=n;j++)
        for(int i=1;i<=n;i++)
        {
            fa[i][j]=fa[fa[i][j-1]][j-1];
        }
}
int lca(int a,int b)
{
    if(deep[a]>deep[b])
        swap(a,b);
    int f=deep[b]-deep[a];
    for(int i=0;(1<<i)<=f;i++)
    {
        if((1<<i)&f)
            b=fa[b][i];
    }
    if(a!=b)
    {
        for(int i=(int)log2(n);i>=0;i--)
        {
            if(fa[a][i]!=fa[b][i])
            {
                a=fa[a][i];
                b=fa[b][i];
            }
        }
        a=fa[a][0];
    }
    return a;
}

int dist(int a,int b)
{
    int k = lca(a,b);
    return dis[a] + dis[b] - 2*dis[k];
}

vector<int> v[100010];

int main(void)
{
    int T,i,j;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(i=1;i<=n;i++)
            e[i].clear();
        for(i=1;i<=m;i++)
            v[i].clear();
        for(i=1;i<n;i++)
        {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            e[x].push_back(edge(y,z));
            e[y].push_back(edge(x,z));
        }
        for(i=1;i<=m;i++)
        {
            int k;
            int x;
            scanf("%d",&k);
            for(j=0;j<k;j++)
            {
                scanf("%d",&x);
                v[i].push_back(x);
            }
        }
        init();
        int q;
        scanf("%d",&q);
        while(q--)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            int minn = inf;
            for(i=0;i<v[a].size();i++)
                for(j=0;j<v[b].size();j++)
                {
                    minn = min(minn,dist(v[a][i],v[b][j]));
                }
            printf("%d\n",minn);
        }
    }

    return 0;
}