树形Dp 2925. 在树上执行操作以后得到的最大分数

2925. 在树上执行操作以后得到的最大分数

两次DFS

cpp 复制代码
class Solution {
public:
    // 节点状态有两种,选和不选,
    // dp(u, fa, 0) 不选u 节点,其他节点都可以选,值为以u为根的子树的所有节点的和- 根节点的值。
    // dp(u, fa, 1) 选u节点, 其他子几点不选。
    vector<vector<int>> g;
    int n;
    vector<long long> gsum;
    void dfs(int u, int fa, vector<int>& values) {
        for (auto v : g[u]) {
            if (v == fa) continue;
            dfs(v, u, values);
            gsum[u] += gsum[v];
        }
        gsum[u] += values[u];
        return;
    }

    vector<long long> dp0;

    vector<long long> dp1;
    void Dfs2(int u, int fa, vector<int>& values) {
        dp1[u] += values[u];
        for (auto v : g[u]) {
            if (v == fa) continue;
            Dfs2(v, u, values);
            if (g[v].size() == 1) { // 叶子节点
                dp1[u] += dp0[v];
            } else {
                dp1[u] += max(dp1[v], dp0[v]);
            }
        }
    }

    long long maximumScoreAfterOperations(vector<vector<int>>& edges, vector<int>& values) {
        n = edges.size() + 1;
        g.resize(n);
        for (auto edge : edges) {
            g[edge[0]].push_back(edge[1]);
            g[edge[1]].push_back(edge[0]);
        }
        gsum.resize(n, 0);
        dfs(0, -1, values);
        cout << endl;
        dp0.resize(n);
        for(int i = 0; i < n; i++) {
            dp0[i] = gsum[i] - values[i];

        }
        dp1.resize(n);
        Dfs2(0, -1, values);
        return max(dp0[0], dp1[0]);
   
    }
};

一次dfs

cpp 复制代码
class Solution {
public:
    // 节点状态有两种,选和不选,
    // dp(u, fa, 0) 不选u 节点,其他节点都可以选,值为以u为根的子树的所有节点的和- 根节点的值。
    // dp(u, fa, 1) 选u节点, 其他子几点不选。
    vector<vector<int>> g;
    int n;
    vector<long long> dp0;
    vector<long long> dp1;
    void Dfs2(int u, int fa, vector<int>& values) {
        dp1[u] += values[u];
        for (auto v : g[u]) {
            if (v == fa) continue;
            Dfs2(v, u, values);
            dp0[u] += dp0[v] + values[v];
            if (g[v].size() == 1) { // 叶子节点, 注意叶子节点的size 为1,不是0
                dp1[u] += dp0[v];
            } else {
                dp1[u] += max(dp1[v], dp0[v]);
            }

        }
    }

    long long maximumScoreAfterOperations(vector<vector<int>>& edges, vector<int>& values) {
        n = edges.size() + 1;
        g.resize(n);
        for (auto edge : edges) {
            g[edge[0]].push_back(edge[1]);
            g[edge[1]].push_back(edge[0]);
        }
        dp0.resize(n);
        dp1.resize(n);
        Dfs2(0, -1, values);
        return max(dp0[0], dp1[0]);

    }
};
相关推荐
~|Bernard|1 小时前
在 PyCharm 里怎么“点鼠标”完成指令同样的运行操作
算法·conda
战术摸鱼大师1 小时前
电机控制(四)-级联PID控制器与参数整定(MATLAB&Simulink)
算法·matlab·运动控制·电机控制
Christo31 小时前
TFS-2018《On the convergence of the sparse possibilistic c-means algorithm》
人工智能·算法·机器学习·数据挖掘
好家伙VCC2 小时前
数学建模模型 全网最全 数学建模常见算法汇总 含代码分析讲解
大数据·嵌入式硬件·算法·数学建模
liulilittle4 小时前
IP校验和算法:从网络协议到SIMD深度优化
网络·c++·网络协议·tcp/ip·算法·ip·通信
bkspiderx5 小时前
C++经典的数据结构与算法之经典算法思想:贪心算法(Greedy)
数据结构·c++·算法·贪心算法
中华小当家呐6 小时前
算法之常见八大排序
数据结构·算法·排序算法
沐怡旸7 小时前
【算法--链表】114.二叉树展开为链表--通俗讲解
算法·面试
一只懒洋洋7 小时前
K-meas 聚类、KNN算法、决策树、随机森林
算法·决策树·聚类
方案开发PCBA抄板芯片解密9 小时前
什么是算法:高效解决问题的逻辑框架
算法