【BZOJ1832】聚会

最新推荐文章于 2023-05-27 15:59:22 发布

sszxzzh

最新推荐文章于 2023-05-27 15:59:22 发布

阅读量569

点赞数 1

分类专栏：最近公共祖先(LCA BZOJ 文章标签： BZOJ LCA

本文链接：https://blog.csdn.net/m0_38083668/article/details/81301942

版权

BZOJ 同时被 2 个专栏收录

104 篇文章 0 订阅

订阅专栏

最近公共祖先(LCA

4 篇文章 0 订阅

订阅专栏

1832: [AHOI2008]聚会

Time Limit: 10 Sec Memory Limit: 64 MB
Submit: 1826 Solved: 780

Description

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

Input

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

Output

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

Sample Input

6 4
1 2
2 3
2 4
4 5
5 6
4 5 6
6 3 1
2 4 4
6 6 6

Sample Output

5 2
2 5
4 1
6 0

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

解析：

树上涉及到距离最短的问题多少跟LCA都有点关系，所以直接找三个点两两的LCA更新答案就行了。

代码：

#include <bits/stdc++.h>
using namespace std;
 
const int Max=500010;
int n,m,size,s,ans;
int depth[Max],first[Max],f[Max][20],vis[Max];
struct shu{int to,next;};
shu edge[Max<<1];
 
inline int get_int()
{
    int x=0,f=1;
    char c;
    for(c=getchar();(!isdigit(c))&&(c!='-');c=getchar());
    if(c=='-') {f=-1;c=getchar();}
    for(;isdigit(c);c=getchar()) x=(x<<3)+(x<<1)+c-'0';
    return x*f;
}
 
inline void build(int x,int y)
{
    edge[++s].next=first[x];
    first[x]=s;
    edge[s].to=y;
}
 
inline void pre()
{
    queue<int>q;
    q.push(1);
    while(!q.empty())
    {
      int point = q.front();
      vis[point] = 1;
      q.pop();
      for(int u=first[point];u;u=edge[u].next)
      {
        int to=edge[u].to;
        if(vis[to]) continue;
        depth[to] = depth[point] + 1;
        f[to][0] = point;
        q.push(to);
      }
    }
}
 
inline int LCA(int x,int y)
{
    if(depth[x] < depth[y]) swap(x,y);
    int len = depth[x] - depth[y];
    for(int i=16;i>=0;i--)
      if(len >= 1<<i) len-=1<<i,x=f[x][i];
    if(x==y) return x;
    for(int i=16;i>=0;i--)
      if(f[x][i] != f[y][i]) x = f[x][i],y = f[y][i];
    return f[x][0];
}
 
inline int mn(int x,int y){return x > y ? y : x;}
 
int main()
{
    n=get_int(),m=get_int();
    for(int i=1;i<=n-1;i++)
    {
      int x=get_int(),y=get_int();
      build(x,y),build(y,x);
    }
 
    pre();
    for(int i=1;i<=16;i++)
      for(int j=1;j<=n;j++)
        if(f[j][i-1]) f[j][i] = f[f[j][i-1]][i-1];
 
    for(int i=1;i<=m;i++)
    {
      int x=get_int(),y=get_int(),z=get_int();
      if(x== y && y==z) cout<<x<<" 0\n";
      int fa1 = LCA(x,y),fa2 = LCA(x,z),fa3 = LCA(y,z);
      int len1 = depth[x] + depth[y] - 2*depth[fa1] + depth[fa1] + depth[z] - 2*depth[LCA(fa1,z)];
      int len2 = depth[x] + depth[z] - 2*depth[fa2] + depth[fa2] + depth[y] - 2*depth[LCA(fa2,y)];
      int len3 = depth[z] + depth[y] - 2*depth[fa3] + depth[fa3] + depth[x] - 2*depth[LCA(fa3,x)];
      int minn = mn(len1,mn(len2,len3));
      if(minn == len1) cout<<fa1<<" "<<len1<<"\n";
      else if(minn == len2) cout<<fa2<<" "<<len2<<"\n";
      else cout<<fa3<<" "<<len3<<"\n";
    }
 
    return 0;
}

2019.11.9更新

以前写的实在看不下去了。。。

代码：

#include <bits/stdc++.h>
using namespace std;

const int Max=500005;
int n,m,tot,ans,id;
int dep[Max],fa[Max],son[Max],size[Max],vis[Max],top[Max],first[Max];
struct shu{int to,next;}edge[Max<<1];
 
inline int get_int()
{
    int x=0,f=1;
    char c;
    for(c=getchar();(!isdigit(c))&&(c!='-');c=getchar());
    if(c=='-') f=-1,c=getchar();
    for(;isdigit(c);c=getchar()) x=(x<<3)+(x<<1)+c-'0';
    return x*f;
}
inline void print(int x)
{
	if(x>9) print(x/10);
	putchar('0'+x%10);
}

inline void build(int x,int y){edge[++tot].next=first[x],first[x]=tot,edge[tot].to=y;}
inline void dfs1(int p)
{
	size[p]=vis[p]=1;
	for(int u=first[p];u;u=edge[u].next)
	{
	  int to=edge[u].to;
	  if(vis[to]) continue;
	  fa[to]=p,dep[to]=dep[p]+1,dfs1(to),size[p]+=size[to];
	  if(size[son[p]]<size[to]) son[p]=to;
	}
}
inline void dfs2(int p,int tp)
{
	top[p]=tp;
	if(!son[p]) return;
	dfs2(son[p],tp);
	for(int u=first[p];u;u=edge[u].next)
	{
	  int to=edge[u].to;
	  if(to==fa[p]||to==son[p]) continue;
	  dfs2(to,to);
	}
}
inline int LCA(int x,int y)
{
	while(top[x]!=top[y])
	{
	  if(dep[top[x]]<dep[top[y]]) swap(x,y);
	  x=fa[top[x]];
	}
	return dep[x]<dep[y] ? x : y;
}
inline int calc(int a,int b,int c,int d){return dep[a]+dep[b]-2*dep[d]+dep[d]+dep[c]-2*dep[LCA(c,d)];}
inline void comp(int a,int b,int c)
{
	int f1=LCA(a,b),f2=LCA(a,c),f3=LCA(b,c),l1,l2,l3;
	l1=calc(a,b,c,f1),l2=calc(a,c,b,f2),l3=calc(b,c,a,f3);
	ans=min(calc(a,b,c,f1),min(calc(a,c,b,f2),calc(b,c,a,f3)));
	if(ans==l1) id=f1;
	else if(ans==l2) id=f2;
	else id=f3;
}
inline void init()
{
	n=get_int(),m=get_int();
	for(int i=1;i<n;i++)
	{
	  int x=get_int(),y=get_int();
	  build(x,y),build(y,x),fa[i]=i;
	}
	dfs1(1);
	dfs2(1,1);
}
inline void solve()
{
	while(m--)
	{
	  int x=get_int(),y=get_int(),z=get_int();
	  if(x==y&&y==z) {print(x),puts(" 0");continue;}
	  comp(x,y,z);
	  print(id),putchar(' '),print(ans),putchar('\n');
	}
}

int main()
{
	init();
	solve();
    return 0;
}