LeetCode 2411.按位或最大的最小子数组长度:一次倒序遍历

【LetMeFly】2411.按位或最大的最小子数组长度:一次倒序遍历

力扣题目链接:https://leetcode.cn/problems/smallest-subarrays-with-maximum-bitwise-or/

给你一个长度为 n 下标从 0 开始的数组 nums ,数组中所有数字均为非负整数。对于 0n - 1 之间的每一个下标 i ,你需要找出 nums 中一个 最小 非空子数组,它的起始位置为 i (包含这个位置),同时有 最大按位或 运算值

  • 换言之,令 Bij 表示子数组 nums[i...j] 的按位或运算的结果,你需要找到一个起始位置为 i 的最小子数组,这个子数组的按位或运算的结果等于 max(Bik) ,其中 i <= k <= n - 1

一个数组的按位或运算值是这个数组里所有数字按位或运算的结果。

请你返回一个大小为 n 的整数数组answer,其中answer[i]是开始位置为 i ,按位或运算结果最大,且 最短 子数组的长度。

子数组 是数组里一段连续非空元素组成的序列。

示例 1:

复制代码
输入:nums = [1,0,2,1,3]
输出:[3,3,2,2,1]
解释:
任何位置开始,最大按位或运算的结果都是 3 。
- 下标 0 处,能得到结果 3 的最短子数组是 [1,0,2] 。
- 下标 1 处,能得到结果 3 的最短子数组是 [0,2,1] 。
- 下标 2 处,能得到结果 3 的最短子数组是 [2,1] 。
- 下标 3 处,能得到结果 3 的最短子数组是 [1,3] 。
- 下标 4 处,能得到结果 3 的最短子数组是 [3] 。
所以我们返回 [3,3,2,2,1] 。

示例 2:

复制代码
输入:nums = [1,2]
输出:[2,1]
解释:
下标 0 处,能得到最大按位或运算值的最短子数组长度为 2 。
下标 1 处,能得到最大按位或运算值的最短子数组长度为 1 。
所以我们返回 [2,1] 。

提示:

  • n == nums.length
  • 1 <= n <= 105
  • 0 <= nums[i] <= 109

解题方法:倒序遍历

解题思路

或运算的性质决定每一位是可以分开来单独看的。假设我们只关注最低的一位,那么问题就转化为了:

从nums[i]开始到哪个元素为止最低的一位出现过 1 1 1。

一旦最低位出现了 1 1 1,子数组的最短长度就确定了,就没必要继续往后遍历了。

现在nums的范围是 1 0 9 10^9 109,二进制下一共有31位,每一位互不影响,单独来看,求最长的子数组就好了。

具体方法

对于某一位(以二进制下的最低位为例),如何判断从nums[i]开始,第一个这一位是1的元素呢?

只需要从后往前遍历,使用一个变量lastPos记录最后一次这一位为1的下标(即这位为1的最小下标),则遍历到nums[i]时,lastPos的值就是从nums[i]开始第一个这一位为1的元素下标。

特别的,我们可以将lastPos的初始值设置为0;由于nums[i]的范围是 1 0 9 10^9 109,所以我们需要一个长度为 31 31 31的lastPos数组,nums[i]开始的最短数组的长度为31位中结果最长的那个。

时空复杂度

  • 时间复杂度 O ( l e n ( n u m s ) × C ) O(len(nums)\times C) O(len(nums)×C),其中 C = log ⁡ max ⁡ ( n u m s [ i ] ) = log ⁡ 1 0 9 = 31 C=\log \max (nums[i])=\log10^9=31 C=logmax(nums[i])=log109=31
  • 空间复杂度 O ( C ) O(C) O(C)

AC代码

C++
cpp 复制代码
/*
 * @Author: LetMeFly
 * @Date: 2025-07-29 23:42:57
 * @LastEditors: LetMeFly.xyz
 * @LastEditTime: 2025-07-30 00:00:01
 */
#if defined(_WIN32) || defined(__APPLE__)
#include "_[1,2]toVector.h"
#endif

class Solution {
public:
    vector<int> smallestSubarrays(vector<int>& nums) {
        vector<int> ans(nums.size());
        vector<int> lastPos(31);
        for (int i = nums.size() - 1; i >= 0; i--) {
            int last = i;
            for (int j = 0; j < 31; j++) {
                if (nums[i] >> j & 1) {
                    lastPos[j] = i;
                } else {
                    last = max(last, lastPos[j]);
                }
            }
            ans[i] = max(ans[i], last - i + 1);
        }
        return ans;
    }
};


