【递归算法】找出所有子集的异或总和再求和

题目链接

文章摘要:

  • 本文介绍了LeetCode题目《所有子集异或总和之和》的解法。通过分析题目要求,提出三步解题思路:找出所有子集、计算子集异或值、求和异或值。采用决策树模型设计回溯算法,利用全局变量path和sum分别记录子集异或值和总和。详细讲解了dfs函数实现、回溯处理和递归出口等关键步骤,并给出Java代码实现。该方法通过异或运算特性简化回溯操作,高效计算所有子集的异或总和。

一、题目解析

题目定义了对数组的异或总和是怎么运算的。

给我们一个数组,要我们返回该数组的所有子集的异或和再求和。

简单来说,我们做这道题目有三个步骤:

  1. 找出数组的所有的子集
  2. 计算所有子集中的所有数字的异或值
  3. 将所有子集的异或值相加

虽然看起来挺麻烦的,但其实,解题步骤和子集那道题目差不多,我们可以把 2 和 3 的逻辑加入到 1 中。

二、算法原理 + 代码实现

决策树

我们第一步就是画出决策树,这里是找出数组的所有子集,与上一道题目的思路一致,有两种方法,这里采用第二种方法作决策树。

我们根据示例2来画:

决策树:

接下来,我们根据决策树来设计代码。

全局变量

我们这里需要计算的是子集中所有元素的异或值 以及异或值之和。因此定义两个 int 类型即可,path 用来记录子集中所有元素的异或值,ret 用来记录异或值之和。

dfs 函数

函数头

我们这里需要通过递归遍历题目所给数组nums,因此参数是数组 nums 和下标 pos。

函数体

在子集那道题目中,我们的 dfs 函数体中做的事情是 "从当前数字往后遍历",在这里也是一样的,只不过是修改的东西从集合类变成里基本数据类型。当拿到一个数字,就把它与 path 进行异或,然后基于这个再递归。

细节问题

回溯

这里我们也是要在回溯的时候恢复现场的,在函数体当中,递归回来的时候就恢复现场。这里可以利用异或的 "抵消" 性质,再次将 path 与当前数字异或,就可以恢复现场了。

剪枝

这里同样不涉及到剪枝操作。

递归出口

这里不需要设置递归出口。我们更新结果是一进入 dfs 函数就更新的,因为决策树的每一个节点都是结果。当函数体中的遍历结束,整个递归也就结束了。

代码实现

Java 复制代码
public class Solution {
    int sum;  // 记录异或值之和
    int path; // 记录子集中所有元素的异或值

    public int subsetXORSum(int[] nums) {
        dfs(nums, 0);
        return sum;
    }

    private void dfs(int[] nums, int pos) {
        sum += path;  // 更新结果
        
        // 从当前数字往后遍历
        for (int i = pos; i < nums.length; i++) {
            path ^= nums[i];
            dfs(nums, i + 1);
            path ^= nums[i];  // 回溯时恢复现场
        }
    }
}

文章到这里就告一段落啦,若有错误请尽管指出!

相关推荐
白羊by8 小时前
YOLOv1~v11 全版本核心演进总览
深度学习·算法·yolo
嗑嗑嗑瓜子的猫9 小时前
Java!它值得!
java·开发语言
墨尘笔尖9 小时前
最大最小值降采样算法的优化
c++·算法
2401_8955213410 小时前
【Spring Security系列】Spring Security 过滤器详解与基于JDBC的认证实现
java·后端·spring
皮卡蛋炒饭.10 小时前
线程的概念和控制
java·开发语言·jvm
一只大袋鼠10 小时前
MyBatis 入门详细实战教程(一):从环境搭建到查询运行
java·开发语言·数据库·mysql·mybatis
程序员老邢10 小时前
【人生底稿・番外篇 05】我的电影江湖:从录像带时代,到港片陪伴的青春岁月
java·程序人生·职场发展·娱乐
sonnet-102910 小时前
函数式接口和方法引用
java·开发语言·笔记
Bat U10 小时前
JavaEE|多线程(二)
java·开发语言
_Evan_Yao11 小时前
RAG中的“Chunk”艺术:我试过10种切分策略后总结的结论
java·人工智能·后端·python·软件工程