Skip to content

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 * 105
  • 0 <= nums[i] <= 109
  • 0 <= 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