Codeforces Round #135 (Div. 2)(树dp)

题目:http://codeforces.com/contest/219/problem/D
题意:给你一棵树,选择一个结点使得可以改变最少的边的方向达到其他所有点
输出改变边的个数,输出结点(多个的话升序输出)
思路:建边,方向对应0,反边对应为1,先dfs统计结点的子树贡献,
dp[i]表示从i出发在i的子树中的总贡献,
然后dfs统计结点的父亲贡献,
f[i]表示从i出发在的向父节点方向的总贡献,从i的父节点走的话可能
继续向父节点走,也可能向i的兄弟树走,所以f[v] = f[u] + (1-w) + (dp[u] - (dp[v]+w))
最后找最小值
这个代码用的f[v] = f[u] + (1-w) - w没怎么理解,没想清楚怎么加的
代码:

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
const int N = 2*1e5+5;

struct edge
{
    int v,w;
};
vector<edge> G[N];
vector<int> V;
int dp[N],f[N];
void dfs1(int u,int fa)
{
    for(auto e : G[u])
    {
        int v = e.v,w = e.w;
        if(v == fa)
            continue;
        dfs1(v,u);
        dp[u] += dp[v] + w;
    }
}
void dfs2(int u,int fa)
{
    for(auto e : G[u])
    {
        int v = e.v,w = e.w;
        if(v == fa)
            continue;
        f[v] = f[u] + (1-w) + (dp[u] - (dp[v]+w));
        dfs2(v,u);
    }
}
int main()
{
    int n;
    scanf("%d",&n);
    int u,v;
    for(int i = 1;i < n;i++)
    {
        scanf("%d%d",&u,&v);
        G[u].push_back({v,0});
        G[v].push_back({u,1});
    }
    dfs1(1,-1);
    dfs2(1,-1);
    for(int i = 1;i <= n;i++)
        f[i] += dp[i];
    int minn = *min_element(f+1,f+n+1);
    printf("%d\n",minn);
    for(int i = 1;i <= n;i++)
        if(f[i] == minn)
            printf("%d ",i);
    printf("\n");
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值