Codeforces Round #359 (Div. 1) B. Kay and Snowflake dfs

最新推荐文章于 2020-06-16 00:35:44 发布

转载最新推荐文章于 2020-06-16 00:35:44 发布 · 276 阅读

dfs 专栏收录该内容

21 篇文章

订阅专栏

本文介绍了一种利用树形动态规划方法解决特定子树分割问题的算法实现。该算法通过递归地遍历树结构并计算每个节点作为根的子树中最佳分割点，确保分割后的子树大小不超过原始子树大小的一半。文中详细展示了如何通过深度优先搜索（DFS）来构造数据结构，并提供了一个完整的C++代码示例。

参考http://www.cnblogs.com/qscqesze/p/5615969.html
**题意：

　　给你一棵树,n点n-1边，m个询问，每次询问你x点及其子树中选择一个点删除，最后这个子树的形成的树枝中节点树不超过siz[x]/2，输出这个点

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include<bits/stdc++.h>
using namespace std;
#define sf scanf
#define pf printf
#define mem(a,b) memset(a,b,sizeof(a));
#define rep(i,a,b) for(int i=(a);i<=(b);++i)
#define MP make_pair
#define ULL unsigned long long
#define LL   long long
#define inf 0x3f3f3f3f
#define md ((ll+rr)>>1)
#define ls (i<<1)
#define rs (ls|1)
#define eps 1e-5
#define ree freopen("in.txt","r",stdin);
#define bug pf("----------------");
#define N 300010
//2017年09月02日12:40:54

int ans[N];
vector<int>G[N];
int fa[N];
int sz[N];
void dfs(int x){
    int len=G[x].size();
    sz[x]=1;
    ans[x]=x;
    for(int i=0;i<len;++i){
        int v=G[x][i]; dfs(v);
        sz[x]+=sz[v];
    }
    for(int i=0;i<len;++i){
        int v=G[x][i];
        if(sz[v]*2>sz[x]){
            ans[x]=ans[v];
        }
    }
    while((sz[x]-sz[ans[x]])*2>sz[x])ans[x]=fa[ans[x]];
}
int main(){
    int n;sf("%d",&n);
    int q;sf("%d",&q);
    rep(i,2,n){
        sf("%d",&fa[i]);
        G[fa[i]].push_back(i);
    }
    dfs(1);
    while(q--){
        int x;sf("%d",&x);
        pf("%d\n",ans[x]);
    }
}