CF1328E-Tree Queries（补） (dfs序)

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN=2e5+5;
const ll INF=3e18;
int n,m,k;
vector<int>v[MAXN];
int a[MAXN],dep[MAXN],l[MAXN],r[MAXN],fa[MAXN],cnt;
void dfs(int u)
{
    l[u]=++cnt;
    for(int  i=0; i<v[u].size(); i++)
    {
        int nxt=v[u][i];
        if(nxt==fa[u])
            continue;
        dep[nxt]=dep[u]+1;
        fa[nxt]=u;
        dfs(nxt);
    }
    r[u]=cnt;
}
bool ok(int x,int y)
{
    return l[x]<=l[y]&&r[x]>=l[y];
}
bool check(int u)
{
    for(int i=1; i<=k; i++)
    {
        if(!ok(a[i],u)&&!ok(fa[a[i]],u))
            return false;
    }
    return true;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin>>n>>m;
    for(int i=1; i<=n-1; i++)
    {
        int x,y;
        cin>>x>>y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    dep[1]=1;
    dfs(1);
    while(m--)
    {
        cin>>k;
        int max_=0,u=0;
        for(int i=1; i<=k; i++)
        {
            cin>>a[i];
            if(dep[a[i]]>max_)
            {
                max_=dep[a[i]];
                u=a[i];
            }
        }
        //cout<<u<<endl;
 
        if(check(u))
            cout<<"YES"<<endl;
        else
            cout<<"NO"<<endl;
    }
    return 0;
}