每日coding(2859. 计算 K 置位下标对应元素的和,nums&=nums-1的妙处)

2859. 计算 K 置位下标对应元素的和

给你一个下标从 0 开始的整数数组 nums 和一个整数 k 。

请你用整数形式返回 nums 中的特定元素之  ,这些特定元素满足:其对应下标的二进制表示中恰存在 k 个置位。

整数的二进制表示中的 1 就是这个整数的 置位 。

例如,21 的二进制表示为 10101 ,其中有 3 个置位。

示例 1:

输入:nums = [5,10,1,5,2], k = 1
输出:13
解释:下标的二进制表示是: 
0 = 0002
1 = 0012
2 = 0102
3 = 0112
4 = 1002 
下标 1、2 和 4 在其二进制表示中都存在 k = 1 个置位。
因此,答案为 nums[1] + nums[2] + nums[4] = 13 。

示例 2:

输入:nums = [4,3,2,1], k = 2
输出:1
解释:下标的二进制表示是: 
0 = 002
1 = 012
2 = 102
3 = 112
只有下标 3 的二进制表示中存在 k = 2 个置位。
因此,答案为 nums[3] = 1 。

提示:

  • 1 <= nums.length <= 1000
  • 1 <= nums[i] <= 105
  • 0 <= k <= 10

这道题目主要是学习两点东西哈,一个是lambda函数,cadn有不少介绍的,但是呢我问了一下文心一言,是这么介绍的

Lambda函数,或匿名函数,在许多编程语言中都是一个非常有用的特性,特别是在Python中。以下是Lambda函数的一些主要优点:

  1. 简洁性:Lambda函数允许你快速定义一个简单的函数,而不需要使用传统的def语句。这对于一些小操作或一次性操作特别有用。
  2. 即时定义:由于Lambda函数是即时定义的,所以在你只需要一个简单的函数定义,而不需要在代码的其他部分多次引用它时,Lambda函数特别有用。
  3. 匿名性:由于它们没有名称,Lambda函数在需要一个函数作为参数传递给另一个函数,但不需要在代码的其他地方引用该函数时,是非常有用的。
  4. 辅助功能:在编写更复杂的函数或算法时,Lambda函数可以作为辅助功能来使用,传递给其他函数。
  5. 可读性:虽然简洁性可能导致一些人认为它们难以阅读,但在许多情况下,Lambda函数实际上可以使代码更清晰和更易于理解。当一个函数的操作非常简单时,使用Lambda函数可以使代码更加简洁。
  6. 灵活性:由于Lambda函数可以非常快速地定义和修改,因此它们在需要快速迭代和修改算法或函数时特别有用。

lambda函数的使用C++——Lambda函数_c++拉姆达函数-CSDN博客

第二个话题,nums &= nums - 1,这个主要是求出

num&(num-1)的用途_num & (num-1)-CSDN博客

class Solution {
public:
    int sumIndicesWithKSetBits(vector<int>& nums, int k) {
        auto bitCount = [](int x) {
            int cnt = 0;
            while (x) {
                cnt += (x % 2);
                x /= 2;
            }
            return cnt;
        };

        int ans = 0;
        for (int i = 0; i < nums.size(); ++i) {
            if (bitCount(i) == k) {
                ans += nums[i];
            }
        }
        return ans;
    }
};
class Solution {
public:
    int sumIndicesWithKSetBits(vector<int>& nums, int k) {
        // 统计位1个数
        auto bitCount = [](int num){
            int c = 0;
            while(num > 0){
                num &= num - 1;
                c++;
            }
            return c;
        };  

        int n = nums.size();
        int sum = 0;
        // 枚举每个下标,如果下标置位数等于k,累计对应的数值
        for(int i = 0; i < nums.size(); i++){
            if(bitCount(i) == k)
                sum += nums[i];
        }
        return sum;
    }
};

  • 19
    点赞
  • 23
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值