用 Python 解锁位运算:统计特定位数下标的数组元素之和
在算法的世界中,位运算虽然看起来偏底层,却往往是高效解决问题的利器。今天我们来聊一个简单但颇具技巧性的问题:如何从数组中找出下标的二进制表示中有恰好 k 个置位(即 1)的那些元素,并求它们的和?
这是一道结合数组遍历和位运算技巧的经典题目,适合用于面试中展示你对 Python 特性的熟练掌握,尤其是 Python 3.10+ 中引入的 bit_count() 方法。
🧩 题目描述
给你一个下标从 0 开始的整数数组 nums 和一个整数 k。
请你返回 nums 中所有满足其下标二进制表示中恰好有 k 个置位的元素之和。
示例 1:
输入:nums = [5,10,1,5,2], k = 1
输出:13
解释:
下标的二进制分别为:
0 = 000
1 = 001 (1 个置位)
2 = 010 (1 个置位)
3 = 011 (2 个置位)
4 = 100 (1 个置位)
符合条件的下标:1、2、4
结果为 nums[1] + nums[2] + nums[4] = 10 + 1 + 2 = 13
示例 2:
输入:nums = [4,3,2,1], k = 2
输出:1
解释:
下标 3 的二进制为 011,包含两个置位
返回 nums[3] =

最低0.47元/天 解锁文章
1186

被折叠的 条评论
为什么被折叠?



