#include <iostream>
using namespace std;
const int N = 200010;
typedef long long LL;
int a[N];
int n;
LL res;
LL sum1,sum2;
int main()
{
cin >> n;
for(int i = 0;i < n;i++) scanf("%d",&a[i]);
for(int i = 0,j = 0;i < n;i++)
{
sum1 += a[i];
sum2 ^= a[i];
while(j < i && sum1 != sum2) sum1 -= a[j],sum2 ^= a[j],j++;
res += (i-j+1);
}
cout << res;
return 0;
}
代码分析:
变量初始化:
n:数组中元素的数量。
a[N]:存储数组元素的变量。
res:存储最终结果,即满足条件的子数组数量。
sum1:连续子数组的元素和。
sum2:连续子数组的元素异或和。
输入处理:
代码首先读取 n,即数组长度。
然后通过循环读取数组 a 的每个元素。
主循环:
该循环遍历数组元素以计算连续子数组。
i:当前子数组的结束位置。
j:当前子数组的开始位置。
计算子数组的和与异或和:
对于每个位置 i,将当前元素 a[i] 加到 sum1(元素和)上,并与 sum2(异或和)进行异或操作。
调整子数组的开始位置 j:
while 循环用来递增 j,直到 sum1(元素和)等于 sum2(异或和)。
如果 sum1 不等于 sum2,则递减 sum1 并对 sum2 进行异或操作以移除 a[j] 的贡献,同时 j 自增。
更新结果:
在找到每个以 i 结尾的满足条件的子数组后,更新结果 res。由于子数组是连续的,以 i 结尾的满足条件的子数组数量为 (i - j + 1)。
输出结果:
所有元素处理完成后,输出 res,即所有满足元素和等于元素异或和的连续子数组的数量。
关键点:
异或运算有一个特性是,相同的数字异或结果为0,因此如果某段子数组的和等于异或和,说明这段子数组中所有相同数字都出现了偶数次。
异或运算也满足交换律和结合律,这使得在递增 j 时可以轻易地移除 j 处元素的贡献并保持正确的异或和。
该问题利用了双指针技术,也称为滑动窗口方法,它可以在 O(n) 的时间复杂度内解决问题,而不需要使用二维的动态规划解法,后者时间复杂度会高达 O(n^2)。
这个代码片段是一个高效的算法,因为它只需要一次遍历就能得到答案。