设为首页 加入收藏

TOP

poj 3107 Godfather(树形dp)
2014-11-23 21:42:21 来源: 作者: 【 】 浏览:5
Tags:poj 3107 Godfather 树形
题目链接: poj-3107
题意
给一颗n个结点的树,节点编号为1~n,问删除一个节点之后,让剩下的分支中节点数量最多的尽量少。
可能有多种方案,按编号顺序输出。
思路
简单的树形dp. 其实连dp都不能算吧...就是直接计数统计
先dfs计算每个节点子树的节点个数tot[i]。
再次dfs更新答案:
f[i] = max( n-tot[i], max{tot[v] | v是i的儿子} );
两个dfs可以合并在一个dfs里完成, 复杂度O(n)
代码
   
/**=====================================================
 *   This is a solution for ACM/ICPC problem
 *
 *   @source      : poj-3107 Godfather
 *   @description : 树形dp
 *   @author      : shuangde
 *   @blog        : blog.csdn.net/shuangde800
 *   @email       : zengshuangde@gmail.com
 *   Copyright (C) 2013/08/30 16:26 All rights reserved. 
 *======================================================*/
#include 
#include 
#include 
#include 
#include 
using namespace std;

typedef pairPII;
typedef long long int64;
const int INF = 0x3f3f3f3f;

const int MAXN = 50010;
int tot[MAXN];
int f[MAXN], minx;

namespace Adj {
    int size, head[MAXN];
    struct Node{
        int v, next; 
    }E[MAXN*2];
    inline void initAdj() {
        size = 0;
        memset(head, -1, sizeof(head));
    }
    inline void addEdge(int u, int v) {
        E[size].v = v;
        E[size].next = head[u];
        head[u] = size++;
    }
}
using namespace Adj;

int n;

int dfs(int u, int fa) {

    tot[u] = 1;
    // count
    for (int e = head[u]; e != -1; e = E[e].next) {
        int v = E[e].v;
        if (v == fa) continue;
        tot[u] += dfs(v, u);
    }
    // 计算答案
    int& ans = f[u] = n - tot[u];
    for (int e = head[u]; e != -1; e = E[e].next) {
        int v = E[e].v;
        if (v == fa) continue;
        ans = max(ans, tot[v]);
    }
    minx = min(minx, ans);
    return tot[u];
}

int main(){

    while (~scanf("%d", &n)) {

        initAdj();
        for (int i = 0; i < n - 1; ++i) {
            int u, v;
            scanf("%d%d", &u, &v);
            addEdge(u, v);
            addEdge(v, u);
        }

        minx = INF;
        dfs(1, -1);

        bool first = true;
        for (int i = 1; i <= n; ++i) 
            if (f[i] == minx) {
                if (first) first = false, printf("%d", i);
                else printf(" %d", i);
            }
        puts("");
    }
    return 0;


】【打印繁体】【投稿】【收藏】 【推荐】【举报】【评论】 【关闭】 【返回顶部
分享到: 
上一篇UVa 531: Compromise 下一篇POJ 1472 Coins (多重背包+滚动..

评论

帐  号: 密码: (新用户注册)
验 证 码:
表  情:
内  容:

·【C语言】动态内存管 (2025-12-27 06:23:20)
·C语言中的内存管理 - (2025-12-27 06:23:16)
·C语言指南:C语言内 (2025-12-27 06:23:14)
·Redis on AWS:Elast (2025-12-27 04:19:30)
·在 Spring Boot 项目 (2025-12-27 04:19:27)