题目描述
给你一个下标从 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、数组元素值<2^31,所以我们枚举0到30的每个比特位。
2、遍历数组,如果第i个比特位上的1的个数>=k,则把2^i加到答案中
复杂度分析
时间复杂度:O(nlogU),其中n为nums的长度,U = max(nums)
空间复杂度:O(1)
代码
#include <vector>
#include <iostream>
using namespace std;
class Solution {
public:
int findKOr(vector<int>& nums, int k) {
int ans = 0;
for (int i = 0; i < 31; ++i) {
int cnt = 0;
for (int num: nums) {
if ((num >> i) & 1) {
++cnt;
}
}
if (cnt >= k) {
ans |= 1 << i;
}
}
return ans;
}
};
int main()
{
Solution solution;
vector<int> nums = {7, 12, 9, 8, 9, 15};
int k = 4;
int res = solution.findKOr(nums, k);
cout << res << endl;
nums = {2, 12, 1, 11, 4, 5};
k = 6;
res = solution.findKOr(nums, k);
cout << res << endl;
nums = {10, 8, 5, 9, 11, 6, 8};
k = 1;
res = solution.findKOr(nums, k);
cout << res << endl;
}
感兴趣的可以给博主点个赞,关注支持一下!到1000粉丝随机抽奖大红包哦~