• 树形DP求树的重心 --SGU 134


    令一个点的属性值为:去除这个点以及与这个点相连的所有边后得到的连通分量的节点数的最大值。

    则树的重心定义为:一个点,这个点的属性值在所有点中是最小的。

    SGU 134 即要找出所有的重心,并且找出重心的属性值。

    考虑用树形DP。

    dp[u]表示割去u点,得到的连通分支的节点数的最大值。

    tot[u]记录以u为根的这棵子树的节点数总和(包括根)。

    则用一次dfs即可预处理出这两个数组。再枚举每个点,每个点的属性值其实为max(dp[u],n-tot[u]),因为有可能最大的连通分支在u的父亲及以上。然后记录答案就可以了。

    代码:

    #include <iostream>
    #include <cstdio>
    #include <cstring>
    #include <cstdlib>
    #include <cmath>
    #include <algorithm>
    #include <vector>
    using namespace std;
    
    vector<int> G[16005];
    int dp[16005],tot[16005];
    vector<int> ans;
    
    void dfs(int u)
    {
        tot[u] = 1;
        for(int i=0;i<G[u].size();i++)
        {
            int v = G[u][i];
            if(tot[v] == -1)
                dfs(v);
            else
                continue;
            dp[u] = max(dp[u],tot[v]);
            tot[u] += tot[v];
        }
    }
    
    int main()
    {
        int n,i,j,u,v;
        scanf("%d",&n);
        for(i=0;i<=n;i++)
            G[i].clear();
        for(i=0;i<n-1;i++)
        {
            scanf("%d%d",&u,&v);
            G[u].push_back(v);
            G[v].push_back(u);
        }
        memset(dp,0,sizeof(dp));
        memset(tot,-1,sizeof(tot));
        dfs(1);
        int mini = Mod;
        for(i=1;i<=n;i++)
        {
            int tmp = max(dp[i],n-tot[i]);
            if(tmp < mini)
            {
                ans.clear();
                ans.push_back(i);
                mini = tmp;
            }
            else if(tmp == mini)
                ans.push_back(i);
        }
        printf("%d %d
    ",mini,ans.size());
        sort(ans.begin(),ans.end());
        printf("%d",ans[0]);
        for(i=1;i<ans.size();i++)
            printf(" %d",ans[i]);
        puts("");
        return 0;
    }
    View Code
  • 相关阅读:
    error_reporting(“E_ALL”)和ini_set(“display_errors”, “on”)的区别?
    linux命令awk的详解
    Ubuntu 能PING IP但不能PING主机域名的解决方法
    从github checkout子文件夹
    zuul简单使用
    docker for windows 10 添加阿里云镜像仓库无效问题
    Spring Boot 进行Bean Validate和Method Validate
    JVM调优-GC参数
    Spring Aop: 关于继承和execution target this @annotation
    ReentrantLock原理
  • 原文地址:https://www.cnblogs.com/whatbeg/p/3908559.html
Copyright © 2020-2023  润新知