「递归算法」:找出所有子集的异或总和再求和

22 阅读2分钟

一、题目

一个数组的 异或总和 定义为数组中所有元素按位 XOR 的结果;如果数组为 ,则异或总和为 0

  • 例如,数组 [2,5,6]异或总和2 XOR 5 XOR 6 = 1

给你一个数组 nums ,请你求出 nums 中每个 子集异或总和 ,计算并返回这些值相加之

**注意:**在本题中,元素 相同 的不同子集应 多次 计数。

数组 a 是数组 b 的一个 子集 的前提条件是:从 b 删除几个(也可能不删除)元素能够得到 a

示例 1:

输入:nums = [1,3]
输出:6
解释:[1,3] 共有 4 个子集:
- 空子集的异或总和是 0 。
- [1] 的异或总和为 1 。
- [3] 的异或总和为 3 。
- [1,3] 的异或总和为 1 XOR 3 = 2 。
0 + 1 + 3 + 2 = 6

示例 2:

输入:nums = [5,1,6]
输出:28
解释:[5,1,6] 共有 8 个子集:
- 空子集的异或总和是 0 。
- [5] 的异或总和为 5 。
- [1] 的异或总和为 1 。
- [6] 的异或总和为 6 。
- [5,1] 的异或总和为 5 XOR 1 = 4 。
- [5,6] 的异或总和为 5 XOR 6 = 3 。
- [1,6] 的异或总和为 1 XOR 6 = 7 。
- [5,1,6] 的异或总和为 5 XOR 1 XOR 6 = 2 。
0 + 5 + 1 + 6 + 4 + 3 + 7 + 2 = 28

示例 3:

输入:nums = [3,4,5,6,7,8]
输出:480
解释:每个子集的全部异或总和值之和为 480 。

提示:

  • 1 <= nums.length <= 12
  • 1 <= nums[i] <= 20

二、思路解析

关于这类模拟实现的题,解题思路都是先画出来决策树,然后再做具体的代码设计。

所以,我们拿 [ 1 , 2 , 3 ] 来举例,决策树是长这样子的👇

编辑

在实现过程,其实只要异或就行了,而细节则是在回溯的时候,得想到再用一次异或,来抵消掉最后异或的那个元素,以达到恢复到上一层的目的。

具体实现请看下面代码👇

三、完整代码

class Solution {
    
    int sum;
    int path;

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

    public 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];
        }
    }
}

以上就是本篇博客的全部内容啦,如有不足之处,还请各位指出,期待能和各位一起进步!