聚会 - 走看看

题目描述

Y岛风景美丽宜人，气候温和，物产丰富。Y岛上有N个城市，有N-1条城市间的道路连接着它们。每一条道路都连接某两个城市。幸运的是，小可可通过这些道路可以走遍Y岛的所有城市。神奇的是，乘车经过每条道路所需要的费用都是一样的。小可可，小卡卡和小YY经常想聚会，每次聚会，他们都会选择一个城市，使得3个人到达这个城市的总费用最小。由于他们计划中还会有很多次聚会，每次都选择一个地点是很烦人的事情，所以他们决定把这件事情交给你来完成。他们会提供给你地图以及若干次聚会前他们所处的位置，希望你为他们的每一次聚会选择一个合适的地点。

输入

第一行两个正整数，N和M。分别表示城市个数和聚会次数。后面有N-1行，每行用两个正整数A和B表示编号为A和编号为B的城市之间有一条路。城市的编号是从1到N的。再后面有M行，每行用三个正整数表示一次聚会的情况：小可可所在的城市编号，小卡卡所在的城市编号以及小YY所在的城市编号。

输出

一共有M行，每行两个数Pos和Cost，用一个空格隔开。表示第i次聚会的地点选择在编号为Pos的城市，总共的费用是经过Cost条道路所花费的费用。

样例输入

样例输出

提示

100%的数据中，N<=500000，M<=500000。
40%的数据中N<=2000，M<=2000。

LCA

最近的点，一定是在两个点的LCA处，枚举一下，取最小值就可以啦~

#include <bits/stdc++.h>
#define maxn 500055
using namespace std;
struct Edge
{
    int v,next;
};
int n,m,t;
struct Map
{
    int head[maxn];
    Edge edge[maxn<<1];
    int cnt;
    int fa[maxn][25];
    bool vis[maxn];
    int depth[maxn];
    inline void init()
    {
        memset(head,-1,sizeof(head));
        cnt=0;
    }
    inline void addedge(int u,int v)
    {
        edge[cnt].v=v;
        edge[cnt].next=head[u];
        head[u]=cnt++;
    }
    inline void bfs()
    {
        queue<int> q;
        memset(vis,false,sizeof(vis));
        memset(depth,0,sizeof(depth));
        vis[1]=true;
        q.push(1);
        depth[1]=1;
        while(!q.empty())
        {
            int u=q.front();
            q.pop();
            for(int i=head[u]; i!=-1; i=edge[i].next)
            {
                int v=edge[i].v;
                if(!vis[v])
                {
                    vis[v]=true;
                    depth[v]=depth[u]+1;
                    fa[v][0]=u;
                    for(int j=1; j<=t; j++)
                    {
                        fa[v][j]=fa[fa[v][j-1]][j-1];
                    }
                    q.push(v);
                }
            }
        }
    }
    inline int lca(int x,int y)
    {
        if(depth[x]>depth[y]) swap(x,y);
        for(int i=t;i>=0;--i)
        {
            if(depth[fa[y][i]]>=depth[x]) y=fa[y][i];
        }
        if(x==y) return x;
        for(int i=t;i>=0;--i)
        {
            if(fa[x][i]!=fa[y][i])
            {
                x=fa[x][i];
                y=fa[y][i];
            }
        }
        return fa[x][0];
    }
    inline int reads()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
    void read()
    {
        scanf("%d%d",&n,&m);
        init();
        t=(int)(log(n)/log(2))+1;
        for(int i=1; i<=n-1; ++i)
        {
            int u,v;
            u=reads();
            v=reads();
            addedge(u,v);
            addedge(v,u);
        }
        bfs();
        for(int i=1;i<=m;++i)
        {
        int x,y,z;
        x=reads();
        y=reads();
        z=reads();
        ask(x,y,z);
        }
    }
    inline void ask(int x,int y,int z)
    {
        int num1=lca(x,y);
        int num2=lca(x,z);
        int num3=lca(y,z);
        int num4=lca(z,num1);
        int ans=depth[x]+depth[y]+depth[z]-3*depth[num4];
        int pos=num4;
        int temp=depth[x]+depth[y]-2*depth[num1]+depth[z]+depth[num1]-2*depth[num4];
        if(ans>temp)
        {
            ans=temp;
            pos=num1;
        }
        temp=depth[x]+depth[z]-2*depth[num2]+depth[y]+depth[num2]-2*depth[num4];
        if(ans>temp)
        {
            ans=temp;
            pos=num2;
        }
        temp=depth[y]+depth[z]-2*depth[num3]+depth[x]+depth[num3]-2*depth[num4];
        if(ans>temp)
        {
            ans=temp;
            pos=num3;
        }
        printf("%d %d
",pos,ans);
    }
 
} Mp;
int main()
{
    Mp.read();
    //printf("%d
",Mp.lca(2,3));
    return 0;
}