题意:
就是给你一个数,然后给你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;
}
总结:
多多思考,画图,思考,画图,多多积累经验吧,发散思维。