[AtCoder 4142]Xor Sum 2

AtCoder 4142

分类:

枚举,尺取法

大意:

给定一个长为n的数列,求有多少区间的xor之和等于区间的数的和(xor指异或)

时空限制:

2000ms/1048576kB

输入格式:

第1行输入一个正整数n,表示有n个数

第2行输入n个数,分别为a[1],a[2]…a[n]

输出格式:

1行，1个数字表示有多少区间的xor之和等于区间的数的和

数据范围:

1<=n<=2*10⁵,0<=a[i]<=2²⁰

---

接下来讲思路
枚举i从1到n
定义一个j只往后推动
每次如果sum与Xor分别与a[j]操作后相等则进行操作
之后如果sum和Xor相等,则答案加上j-i+1即sum或者说Xor长度(j-i+1是指包括a[i]的连续几个数,例如长度为3即a[i],a[i]+a[i+1],a[i]+a[i+1]+a[i+2])
再然后每次把a[i]去掉
最后输出答案即可

#include<bits/stdc++.h>
using namespace std;
signed main(){
	int n;
	scanf("%d",&n);
	int a[n+1];
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	int sum=0,Xor=0,j=0,ans=0;
	for(int i=1;i<=n;i++){
		while(j<n&&((sum+a[j+1])==(Xor^a[j+1]))){
			sum+=a[j+1];
			Xor^=a[j+1];
			j++;
		}
		if(sum==Xor){
			ans+=j-i+1;
		}
		sum-=a[i];
		Xor^=a[i];
	}
	printf("%d\n",ans);
    return 0;
}