题目
给出n个数字 a_1,…,a_n,问最多有多少不重叠的非空区间,使得每个区间内数字的xor(异或和)都等于0
假设输入:
1,2,3,3,2,1
输出:
2
解释:
如果拆成两部分来看
[1,2,3] [3,2,1] 那么不重叠非空区间有两个,并且xor都为0,那么结果为2
[1,2] [3,3] [2,1] 那么不重叠非空区间有三个,但是只有[3,3]的xor为0,结果为1
我们要输出多的那个拆法,那么肯定是输出2
分析
这道题用动态规划来解
设当前指针指向的元素最下标为i,那么有下面两种情况
1、假设i不参与xor,那么[0…i]之间满足条件的区间的最大个数和[0…i-1]之间的一样,因为i不参与前面的xor,并且自己没办法xor
2、假设i参与xor,并且i所在的区间xor为0,设这个区间的起始下标为k,并且k为从i向前找到的离i最近的起始位置(就是这个区间是包含i并且xor为0的所有区间中的最短区间,k为这个区间的起始位置),同时假设我们知道[0…k-1]之间的最大个数为m个,那么整个数组[0…i]位置可满足条件的区间的最大个数为m+1个
知道以上两种情况之后,我们让 i
从 0
出发遍历下面的数组,并且使用dp[i]
记录 i
位置的最大满足条件的区间个数,并且设第一个xor为0的位置为-1,从i向前找k的时候,只要k-1为-1,那么dp[i]就记录为1
1,2,3,3,2,1
1、当 i = 0 的时候,xor 为不为零,并且为起始位置也不用向前找,所以dp[0]=0
2、当 i = 1 的时候,有两种情况,
- 第一种:假设2参与前面的xor,那么此时
dp[1]
应该等于dp[k-1] + 1
,前面一直到头也没有找到k,所以dp[1] = 0 - 第二种:假设2不参与前面的xor,那么
dp[1] = dp[1-1]
,还是等于0 - 将两种情况都考虑下来,取最大值存到dp[1],其实还是0
3、当 i = 2 的时候,还是上面两种情况
- 第一种:假设3参与前面的xor,那么此时
dp[2]
应该等于dp[k-1] + 1
,找到头发现k为0,那么k-1等于-1,此时记录dp[2] = 1
- 第二种:假设3不参与前面的xor,那么
dp[2] = dp[2-1] = 0
- 从两种情况中保留最大值给dp[2],那么此时dp[2] 为 1
依次类推直到 i 指向最后一个元素
那么如何找到k的位置呢,其实我们不需要特意去找k的位置,我们只要找k-1的位置就可以了,从第0个元素逐个地向后xor,上次出现相同xor的位置就是k-1的位置,所以我们刚开始要记录-1的位置为xor等于0,那么这样的话就有一个集合特别适合存储这种记录,HashMap,key为xor,value为出现的位置,每次xor都记录一下,出现相同xor的时候就更新
实现
原理和思路都讲完了,上代码
public static int mostEOR(int arr[]){
if(arr == null || arr.length == 0){
return 0;
}
//记录当前位置之前满足条件的区间个数
int dp[] = new int[arr.length];
//记录xor上次出现的位置
Map<Integer,Integer> map = new HashMap<>();
map.put(0, -1);//设第一次xor的位置为-1
int lastXor = 0;
for (int i = 0; i < arr.length; i++){
lastXor ^= arr[i];
if(map.containsKey(lastXor)){
int pre = map.get(lastXor);
//如果上次出现xor=0的位置为-1,那么当前位置直接记录为1,否则为dp[k-1] + 1
dp[i] = pre == -1 ? 1 : (dp[pre] + 1);
}
//当i大于0的时候,选dp[i]和dp[i-1]中最大值保留到dp[i]上
if(i > 0){
dp[i] = Math.max(dp[i-1], dp[i]);
}
//记录每个xor的值最晚出现的位置
map.put(lastXor, i);
}
return dp[arr.length - 1];
}