JAVA程序设计:子数组按位或操作(LeetCode:898)

我们有一个非负整数数组 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 <= A.length <= 50000
0 <= A[i] <= 10^9

思路:容易想到的方法是对于A[i],将以A[i]结尾的连续子数组能组成的不同答案数加入答案,并保证不重复添加即可,但是由于复杂度为O(n^2)使得我们不得不想更优的方法,可以想到的的是若以A[i]结尾,往前枚举起点j,对于A[j]|A[j+1]|...|A[i]随着j的减小而单调不减。因此我们考虑对于每个i,将以A[i]结尾的子数组能组成的答案存入set,这是以A[i]结尾的所有或的结果,若此时i=i+1,我们只需要用A[i+1]与A[i]结尾组成的答案集合进行异或即可,而由于A[i]的大小不超过1e9,因此其二进制表示不会超过32位,所以说集合Set的尺寸并不会超过32。

class Solution {
    public int subarrayBitwiseORs(int[] A) {
    	
    	Set ans=new HashSet<>();
    	Set cur=new HashSet<>();
    	
    	cur.add(0);
    	for(int x : A) {
    		Set cur2=new HashSet<>();
    		for(int y : cur)
    			cur2.add(x|y);
    		cur2.add(x);
    		cur=cur2;
    		ans.addAll(cur);
    	}
    	
    	return ans.size();
    }
}

 

你可能感兴趣的:(JAVA程序设计:子数组按位或操作(LeetCode:898))