洛谷-P3379-LCA-模板题

洛谷-P3379-LCA-模板题

传送门

什么是LCA??
具体请查看下面链接
**

https://blog.csdn.net/qq_44624316/article/details/105165911

**

这是一道模板题。
卡vector,卡cin.
所以老实点用邻接表和scanf把~

上面说的很详细啦~
用的是倍增。暴力的优化。
鬼知道我理解倍增理解了一下午emmm。
(主要是存储和优化位置)
然后!!自信满满以为理解了。
开始写博客了。。
发现。似乎???又什么都不会了。
我的内心:???

后来又去各种画图理解了一遍~呼,终于完成了一篇博客。
理解之后就很简单啦!二进制的运用~

话不多说。

代码部分:

#include <bits/stdc++.h>
using namespace std;
const int N = 5e5 + 10;

struct node
{
	int to;
	int next;
}edge[N << 1];
int head[N];
int n, m, s;
int cnt;
int lg[N];
int dep[N];
int fa[N][25];

void add(int x, int y)
{
	edge[++cnt].to = y;
	edge[cnt].next = head[x];
	head[x] = cnt;
}

void dfs(int now, int fath)
{
	fa[now][0] = fath;
	dep[now] = dep[fath] + 1;
	for (int i = 1; i <= lg[dep[now]]; i++)
	{
		fa[now][i] = fa[fa[now][i - 1]][i - 1];
	}
	for (int i = head[now]; i; i = edge[i].next)
	{
		if (edge[i].to != fath)
		{
			dfs(edge[i].to, now);
		}
	}
}

int LCA(int x, int y)
{
	if (dep[x] < dep[y])
	{
		swap(x, y);
	}
	while (dep[x] > dep[y])
	{
		x = fa[x][lg[dep[x] - dep[y]] - 1];
	}
	if (x == y)
	{
		return x;
	}
	for (int k = lg[dep[x]] - 1; k >= 0; k--)
	{
		if (fa[x][k] != fa[y][k])
		{
			x = fa[x][k];
			y = fa[y][k];
		}
	}
	return fa[x][0];
}

int main()
{
	scanf ("%d%d%d", &n, &m, &s);
	for (int i = 1; i < n; i++)
	{
		int u1, u2;
		scanf ("%d%d", &u1, &u2);
		add(u1, u2);
		add(u2, u1);
	}
	for (int i = 1; i <= n; i++)
	{
		lg[i] = lg[i - 1] + (1 << lg[i - 1] == i);
	}
	dfs(s, 0);
	for (int i = 1; i <= m; i++)
	{
		int x, y;
		scanf ("%d%d", &x, &y);
		cout << LCA(x, y) << endl;
	}
	return 0;
}
发布了127 篇原创文章 · 获赞 3 · 访问量 3224

猜你喜欢

转载自blog.csdn.net/qq_44624316/article/details/105167889
今日推荐