洛谷P4281 [AHOI2008]紧急集合 / 聚会 树剖+LCA


洛谷P4281 [AHOI2008]紧急集合 / 聚会


标签

  • 树链剖分
  • LCA

简明题意

  • 给一棵树,有多组询问,每组询问会给你3个点,让你求树上某一点x,使得x到所给三个点的距离总和最短

思路

  • 容易发现,3个点相互之间的LCA中,取深度最深的那个,就是要集结的点,从这一点出发到三个点的距离之和最短

注意事项

  • 求树上两点u、v的距离时,如果用深度求,需要分类讨论,讨论他们是否再同一条链上,不在同一条链上时还需要求他们的LCA

总结


AC代码

#include<cstdio>
#include<vector>
#include<algorithm>
#include<cmath>
using namespace std;

const int maxn = 5e5 + 10;

int n, m;
vector<int> g[maxn];

int dep[maxn], siz[maxn], son[maxn], fa[maxn];
void dfs1(int u, int f, int deep)
{
   fa[u] = f;
   dep[u] = deep;
   siz[u] = 1;

   int max_son = -1;
   for (auto& v : g[u])
   	if (v != f)
   	{
   		dfs1(v, u, deep + 1);
   		siz[u] += siz[v];
   		if (siz[v] > max_son)
   			max_son = siz[v], son[u] = v;
   	}
}

int id[maxn], cnt, top[maxn];
void dfs2(int u, int topf)
{
   id[u] = ++cnt;
   top[u] = topf;

   if (son[u])
   {
   	dfs2(son[u], topf);
   	for (auto& v : g[u])
   		if (v != fa[u] && v != son[u])
   			dfs2(v, v);
   }
}

int LCA(int u, int v)
{
   while (top[u] != top[v])
   {
   	if (dep[top[v]] < dep[top[u]]) swap(u, v);
   	v = fa[top[v]];
   }
   return dep[u] < dep[v] ? u : v;
}

int ask_path(int u, int v)
{
   int cnt = 0;
   while (top[u] != top[v])
   {
   	if (dep[top[v]] < dep[top[u]]) swap(u, v);
   	cnt += id[v] - id[top[v]] + 1;
   	v = fa[top[v]];
   }
   return cnt + abs(id[u] - id[v]);
}

int cal(int o, int x, int y, int z)
{
   return ask_path(o, x) + ask_path(o, y) + ask_path(o, z);
}

void solve()
{
   scanf("%d%d", &n, &m);
   for (int i = 1; i < n; i++)
   {
   	int u, v;
   	scanf("%d%d", &u, &v);

   	g[u].push_back(v), g[v].push_back(u);
   }

   dfs1(1, 1, 1);
   dfs2(1, 1);

   while (m--)
   {
   	int x, y, z;
   	scanf("%d%d%d", &x, &y, &z);

   	int k = LCA(x, y), t;
   	if (dep[k] < dep[t = LCA(y, z)]) k = t;
   	if (dep[k] < dep[t = LCA(x, z)]) k = t;

   	printf("%d %d\n", k, cal(k, x, y, z));
   }
}

int main()
{
   freopen("Testin.txt", "r", stdin);
   solve();
   return 0;
}

双倍经验

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值