Codeforces Round #629 (Div. 3)E. Tree Queries(树)

【Codeforces Round #629 (Div. 3)E. Tree Queries(树)】题意就是m次询问,问你每次询问的k个点能否都在从根节点到一个点的路径距离1或者就在路径上。
我们发现对于一个节点,如果存在两个以上孙子节点,显然输出NO,那怎么判断何时输出YES呢?显然只要不存在这样两个孙子结点即可,也就是说对于任意节点只可能有多个儿子节点就输出YES,那么判断所有节点的父节点是不是都在一条路径上即可。因为路径可能很长,要用倍增法预处理求出每个点的祖先节点。

#include using namespace std; int n,m; #define MAXN 205000 #define ll long long int fa[MAXN][30]; vectore[MAXN]; int dep[MAXN]; int step[30]; void dfs(int x,int pre) { for(auto it:e[x]) { if(it==pre)continue; fa[it][0]=x; dep[it]=dep[x]+1; for(int i=1; step[i]<=dep[it]; i++) fa[it][i]=fa[fa[it][i-1]][i-1]; dfs(it,x); } } int find(int x,int dep) { int pa=x; for(int i=0; step[i]<=dep; i++) { if((dep&step[i])==step[i]) pa=fa[pa][i]; } return pa; } int main() { step[0]=1; for(int i=1; i<=20; i++) step[i]=step[i-1]*2ll; int n,m; scanf("%d%d",&n,&m); for(int i=1; iv; vector>cmp; for(int i=0; i=1; i--) { int now=cmp[i].second; int pre=cmp[i-1].second; if(now==pre)continue; else { if(find(now,dep[now]-dep[pre])!=pre) boolean=0; } } if(boolean)printf("YES\n"); else printf("NO\n"); } }

    推荐阅读