codeforces-1328E-Tree Queries

codeforces-1328E-Tree Queries

传送门：https://codeforces.com/problemset/problem/1328/E

题意：n个结点的树 m次询问 让你找一条从根结点1开始的路径 使一些点在路径上或离路径上的点距离为1 问能不能找到

因为是要求这些点在路径上或者离路径上的点的距离为1，所以可以把每个点处理为他的父节点，然后再考虑（妙）

这样问题就变成这些点是不是都在一条路上，这条路还得从1开始

按照深度排序，看深的能不能走到浅的那里（看最近公共祖先是不是深度小的那个点），就可以啦

算是一道lca吧

关于倍增lca可以看我https://www.cnblogs.com/YangKun-/p/12524015.html的另一篇博客

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=1e6+10;
const int inf=0x3f3f3f3f;
vector<int>ve[200009];
int vis[200009],father[200009][25];
int now[200009];
int deep[200009];
void dfs(int u)
{
    for(int i=0; i<ve[u].size(); i++)
    {
        int v=ve[u][i];
        if(vis[v]==0)
        {
            father[v][0]=u;//记得初始化
            vis[v]=1;
            deep[v]=deep[u]+1;
            dfs(v);
        }
    }
}
void st(int n)
{
    for(int j=1; (1<<j)<=n; j++)
        for(int i=1; i<=n; i++)
            father[i][j]=father[father[i][j-1]][j-1];
}
int lca(int u,int v)
{
    int h=deep[u]-deep[v];
    for(int i=0; i<20; i++)
    {
        if(h&(1<<i))
        {
            u=father[u][i];
        }
    }
    if(u==v) return u;
    else return -1;
}
int cmp(int a,int b)
{
    return deep[a]>deep[b];
}
int main()
{
    int n,m,l,r;
    scanf("%d%d",&n,&m);
    for(int i=1; i<n; i++)
    {
        scanf("%d%d",&l,&r);
        ve[l].push_back(r);
        ve[r].push_back(l);
    }
    vis[1]=1;
    deep[1]=1;
    father[1][0]=0;
    dfs(1);
    st(n);
//    for(int i=0;i<=n;i++) printf("deep[%d]=%d
",i,deep[i]);
    while(m--)
    {
        int k;
        scanf("%d",&k);
        for(int i=1; i<=k; i++)
        {
            scanf("%d",&now[i]);
            now[i]=father[now[i]][0];
//            printf("%d ",now[i]);
        }
        sort(now+1,now+1+k,cmp);
        int flag=0;
        for(int i=1; i<k; i++)
        {
            if(lca(now[i],now[i+1])==-1)
            {
                flag=1;
                break;
            }
        }
        if(flag) printf("NO
");
        else printf("YES
");
    }
    return 0;
}