leetcode每日一题(好几天之前的) -- 3068.最大节点价值之和

思路

一开始,我是想利用异或的特性,重复异或两次变为原值,那么只要异或后的值更大,就更新

  • 但是这和题意相违背,因为题目做处理是对边整体进行的 (我写的版本是对顶点分别控制的)

但即使加上一起变动的特性,这个思路也是不对的

  • 如果一个顶点有多条边相连 & 当前已经对[先遍历到的边]异或过
  • 那么到了同顶点的其他边时就无法再变动了 -- 因为异或后会变小(变为原值),但是可能这里对另一个顶点异或后会带来全局的更优解

总之还是去看了题解

贪心

梳理一下题意

  • 规定是可以对相邻两个节点同时进行异或操作,但是树中任意两个节点之间都是可以联通的(树 -- 无环且连通的无向图)
  • 所以如果对一条路径上的所有边都进行异或操作,最终只有路径端点被异或,其他节点都没变(异或两次恢复了)
  • 于是,原先的 [只有相邻节点可以被异或] 可以转换成 [对任意两个节点异或]

于是,可以先对每个节点进行异或,计算出和原来相比的差值

  • 如果差值为正,说明异或后可以使总和变大,那么就加入到结果中
  • 既然要求和的最大值 & 可以对任意两点进行异或,那就直接按差值排序 ,然后再成对使用

树形dp

递归

本题中,每个节点有两种选择:异或该节点 / 不异或该节点

  • 而这里的选择取决于该节点的子节点选择了什么,子节点的选择又取决于孙子节点...
  • 于是形成了自顶向下的依赖关系,所以最好是使用递归逻辑

预处理

将原先给定边的形式,转换为邻接表

  • 方便后续的递归操作(dfs)

未完待续....

代码

cpp 复制代码
class Solution {
public:
    long long maximumValueSum(vector<int>& nums, int k,
                              vector<vector<int>>& edges) {
        long long res = accumulate(nums.begin(), nums.end(), 0ll);
        vector<int> diff;
        for (auto& a : nums) {
            diff.push_back((a ^ k) - a);
        }
        sort(diff.begin(), diff.end());
        for (int i = diff.size() - 1; i > 0 && diff[i] + diff[i - 1] >= 0;
             i -= 2) {
            res += max(0, diff[i] + diff[i - 1]);
        }
        return res;
    }
};
cpp 复制代码
class Solution {
    // dfs函数返回一个pair<long long, long long>:
    // first  -> 当前节点u"不进行异或操作"时,子树的最大价值和
    // second -> 当前节点u"进行异或操作"时,子树的最大价值和
    pair<long long, long long> dfs(int node, int parent,
                                   const vector<vector<int>>& graph,
                                   const vector<int>& nums, int k) {
        long long sumIfNotXor = 0;
        long long sumIfXor = LLONG_MIN;

        for (int child : graph[node]) {
            //
            if (child == parent)
                continue;
            auto [childNotXor, childXor] = dfs(child, node, graph, nums, k);

            long long newSumIfXor =
                max(sumIfXor + childNotXor, sumIfNotXor + childXor);
            long long newSumIfNotXor =
                max(sumIfNotXor + childNotXor, sumIfXor + childXor);

            sumIfXor = newSumIfXor;
            sumIfNotXor = newSumIfNotXor;
        }

        long long resultNotXor =
            max(sumIfNotXor + nums[node], sumIfXor + (nums[node] ^ k));
        long long resultXor =
            max(sumIfXor + nums[node], sumIfNotXor + (nums[node] ^ k));

        return {resultNotXor, resultXor};
    }

public:
    long long maximumValueSum(vector<int>& nums, int k,
                              vector<vector<int>>& edges) {
        int n = (int)nums.size();
        vector<vector<int>> graph(n);
        // 建图,邻接表形式
        for (auto& edge : edges) {
            int u = edge[0], v = edge[1];
            graph[u].push_back(v);
            graph[v].push_back(u);
        }

        // 从根节点0开始DFS,返回两个状态的最大值,我们要的是未异或状态的最大值
        return dfs(0, -1, graph, nums, k).first;
    }
};
相关推荐
又见野草14 小时前
软件设计师知识点总结:数据结构与算法(超级详细)
数据结构·算法·排序算法
GalaxyPokemon15 小时前
有一个服务器,用于提供HTTP服务,但是需要限制每个用户在任意的100秒内只能请求60次,怎么实现这个功能
算法
fl17683115 小时前
基于opencv+Mediapipe+CNN实现用手势识别控制对鼠标操控python源码+项目说明+设计文档
算法
K 旺仔小馒头15 小时前
优选算法:01 双指针巧解移动零问题
c++·算法·刷题
sali-tec16 小时前
C# 基于halcon的视觉工作流-章49-网面破损
开发语言·图像处理·算法·计算机视觉·c#
ysa05103016 小时前
Fenwick 树进行快速统计
算法
im_AMBER16 小时前
Leetcode 33
算法·leetcode·职场和发展
andyguo16 小时前
全面解读大型语言模型测评:从认知演进到实操框架
人工智能·算法
lzptouch17 小时前
线性回归算法
算法·回归·线性回归
曹牧17 小时前
C#:数组不能使用Const修饰符
java·数据结构·算法