2917.找出数组中的 K-or 值

【LetMeFly】2917.找出数组中的 K-or 值：基础位运算

力扣题目链接：https://leetcode.cn/problems/find-the-k-or-of-an-array/

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

nums 中的 K-or 是一个满足以下条件的非负整数：

• 只有在 nums 中，至少存在 k 个元素的第 i 位值为 1 ，那么 K-or 中的第 i 位的值才是 1 。

返回 nums 的 K-or 值。

注意 ：对于整数 x ，如果 (2i AND x) == 2i ，则 x 中的第 i 位值为 1 ，其中 AND 为按位与运算符。

 

示例 1：

输入：nums = [7,12,9,8,9,15], k = 4
输出：9
解释：nums[0]、nums[2]、nums[4] 和 nums[5] 的第 0 位的值为 1 。
nums[0] 和 nums[5] 的第 1 位的值为 1 。
nums[0]、nums[1] 和 nums[5] 的第 2 位的值为 1 。
nums[1]、nums[2]、nums[3]、nums[4] 和 nums[5] 的第 3 位的值为 1 。
只有第 0 位和第 3 位满足数组中至少存在 k 个元素在对应位上的值为 1 。因此，答案为 2^0 + 2^3 = 9 。

示例 2：

输入：nums = [2,12,1,11,4,5], k = 6
输出：0
解释：因为 k == 6 == nums.length ，所以数组的 6-or 等于其中所有元素按位与运算的结果。因此，答案为 2 AND 12 AND 1 AND 11 AND 4 AND 5 = 0 。

示例 3：

输入：nums = [10,8,5,9,11,6,8], k = 1
输出：15
解释：因为 k == 1 ，数组的 1-or 等于其中所有元素按位或运算的结果。因此，答案为 10 OR 8 OR 5 OR 9 OR 11 OR 6 OR 8 = 15 。

 

提示：

• 1 <= nums.length <= 50
• 0 <= nums[i] < 231
• 1 <= k <= nums.length

方法一：基础位运算

想AC这道题，只需要懂得两个位运算操作：

1. 计算$t$二进制下第$i+1$位是否为$1$：$(t >> i) & 1$
2. 将$ans$二进制下的第$i+1$位置为$1$：$ans |= (1 << i)$

因$0\leq nums[i] \le 2^{31}$，所以用变量$i$从$0$到$30$枚举每一位，统计所有数字中这一位为$1$的个数，若达到$k$则令答案的这一位为$1$。

• 时间复杂度$O(len(nums)\times \log nums[i])$，其中$\log nums[i]=31$
• 空间复杂度$O(1)$

AC代码

C++

class Solution {
public:
    int findKOr(vector<int>& nums, int k) {
        int ans = 0;
        for (int i = 0; i < 32; i++) {  // nums[i] < 2^31不是≤，因此这里其实i = 0到i < 31即可
            int cnt = 0;
            for (int t : nums) {
                cnt += ((t >> i) & 1);
            }
            if (cnt >= k) {
                ans |= (1 << i);
            }
        }
        return ans;
    }
};

Python

# from typing import List

class Solution:
    def findKOr(self, nums: List[int], k: int) -> int:
        ans = 0
        for i in range(31):
            cnt = 0
            for t in nums:
                cnt += ((t >> i) & 1)
            if cnt >= k:
                ans |= (1 << i)
        return ans

同步发文于CSDN和我的个人博客，原创不易，转载经作者同意后请附上原文链接哦~
Tisfy：https://letmefly.blog.csdn.net/article/details/136497896