2411. 按位或最大的最小子数组长度

https://leetcode.cn/problems/smallest-subarrays-with-maximum-bitwise-or/description/

给你一个长度为 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:

1
2
3
4
5
6
7
8
9
10
输入: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:

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

提示:

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

首先上场的还是我们的暴力破解法,毕竟大力出奇迹嘛。

按照题目要求,遍历 nums 对于每一个 nums[i] 再次遍历 nums[i]…nums[nums.length - 1] 。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
class Solution {
public int[] smallestSubarrays(int[] nums) {
int[] ans = new int[nums.length];
for (int i = 0; i < nums.length; i++) {
ans[i] = 1;
int max = nums[i];
int temp = nums[i];
for (int j = i + 1; j < nums.length; j++) {
temp |= nums[j];
if (temp > max) {
max = temp;
ans[i] = j - i + 1;
}
}
}
return ans;
}
}

当然,还有一种方式,遍历 nums ,再从 i - 1 逆向遍历 nums[j] ,如果 (nums[i] | nums[j]) > nums[j] 则更新 nums[j] = nums[j] | nums[i]ans[j] = i - j + 1

1
2
3
4
5
6
7
8
9
10
11
12
13
class Solution {
public int[] smallestSubarrays(int[] nums) {
int[] ans = new int[nums.length];
for (int i = 0; i < nums.length; i++) {
ans[i] = 1;
for (int j = i - 1; j >= 0 && (nums[i] | nums[j]) > nums[j]; j--) {
nums[j] |= nums[i];
ans[j] = i - j + 1;
}
}
return ans;
}
}
0%