CF-Round# 629-div3-E题

CF-Round# 629-div3-E题

E. Tree Queries

传送门

这道题。LCA。
我为了LCA，花了一下午，认为自己懂得很详尽了。
然而。开始写博客的时候。
内心：你懂个🔨啊？？
于是我又理解加操作了一遍。

LCA详解请参考如下链接:（介绍的很详细了）
**

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

**
其实这道题也可以不用LCA来写哈~
这里提供LCA思路。

题目大意：给你一棵树。再给你一些结点。问能否找到一条路径使得这些结点是这条路径中的一员或者使得某些结点到这条路径的某个结点的长度小于1。

思路：
我们可以把这些结点的深度求出来。
然后按照深度sort一下这些结点。
之后两两LCA出这些结点的公共祖先。
如果满足两个结点的到公共祖先的距离都大于1的话。说明不能满足题目要求。
你想想。如果两个结点到lca的距离都小于1的话。那么选择任意一条路径包含这两个结点中任意一个结点就行
如果有一个结点到lca的路径大于1，一个小于1.那么我们选择路径的时候就选择大于1的那条。可以保证另一个到路径某一个结点的距离小于1.

（应该还可以找到一条最深路径，然后判断不在路径上的结点的距离是否小于1的解法）

我wa在了101的test case上;
一头栽在这里。
我有点莫名其妙。。
然后仔细检查发现是双重循环的内部循环条件用重了，emmm。
下次要注意了！！bug！！

代码部分：

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

int n, m;
int lg[N];
int fa[N][25];
int dep[N];
int a[N];
vector<int> v[N << 1];

void dfs(int x, int fath)
{
	fa[x][0] = fath;
	dep[x] = dep[fath] + 1;
	
	for (int i = 1; i <= lg[dep[x]]; i++)
	{
		fa[x][i] = fa[fa[x][i - 1]][i - 1];
	}
	int n = v[x].size();
	for (int i = 0; i < n; i++)
	{
		if (v[x][i] != fath)
		{
			dfs(v[x][i], x);
		}
	}
}

bool cmp(int a, int b)
{
	return dep[a] < dep[b];
}

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 i = lg[dep[x]] - 1; i >= 0; i--)
	{
		if (fa[x][i] != fa[y][i])
		{
			x = fa[x][i];
			y = fa[y][i];
		}
	}
	return fa[x][0];
}

int main()
{
	scanf ("%d%d", &n, &m);
	for (int i = 1; i <= n; i++)
	{
		lg[i] = lg[i - 1] + (1 << lg[i - 1] == i);
	}
	for (int i = 1; i < n; i++)
	{
		int u1, u2;
		scanf ("%d%d", &u1, &u2);
		v[u1].push_back(u2);
		v[u2].push_back(u1);
	}
	dfs(1, 0);
	for (int i = 1; i <= m; i++)
	{
		int k;
		int flag = 0;
		scanf ("%d", &k);
		for (int i = 1; i <= k; i++)
		{
			scanf ("%d", &a[i]);
		}
		sort(a + 1, a + k + 1, cmp);
		if (k == 1)
		{
			cout << "YES\n";
			continue;
		}
		for (int j = 1; j < k; j++)
		{
			int ance = LCA(a[j], a[j + 1]);
			if (dep[a[j]] - dep[ance] > 1 && dep[a[j + 1]] - dep[ance] > 1)
			{
				flag = 1;
			}
		}
		if (flag)
		{
			cout << "NO\n";
		}
		else
		{
			cout << "YES\n";
		}
	}
	return 0;
}