【LeetCode每日一题】——1863.找出所有子集的异或总和再求和

一【题目类别】

• 回溯

二【题目难度】

• 简单

三【题目编号】

• 1863.找出所有子集的异或总和再求和

四【题目描述】

• 一个数组的 异或总和 定义为数组中所有元素按位 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

七【解题思路】

• 本题的关键是得到目标数组的所有子集（不去重），所以对于目标数组中的任意一个元素都有：
  • 选取当前元素加入子集中
  • 不选取当前元素加入子集中
• 所以我们回溯的时候就应该按照上面两种情况进行选取，这样就能获取到目标数组可以构成的所有子集了
• 获取到有效子集后，就可以对其进行异或操作了
• 当然还要注意回溯的退出条件，即当前遍历数组的索引是否已经到了数组的末尾
• 最后返回结果即可，具体实现可以参考下面的代码

八【时间频度】

• 时间复杂度：$O(2^n)$，$n$为传入的参数的数组的长度
• 空间复杂度：$O(n)$，$n$为传入的参数的数组的长度

九【代码实现】

1. Java语言版

class Solution {
    public int subsetXORSum(int[] nums) {
        // 保存结果
        int[] res = new int[1];
        res[0] = 0;
        // 调用回溯函数计算所有子集的异或总和
        backTrack(nums, nums.length, 0, 0, res);
        // 返回结果
        return res[0];
    }

    // 使用回溯方法计算所有子集的异或总和
    private void backTrack(int[] nums, int numsSize, int index, int xor, int[] res) {
        if (index == numsSize) {
            res[0] += xor;
            return;
        }
        backTrack(nums, numsSize, index + 1, xor ^ nums[index], res);       // 选择当前元素在子集中
        backTrack(nums, numsSize, index + 1, xor, res);                     // 不选择当前元素在子集中
    }
}

2. Python语言版

class Solution:
    def subsetXORSum(self, nums: List[int]) -> int:
        # 保存结果
        res = 0

        # 使用回溯方法计算所有子集的异或总和
        def backTrack(xor, index):
            nonlocal res
            if index == len(nums):
                res += xor
                return
            backTrack(xor ^ nums[index], index + 1) # 选择当前元素在子集中
            backTrack(xor, index + 1)               # 不选择当前元素在子集中
        backTrack(0, 0)

        # 返回结果
        return res

3. C语言版

// 使用回溯方法计算所有子集的异或总和
void backTrack(int* nums, int numsSize, int index, int xor, int* res)
{
    if (index == numsSize)
    {
        *res += xor;
        return;
    }
    backTrack(nums, numsSize, index + 1, xor ^ nums[index], res);   // 选择当前元素在子集中
    backTrack(nums, numsSize, index + 1, xor, res);                 // 不选择当前元素在子集中
}

int subsetXORSum(int* nums, int numsSize)
{
    // 保存结果
    int res = 0;
    // 调用回溯函数计算所有子集的异或总和
    backTrack(nums, numsSize, 0, 0, &res);
    // 返回结果
    return res;
}

十【提交结果】

1. Java语言版
   

2. Python语言版
   

3. C语言版