算法竞赛进阶指南 聚会 (LCA)

题面

Y岛风景美丽宜人,气候温和,物产丰富。

Y岛上有N个城市(编号1,2,…,N),有N-1条城市间的道路连接着它们。

每一条道路都连接某两个城市。

幸运的是,小可可通过这些道路可以走遍Y岛的所有城市。

神奇的是,乘车经过每条道路所需要的费用都是一样的。

小可可,小卡卡和小YY经常想聚会,每次聚会,他们都会选择一个城市,使得3个人到达这个城市的总费用最小。

由于他们计划中还会有很多次聚会,每次都选择一个地点是很烦人的事情,所以他们决定把这件事情交给你来完成。

他们会提供给你地图以及若干次聚会前他们所处的位置,希望你为他们的每一次聚会选择一个合适的地点。

输入格式
第一行两个正整数,N和M,分别表示城市个数和聚会次数。

后面有N-1行,每行用两个正整数A和B表示编号为A和编号为B的城市之间有一条路。

再后面有M行,每行用三个正整数表示一次聚会的情况:小可可所在的城市编号,小卡卡所在的城市编号以及小YY所在的城市编号。

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

数据范围
N≤500000,M≤500000
输入样例:
6 4
1 2
2 3
2 4
4 5
5 6
4 5 6
6 3 1
2 4 4
6 6 6
输出样例:
5 2
2 5
4 1
6 0

思路

首先耿直的想法就是求三个点的lca直接嗯搞,但是样例直接否定了这种想法,因为如果三点在一条链上,那么三点lca并不是最优的,所以这个时候我们考虑一下贪心,首先让两点汇合,然后让第三个点前往两点汇合所在地,因为这个一定比两点前往第三点要优秀,然后枚举哪两个点线汇合,比较一下大小就可以了。

代码实现

#include<cstdio>
#include<algorithm>
#include<functional>
#include<vector>
#include<numeric>
#include<queue>
#include<map>
#include<set>
#include<iostream>
#include<cstring>
#include<cmath>
using namespace std;
#define rep(i,f_start,f_end) for (int i=f_start;i<=f_end;++i)
#define per(i,n,a) for (int i=n;i>=a;i--)
#define MT(x,i) memset(x,i,sizeof(x) )
#define rev(i,start,end) for (int i=start;i<end;i++)
#define inf 0x3f3f3f3f
#define mp(x,y) make_pair(x,y)
#define lowbit(x) (x&-x)
#define MOD 1000000007
#define exp 1e-8
#define N 1000005 
#define fi first 
#define se second
#define pb push_back
typedef long long ll;
const ll INF=0x3f3f3f3f3f3f3f3f;
typedef vector <int> VI;
typedef pair<int ,int> PII;
typedef pair<int ,PII> PIII;
ll gcd(ll a,ll b){return b==0?a:gcd(b,a%b);}
ll lcm(ll a,ll b){return a/gcd(a,b)*b;};
void check_max (int &a,int b) { a=max (a,b);}
void check_min (int &a,int b) { a=min (a,b);}
inline int read() {
    char ch=getchar(); int x=0, f=1;
    while(ch<'0'||ch>'9') {
        if(ch=='-') f=-1;
        ch=getchar();
    } while('0'<=ch&&ch<='9') {
        x=x*10+ch-'0';
        ch=getchar();
    } return x*f;
}
const int maxn=5e5+100;
int n,m,s;
int deep[maxn];
struct Lca {
    int head[maxn],cnt=0;
    int fa[maxn][22];
    struct node {
        int v,next;
    }e[maxn<<1];
    void init () {
        MT (head,0);
        cnt=0;
    }
    void add_edge (int u,int v) {
        e[++cnt].v=v;
        e[cnt].next=head[u];
        head[u]=cnt;
    }
    void dfs (int x,int y) {
        deep[x]=deep[y]+1;
        fa[x][0]=y;
        for (int i=1;(1<<i)<=deep[x];i++) fa[x][i]=fa[fa[x][i-1]][i-1];
        for (int i=head[x];i;i=e[i].next) if (e[i].v!=y) dfs (e[i].v,x);
    }
    int LCA (int x,int y) {
        if (deep[x]<deep[y]) swap (x,y);
        while (deep[x]>deep[y]) x=fa[x][__lg(deep[x]-deep[y])];
        if (x==y) return x;
        for (int k=__lg (deep[x]);k>=0;k--) if (fa[x][k]!=fa[y][k]) {
            x=fa[x][k];
            y=fa[y][k];
        }
        return fa[x][0];
    }
}g1;

int main () {
    scanf ("%d%d",&n,&m);
    g1.init ();
    for (int i=1;i<n;i++) {
        int x,y;
        scanf ("%d%d",&x,&y);
        g1.add_edge (x,y);
        g1.add_edge (y,x);
    }
    g1.dfs (1,0);
    for (int i=1;i<=m;i++) {
        int x,y,z,c_x,c_y,c_z,dx,dy,dz;
        scanf ("%d%d%d",&x,&y,&z);
        c_x=g1.LCA(x,y),dx=deep[x]+deep[y]-deep[c_x]+deep[z]-2*deep[g1.LCA (z,c_x)];
        c_y=g1.LCA(y,z),dy=deep[y]+deep[z]-deep[c_y]+deep[x]-2*deep[g1.LCA(x,c_y)];
		c_z=g1.LCA(x,z),dz=deep[x]+deep[z]-deep[c_z]+deep[y]-2*deep[g1.LCA(y,c_z)];
        if (dx>dy) dx=dy,c_x=c_y;
        if (dx>dz) dx=dz,c_x=c_z;
        printf ("%d %d\n",c_x,dx);
    }
}
posted @ 2020-09-29 16:39  Luglucky  阅读(194)  评论(0编辑  收藏  举报