834. 树中距离之和(hard)

本文介绍了一种使用动态规划的方法,通过深度优先和广度优先遍历构建树的数据结构。作者详细讲解了如何利用递归实现dfs和dfs2函数,以计算节点间的距离之和。实例展示了从给定边的关系中构造树的过程,并提供了一个C++解决方案类实例。
摘要由CSDN通过智能技术生成

树的dp,

讲解:https://www.youtube.com/watch?v=gi2maECPOB0&ab_channel=M.C%E7%B1%B3%E5%BC%80%E6%9C%97%E5%9F%BA%E7%BD%97

建树:

0:1,2

1:0

2:0,3,4,5

3:2

4:2

5:2

/*树的dp,
讲解:https://www.youtube.com/watch?v=gi2maECPOB0&ab_channel=M.C%E7%B1%B3%E5%BC%80%E6%9C%97%E5%9F%BA%E7%BD%97
建树:
0:1,2
1:0
2:0,3,4,5
3:2
4:2
5:2
*/
class Solution {
public:
    vector<set<int> > tree;
    int N;

    // ans是保存子节点的距离之和,count计算有多少个子节点(包括自己)
    void dfs(int cur, int par, vector<int>& ans, int* count)
    {
        for(auto child : tree[cur])
        {
            if(child != par)
            {
                dfs(child, cur, ans, count);
                count[cur] += count[child];
                ans[cur] += (ans[child] + count[child]);//先求出0到其他所有点的距离之和
                //cout << cur << ":" << ans[cur] << " " << count[cur] << endl;
            }
        }
    }
    void dfs2(int cur,int par, vector<int>& ans, int* count)
    {
        for(auto child : tree[cur])
        {
            if(child != par)
            {
                ans[child] = ans[cur] + (N - count[child]) - count[child];
                dfs2(child, cur, ans, count);
            }
        }
    }
    
    vector<int> sumOfDistancesInTree(int N, vector<vector<int>>& edges) {
        this->N = N;
        int len = edges.size();
        //计算每个节点有多少个子节点 count(包括自身)
        int* count = new int[N];
        for(int i = 0;i < N; i++) count[i] = 1;
        //计算当前节点到其他节点的距离之和
        vector<int> ans(N, 0);
        if(len < 1) return ans;

        for(int i = 0;i < N;i++)
        {
            set<int> s;//存放树节点的边
            tree.push_back(s);
        }
        //建树
        for(int i = 0;i < len;i++)
        {
            tree[edges[i][0]].insert(edges[i][1]);
            tree[edges[i][1]].insert(edges[i][0]);
        }

        dfs(0, -1, ans, count);
        dfs2(0, -1, ans, count);
        return ans;
    }
};

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值