ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

仓鼠找 sugar

2022-05-25 22:01:58  阅读:175  来源: 互联网

标签:include dist int sugar fa depth lca 仓鼠


仓鼠找 sugar

LCA

SCUACM2022集训前训练-图论 - Virtual Judge (vjudge.net)

  1. 首先要观察出一个结论:若 a - b 的路径与 c - d 的路径相交,设 a, b 的 LCA 为 x; c, d 的 LCA 为 y

    则有 x 在 c - d 路径上 或 y 在 a - b 路径上

  2. 如何判断一个点是否在一条路径上:

    可观察到一个性质:若 x 在 a,b 路径上,则 \(dist(a,x) + dist(b,x) == dist(a,b)\)

  3. 求 \(dist(a,b)\): 设 \(t=lca(a,b)\), \(dist(a,b)=depth[a]-depth[t]+depth[b]-depth[t]\)

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;
typedef long long ll;
const int N = 1e5 + 10, M = 21;
int n, m, Q;
int fa[N][M], depth[N];
vector<vector<int> > G(N);
queue<int> q;
void add(int a, int b)
{
	G[a].push_back(b);
}

void bfs(int root)
{
	q.push(root);
	memset(depth, 0x3f, sizeof depth);
	depth[0] = 0;
	depth[root] = 1;
	while(!q.empty())
	{
		int t = q.front();
		q.pop();
		for (auto to : G[t])
		{
			if (depth[to] != depth[t] - 1)
			{
				depth[to] = depth[t] + 1;
				fa[to][0] = t;
				q.push(to);
				for (int k = 1; k < M; k++)
					fa[to][k] = fa[fa[to][k-1]][k-1];
			}
		}
	}
}

int lca(int a, int b)
{
	if (depth[a] < depth[b])
		swap(a, b);
	for (int k = M-1; k >= 0; k--)
		if (depth[fa[a][k]] >= depth[b])
			a = fa[a][k];
	if (a == b)
		return a;
	for (int k = M-1; k >= 0; k--)
		if (fa[a][k] != fa[b][k])
		{
			a = fa[a][k];
			b = fa[b][k];
		}
	return fa[a][0];
}
int dist(int a, int b)
{
	int t = lca(a, b);
	return depth[a] + depth[b] - 2 * depth[t];
}
int main()
{
	ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
	cin >> n >> Q;
	for (int i = 1; i < n; i++)
	{
		int u, v;
		cin >> u >> v;
		add(u, v), add(v, u);
	}
	bfs(1);
	while(Q--)
	{
		int a, b, c, d;
		cin >> a >> b >> c >> d;
		int x = lca(a, b), y = lca(c, d);
		if (dist(x, c) + dist(x, d) == dist(c, d) || dist(y, a) + dist(y, b) == dist(a, b))
			cout << "Y" << endl;
		else
			cout << "N" << endl;
	}
	return 0;
}

标签:include,dist,int,sugar,fa,depth,lca,仓鼠
来源: https://www.cnblogs.com/hzy717zsy/p/16311097.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有