树学

树学

https://ac.nowcoder.com/acm/problem/201400

题意:
给你一颗树,你选择一个节点当根,求所有点的深度和最少为多少?

思路:
树状dp+换根
一开始随便选一个点当根计算结果。
dp[i]表示以i为根的子树节点的深度和。
se[i]表示以i为根的子树的节点数目。
换根:
dp[v]=dp[u]+n-2*se[v];(v为u的子节点,根从u转向v时,以v为根的子树中的节点深度全减一,其余节点深度加一)

代码:

#include <bits/stdc++.h>
typedef long long ll;

using namespace std;

const ll inf=1000000007;

inline int read()
{
    int x=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')
            f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        x=(x<<1)+(x<<3)+(ch^48);
        ch=getchar();
    }
    return x*f;
}

vector<int> g[1000005];
ll dp[1000005], se[1000005];
int n;

void dfs(int v,int f,int d)
{
    dp[v]=d;
    se[v]=1;
    for(int i=0;i<g[v].size();i++)
    {
        if(g[v][i]!=f)
        {
            dfs(g[v][i],v,d+1);
            dp[v]+=dp[g[v][i]];
            se[v]+=se[g[v][i]];
        }
    }
}

void dfs1(int v,int f)
{
    if(f!=-1)
    {
        dp[v]=dp[f]+n-2*se[v];
    }
    for(int i=0;i<g[v].size();i++)
    {
        if(g[v][i]!=f)
        {
            dfs1(g[v][i],v);
        }
    }
}

int main()
{
    n=read();
    for(int i=1;i<n;i++)
    {
        int u=read(), v=read();
        g[u].push_back(v);
        g[v].push_back(u);
    }
    dfs(1,-1,0);
    dfs1(1,-1);
    ll ans=inf*inf;
    for(int i=1;i<=n;i++)
    {
        ans=min(ans,dp[i]);
    }
    cout << ans << endl;
    return 0;
}
全部评论

相关推荐

评论
1
1
分享

创作者周榜

更多
正在热议
更多
# AI面会问哪些问题? #
24476次浏览 480人参与
# 中国电信笔试 #
31011次浏览 283人参与
# 厦门银行科技岗值不值得投 #
7431次浏览 186人参与
# 你的实习产出是真实的还是包装的? #
18635次浏览 329人参与
# 如果秋招能重来,我会____ #
96616次浏览 500人参与
# 春招至今,你的战绩如何? #
59439次浏览 535人参与
# 开放七大实习专项,百度暑期实习值得冲吗 #
14048次浏览 209人参与
# i人适合做什么工作 #
36838次浏览 124人参与
# 我是面试官,请用一句话让我破防 #
79444次浏览 219人参与
# 哪些公司真双非友好? #
69176次浏览 287人参与
# 找AI工作可以去哪些公司? #
7561次浏览 179人参与
# 从事AI岗需要掌握哪些技术栈? #
7528次浏览 238人参与
# 面试尴尬现场 #
220729次浏览 861人参与
# 投递几十家公司,到现在0offer,大家都一样吗 #
339792次浏览 2163人参与
# 五一之后,实习真的很难找吗? #
102792次浏览 584人参与
# 金三银四,你的春招进行到哪个阶段了? #
21492次浏览 275人参与
# 你做过最难的笔试是哪家公司 #
29736次浏览 182人参与
# 你小时候最想从事什么职业 #
159832次浏览 2072人参与
# 阿里笔试 #
176137次浏览 1300人参与
# 应届生第一份工资要多少合适 #
20463次浏览 84人参与
# 一张图晒出你司的标语 #
3784次浏览 71人参与
# 面试被问期望薪资时该如何回答 #
382445次浏览 2163人参与
牛客网
牛客网在线编程
牛客网题解
牛客企业服务