CFdiv3-Passable Paths-(思维+lca)

G

题意：
就是给你一个数，然后给你m次查询，每次查询给你一些点，然后问你这些点是否在一条链上。

思考：
看到这题，画了画图模拟了一下，发现对于一个点，如果这个点的所有儿子所代表的子树中有集合中的点，如果>=3个儿子有就不行。因为这就分叉了，没法走一条链。所以直接dfs到now点，看看儿子所在的子树中有几个是有集合点，同时看看他自己父亲节点上面有没有。判断一下即可。这样的复杂度是q*O(n)的。由于easy版本只查询5次所以可以。
对于hard版本，查询1e5次。那么必须要O(n)的，刚开始想了下预处理？但是点集那么多，预处理肯定不行。然后就又感觉，只要任意两点的lca都在点集里面就行，但是任意两点的lca是没法求的，而且发现这样也是错的，比如整个图都选。然后貌似就没好好画图理解了，其实还是要画画图多思考思考的。其实可以先把所有点共同的lca求出来。那么既然这个集合的点在一条链上，那么我先把深度最深的那个点求出来。然后对集合中的每个点和最深点求一下lca，看看是否等于总lca，如果等于总lca，那么代表在链的另一边，如果不等于，那么代表在链的这一边。所以分成两组后，按深度从小到大排序后，这个点必须是下一个点的祖先节点，只有这样才是两条链通过总lca链接起来。所以呀，多画画图思考思考。而且最让我自己疑惑的是自己竟然没想过过着有求一堆点的总lca这种思想…。

代码：

easy版本：

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
		
using namespace std;
const int mod = 1e9+7,inf = 1e18;
const int N = 2e5+10,M = 2010;

int T,n,m,k;
int va[N];
int vis[N];
int suc;

vector<int> e[N];

bool dfs(int now,int p,int hav)
{
	int sum = hav;
	hav |= vis[now];
	int ok = vis[now];
	for(auto spot:e[now])
	{
		if(spot==p) continue;
		if(dfs(spot,now,hav)) ok = 1,sum++;
	}
	if(sum>=3) suc = 0;
	return ok;
}

signed main()
{
	IOS;
	cin>>n;
	for(int i=1;i<n;i++)
	{
		int a,b;
		cin>>a>>b;
		e[a].pb(b);
		e[b].pb(a);
	}
	cin>>m;
	while(m--)
	{
		cin>>k;
		for(int i=1;i<=n;i++) vis[i] = 0;
		for(int i=1;i<=k;i++)
		{
			int x;
			cin>>x;
			vis[x] = 1;
		}
		suc = 1;
		for(int i=1;i<=n;i++)
		{
			if(vis[i])
			{
				dfs(i,0,0);
				break;
			}
		}
		suc?cout<<"YES\n":cout<<"NO\n";
	}
	return 0;
}

hard版本:
#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
		
using namespace std;
const int mod = 1e9+7,inf = 1e18;
const int N = 2e5+10,M = 2010;

int T,n,m,k;
int va[N];
int acc[N][25],cnt = 22;
int dep[N];

vector<int > e[N];

void dfs(int now,int p)
{
	acc[now][0] = p;
	dep[now] = dep[p]+1;
	for(auto spot:e[now])
	{
		if(spot==p) continue;
		dfs(spot,now);
	}
}

int lca(int a,int b)
{
	if(dep[a]<dep[b]) swap(a,b);
	for(int i=cnt;i>=0;i--)
	{
		if(dep[acc[a][i]]>=dep[b])
		a = acc[a][i];
	}
	if(a==b) return a;
	for(int i=cnt;i>=0;i--)
	{
		if(acc[a][i]!=acc[b][i])
		{
			a = acc[a][i];
			b = acc[b][i];
		}
	}
	return acc[a][0];
}

signed main()
{
	IOS;
	cin>>n;
	for(int i=1;i<n;i++)
	{
		int a,b;
		cin>>a>>b;
		e[a].pb(b);
		e[b].pb(a);
	}
	dfs(1,0);
	for(int i=1;i<=cnt;i++)
	{
		for(int j=1;j<=n;j++)
		acc[j][i] = acc[acc[j][i-1]][i-1];
	}
	cin>>m;
	vector<PII > v1,v2;
	while(m--)
	{
		cin>>k;
		v1.clear(),v2.clear();
		int down = -1;
		for(int i=1;i<=k;i++)
		{
			cin>>va[i];
			if(down==-1||dep[down]<dep[va[i]]) down = va[i];
		}
		int up = va[1];
		for(int i=2;i<=k;i++) up = lca(up,va[i]);
		for(int i=1;i<=k;i++)
		{
			if(lca(down,va[i])==up) v1.pb({dep[va[i]],va[i]});
			else v2.pb({dep[va[i]],va[i]});
		}
		int suc = 1;
		sort(v1.begin(),v1.end());
		sort(v2.begin(),v2.end());
		for(int i=0;i<(int)v1.size()-1;i++)
		{
			int now = lca(v1[i].se,v1[i+1].se);
			if(now!=v1[i].se) suc = 0;
		}
		for(int i=0;i<(int)v2.size()-1;i++)
		{
			int now = lca(v2[i].se,v2[i+1].se);
			if(now!=v2[i].se) suc = 0;
		}
		suc?cout<<"YES\n":cout<<"NO\n";
	}
	return 0;
}

总结：
多多思考，画图，思考，画图，多多积累经验吧，发散思维。