#if defined(_WIN32) || defined(__APPLE__)
/*
 [1,0,2,1,3]
*/
int main() {
    string s;
    while (cin >> s) {
        vector<int> v = stringToVector(s);
        Solution sol;
        debug(sol.smallestSubarrays(v));
    }
    return 0;
}
#endif
Python
python 复制代码
'''
Author: LetMeFly
Date: 2025-07-29 23:42:57
LastEditors: LetMeFly.xyz
LastEditTime: 2025-07-30 10:14:12
'''
from typing import List

class Solution:
    def smallestSubarrays(self, nums: List[int]) -> List[int]:
        ans = [0] * len(nums)
        lastPos = [0] * 31
        for i in range(len(nums) - 1, -1, -1):
            last = i
            for j in range(31):
                if nums[i] >> j & 1:
                    lastPos[j] = i
                else:
                    last = max(last, lastPos[j])
            ans[i] = max(ans[i], last - i + 1)
        return ans
Java
java 复制代码
/*
 * @Author: LetMeFly
 * @Date: 2025-07-29 23:42:57
 * @LastEditors: LetMeFly.xyz
 * @LastEditTime: 2025-07-31 10:45:02
 */
class Solution {
    public int[] smallestSubarrays(int[] nums) {
        int[] ans = new int[nums.length];
        int[] lastPos = new int[31];
        for (int i = nums.length - 1; i >= 0; i--) {
            int last = i;
            for (int j = 0; j < 31; j++) {
                if ((nums[i] >> j & 1) == 1) {
                    lastPos[j] = i;
                } else {
                    last = Math.max(last, lastPos[j]);
                }
            }
            ans[i] = Math.max(ans[i], last - i + 1);
        }
        return ans;
    }
}
Go
go 复制代码
/*
 * @Author: LetMeFly
 * @Date: 2025-07-29 23:42:57
 * @LastEditors: LetMeFly.xyz
 * @LastEditTime: 2025-07-30 10:15:13
 */
package main

func smallestSubarrays(nums []int) []int {
    ans := make([]int, len(nums))
    lastPos := make([]int, 31)
    for i := len(nums) - 1; i >= 0; i-- {
        last := i
        for j := 0; j < 31; j++ {
            if nums[i] >> j & 1 == 1 {
                lastPos[j] = i
            } else {
                last = max(last, lastPos[j])
            }
        }
        ans[i] = max(ans[i], last - i + 1)
    }
    return ans
}
Rust
rust 复制代码
/*
 * @Author: LetMeFly
 * @Date: 2025-07-29 23:42:57
 * @LastEditors: LetMeFly.xyz
 * @LastEditTime: 2025-07-30 21:00:40
 */
use std::cmp::max;

impl Solution {
    pub fn smallest_subarrays(nums: Vec<i32>) -> Vec<i32> {
        let mut ans: Vec<i32> = vec![0; nums.len()];
        let mut lastPos: Vec<i32> = vec![0; 31];
        for i in (0..ans.len()).rev() {
            let ii32: i32 = i.try_into().unwrap();
            let mut last: i32 = ii32;
            for j in (0..31) {
                if nums[i] >> j & 1 == 1 {
                    lastPos[j] = ii32;
                } else {
                    last = max(last, lastPos[j])
                }
            }
            ans[i] = max(ans[i], last - ii32 + 1)
        }
        ans
    }
}

同步发文于CSDN和我的个人博客,原创不易,转载经作者同意后请附上原文链接哦~

千篇源码题解已开源

相关推荐
草莓熊Lotso31 分钟前
【数据结构初阶】--排序(一):直接插入排序,希尔排序
c语言·数据结构·经验分享·其他·排序算法
小拇指~1 小时前
梯度下降的基本原理
人工智能·算法·计算机视觉
小徐不徐说1 小时前
C++ 模板与 STL 基础入门:从泛型编程到实战工具集
开发语言·数据结构·c++·qt·面试
艾莉丝努力练剑1 小时前
【C/C++】类和对象(上):(一)类和结构体,命名规范——两大规范,新的作用域——类域
java·c语言·开发语言·c++·学习·算法
TDengine (老段)2 小时前
TDengine 中 TDgp 中添加机器学习模型
大数据·数据库·算法·机器学习·数据分析·时序数据库·tdengine
2202_756749693 小时前
04 基于sklearn的机械学习-梯度下降(上)
人工智能·算法·机器学习
草莓爱芒果3 小时前
Spring Boot中使用Bouncy Castle实现SM2国密算法(与前端JS加密交互)
java·spring boot·算法
晚云与城3 小时前
【数据结构】-----排序的艺术画卷
数据结构·算法·排序算法