3097.或值至少为K的最短子数组II
https://leetcode.cn/problems/shortest-subarray-with-or-at-least-k-ii/
给你一个 非负 整数数组 nums 和一个整数 k 。
如果一个数组中所有元素的按位或运算 OR 的值 至少 为 k ,那么我们称这个数组是 特别的 。
请你返回 nums 中 最短特别非空 子数组的长度,如果特别子数组不存在,那么返回 -1 。
示例 1:
输入:nums = [1,2,3], k = 2
输出:1
解释:
子数组 [3] 的按位 OR 值为 3 ,所以我们返回 1 。
示例 2:
输入:nums = [2,1,8], k = 10
输出:3
解释:
子数组 [2,1,8] 的按位 OR 值为 11 ,所以我们返回 3 。
示例 3:
输入:nums = [1,2], k = 0
输出:1
解释:
子数组 [1] 的按位 OR 值为 1 ,所以我们返回 1 。
提示:
1 <= nums.length <= 2 * 1050 <= nums[i] <= 1090 <= k <= 109
python
from typing import List
class Solution:
def minimumSubarrayLength(self, nums: List[int], k: int) -> int:
bits = [0] * 30
minv = float('inf')
def sum_bits():
return sum([1 << i for i in range(30) if bits[i] > 0])
left = 0
for right in range(len(nums)):
for i in range(30):
if nums[right] & (1 << i):
bits[i] += 1
while left <= right and sum_bits() >= k:
minv = min(minv, right - left + 1)
for i in range(30):
if nums[left] & (1 << i):
bits[i] -= 1
left += 1
return minv if minv != float('inf') else -1