加载中...
898-子数组按位或操作(Bitwise ORs of Subarrays)
发表于:2021-12-03 | 分类: 中等
字数统计: 1.1k | 阅读时长: 5分钟 | 阅读量:

原文链接: https://leetcode-cn.com/problems/bitwise-ors-of-subarrays

英文原文

We have an array arr of non-negative integers.

For every (contiguous) subarray sub = [arr[i], arr[i + 1], ..., arr[j]] (with i <= j), we take the bitwise OR of all the elements in sub, obtaining a result arr[i] | arr[i + 1] | ... | arr[j].

Return the number of possible results. Results that occur more than once are only counted once in the final answer

 

Example 1:

Input: arr = [0]
Output: 1
Explanation: There is only one possible result: 0.

Example 2:

Input: arr = [1,1,2]
Output: 3
Explanation: The possible subarrays are [1], [1], [2], [1, 1], [1, 2], [1, 1, 2].
These yield the results 1, 1, 2, 1, 3, 3.
There are 3 unique values, so the answer is 3.

Example 3:

Input: arr = [1,2,4]
Output: 6
Explanation: The possible results are 1, 2, 3, 4, 6, and 7.

 

Constraints:

  • 1 <= nums.length <= 5 * 104
  • 0 <= nums[i] <= 109

中文题目

我们有一个非负整数数组 A

对于每个(连续的)子数组 B = [A[i], A[i+1], ..., A[j]] ( i <= j),我们对 B 中的每个元素进行按位或操作,获得结果 A[i] | A[i+1] | ... | A[j]

返回可能结果的数量。 (多次出现的结果在最终答案中仅计算一次。)

 

示例 1:

输入:[0]
输出:1
解释:
只有一个可能的结果 0 。

示例 2:

输入:[1,1,2]
输出:3
解释:
可能的子数组为 [1],[1],[2],[1, 1],[1, 2],[1, 1, 2]。
产生的结果为 1,1,2,1,3,3 。
有三个唯一值,所以答案是 3 。

示例 3:

输入:[1,2,4]
输出:6
解释:
可能的结果是 1,2,3,4,6,以及 7 。

 

提示:

  1. 1 <= A.length <= 50000
  2. 0 <= A[i] <= 10^9

通过代码

官方题解

方法一:集合

分析

显然,最简单的方法就是枚举所有满足 i <= j(i, j),并计算出不同的 result(i, j) = A[i] | A[i + 1] | ... | A[j] 的数量。由于 result(i, j + 1) = result(i, j) | A[j + 1],因此我们可以在 $O(N^2)$ 的时间复杂度计算出所有的 result(i, j),其中 $N$ 是数组 A 的长度。

我们尝试优化一下这种最简单的枚举方法。可以发现,对于固定的 jresult(j, j), result(j - 1, j), result(j - 2), j, ..., result(1, j) 的值是单调不降的,因为将 result(k, j)A[k - 1] 做按位或操作,得到的结果 result(k - 1, j) 一定不会变小。并且,根据按位或操作的性质,如果把 result(k, j)result(k - 1, j) 都用二进制表示,那么后者将前者二进制表示中的若干个 0 变成了 1

由于数组 A 中都是小于 10^9 的正整数,它们的二进制表示最多只有 32 位。因此从 result(j, j) 开始到 result(1, j) 结束,最多只会有 320 变成了 1,也就是说,result(j, j), result(j - 1, j), result(j - 2), j, ..., result(1, j) 中最多只有 32 个不同的数。这样我们就可以维护一个集合,存储所有以 j 为结尾的 result 值。当结尾从 j 枚举到 j + 1 时,我们将集合中的每个数对 A[j + 1] 做按位或操作,得到的新的 result 值也不会超过 32 个。

算法

我们用一个集合 cur 存储以 j 为结尾的 result 值,即所有满足 i <= jA[i] | ... | A[j] 的值。集合的大小不会超过 32

[sol1]
class Solution { public int subarrayBitwiseORs(int[] A) { Set<Integer> ans = new HashSet(); Set<Integer> cur = new HashSet(); cur.add(0); for (int x: A) { Set<Integer> cur2 = new HashSet(); for (int y: cur) cur2.add(x | y); cur2.add(x); cur = cur2; ans.addAll(cur); } return ans.size(); } }
[sol1]
class Solution(object): def subarrayBitwiseORs(self, A): ans = set() cur = {0} for x in A: cur = {x | y for y in cur} | {x} ans |= cur return len(ans)

复杂度分析

  • 时间复杂度:$O(N \log W)$,其中 $N$ 是数组 A 的长度,$W$ 是 A 中最大的数。

  • 空间复杂度:$O(N \log W)$。在给出的代码中用集合 ans 存放了所有答案,会使用 $O(N \log W)$ 的空间。但这题只要返回答案的数量,因此可以只用一个变量对集合 cur 的大小进行累加,这样空间复杂度可以降低为 $O(\log W)$。

统计信息

通过次数 提交次数 AC比率
5592 16468 34.0%

提交历史

提交时间 提交结果 执行时间 内存消耗 语言
上一篇:
894-所有可能的满二叉树(All Possible Full Binary Trees)
下一篇:
897-递增顺序搜索树(Increasing Order Search Tree)
本文目录
本文目录