zoukankan      html  css  js  c++  java
  • Codeforces Round #359 (Div. 2) D. Kay and Snowflake 树的重心

    题目链接:

    题目

    D. Kay and Snowflake

    time limit per test
    3 seconds
    memory limit per test
    256 megabytes
    input
    standard input
    output
    standard output

    问题描述

    After the piece of a devilish mirror hit the Kay's eye, he is no longer interested in the beauty of the roses. Now he likes to watch snowflakes.

    Once upon a time, he found a huge snowflake that has a form of the tree (connected acyclic graph) consisting of n nodes. The root of tree has index 1. Kay is very interested in the structure of this tree.

    After doing some research he formed q queries he is interested in. The i-th query asks to find a centroid of the subtree of the node vi. Your goal is to answer all queries.

    Subtree of a node is a part of tree consisting of this node and all it's descendants (direct or not). In other words, subtree of node v is formed by nodes u, such that node v is present on the path from u to root.

    Centroid of a tree (or a subtree) is a node, such that if we erase it from the tree, the maximum size of the connected component will be at least two times smaller than the size of the initial tree (or a subtree).

    输入

    The first line of the input contains two integers n and q (2 ≤ n ≤ 300 000, 1 ≤ q ≤ 300 000) — the size of the initial tree and the number of queries respectively.

    The second line contains n - 1 integer p2, p3, ..., pn (1 ≤ pi ≤ n) — the indices of the parents of the nodes from 2 to n. Node 1 is a root of the tree. It's guaranteed that pi define a correct tree.

    Each of the following q lines contain a single integer vi (1 ≤ vi ≤ n) — the index of the node, that define the subtree, for which we want to find a centroid.

    输出

    For each query print the index of a centroid of the corresponding subtree. If there are many suitable nodes, print any of them. It's guaranteed, that each subtree has at least one centroid.

    样例

    input

    7 4
    1 1 3 3 5 3
    1
    2
    3
    5

    output

    3
    2
    3
    6

    题意

    求需要查询的子树的重心。

    题解

    对于点u,它的重心会在它的以重儿子为根的子树的重心和u的路径上。

    可以用dfs直接暴力递归。(如果u的重儿子子树的重心深度比较高,那么u的重心深度也会比较高,严格的时间证明也不懂。。)

    代码

    #include<iostream>
    #include<cstdio>
    #include<cstring>
    #include<vector>
    using namespace std;
    
    const int maxn = 3e5 + 10;
    
    int n, m;
    
    int siz[maxn], ans[maxn],f[maxn];
    vector<int> G[maxn];
    void dfs(int u) {
    	siz[u] = 1; ans[u] = u;
    	for (int i = 0; i < G[u].size(); i++) {
    		int v = G[u][i];
    		dfs(v);
    		siz[u] += siz[v];
    	}
    	for (int i = 0; i < G[u].size(); i++) {
    		int v = G[u][i];
    		if (siz[v] * 2 > siz[u]) {
    			ans[u] = ans[v];
    			break;
    		}
    	}
    	while ((siz[u] - siz[ans[u]]) * 2>siz[u]) ans[u] = f[ans[u]];
    }
    
    int main() {
    	scanf("%d%d", &n, &m);
    	for (int i = 2; i <= n; i++) {
    		scanf("%d", f + i);
    		G[f[i]].push_back(i);
    	}
    	dfs(1);
    	while (m--) {
    		int v; scanf("%d", &v);
    		printf("%d
    ", ans[v]);
    	}
    	return 0;
    }
    
  • 相关阅读:
    典型案例道出“服务台”的价值
    银监会拟允许银行理财产品直接投资
    解读中国版存款保险制度:差别费率+强监管色彩
    央行牵头互联网金融“顶层设计”引业内关注
    央行降息 是农村互联网金融的救命稻草?
    历史上最伟大的 12 位程序员
    年关将至业内警示P2P跑路风险
    央行启动我国征信自律组织研究课题
    windows下开启redis拓展
    php使用curl新增微信临时素材(上传图片)
  • 原文地址:https://www.cnblogs.com/fenice/p/5622024.html
Copyright © 2011-2022 走看看