Leecode(74)1863.找出所有子集的异或总和再求和-简单

最新推荐文章于 2024-09-12 18:58:16 发布

qq_51999844

最新推荐文章于 2024-09-12 18:58:16 发布

阅读量933

点赞数 13

文章标签：算法数据结构

本文链接：https://blog.csdn.net/qq_51999844/article/details/141126657

版权

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

from typing import List

class Solution:
    def subsetXORSum(self, nums: List[int]) -> int:
        total_sum = 0
        n = len(nums)
        
        # 2^n possible subsets
        for i in range(1 << n):
            xor_sum = 0
            for j in range(n):
                if i & (1 << j):
                    xor_sum ^= nums[j]
            total_sum += xor_sum
        
        return total_sum

1 << n 的意义：

1 << n 是位移运算，用于生成 2^n。在这里，n 是数组 nums 的长度。位移运算 1 << n 会将数字 1 向左移动 n 位，结果是 2^n。例如，如果 n = 3，那么 1 << 3 就是 8，即 2^3。

for i in range(1 << n):
这个循环遍历了所有可能的子集。1 << n 是 2^n，所以 i 从 0 到 2^n - 1，每个 i 表示一个不同的子集。

xor_sum = 0
初始化 xor_sum 变量，用于存储当前子集的异或和。

for j in range(n):
遍历数组中的每个元素。

if i & (1 << j):
这个条件检查 i 的第 j 位是否为 1。如果是，说明当前子集包含第 j 个元素。

xor_sum ^= nums[j]
将当前元素的值与 xor_sum 进行异或操作。异或是累积的，表示当前子集的异或和。

total_sum += xor_sum
将当前子集的异或和加到 total_sum 中，累加所有子集的异或和

例子

假设 nums = [1, 2, 3]，我们来看一下 i = 5 时的情况：

转换 i 为二进制：

i = 5，二进制是 101。

检查每一位：

对于 j = 0：
1 << 0 是 0001（即 1）。
i & (1 << 0) 是 101 & 001，结果是 001。结果不为 0，所以第 0 位为 1，我们选择了 nums[0]（即 1）。
对于 j = 1：
1 << 1 是 0010（即 2）。
i & (1 << 1) 是 101 & 010，结果是 000。结果为 0，所以第 1 位为 0，我们没有选择 nums[1]（即 2）。
对于 j = 2：
1 << 2 是 0100（即 4）。
i & (1 << 2) 是 101 & 100，结果是 100。结果不为 0，所以第 2 位为 1，我们选择了 nums[2]（即 3）。

计算异或和：

所以 i = 5 选择了 nums[0] 和 nums[2]，即 [1, 3]。
xor_sum 的计算是 1 ^ 3，结果是 2。