原题链接:C. Even Subarrays
题目大意:
给定一段序列,求有多少个连续子区间满足:区间异或和的因子数量为偶数(规定0的因子数量为奇数)。
数据范围:1≤t≤104,2≤n≤2⋅105,1≤ai≤n,保证所有测试案例的n总和不超过2⋅105
解题思路:
前置知识:平方数的因子数量为奇数,其他数的因子数量均为偶数
因此只需要算出有多少个区间的异或和为平方数即可
我们可以先求出前缀异或和s,即si=a1⊕a2⊕a3…⊕ai,然后可用sr⊕sl-1求出区间异或和,进而得到sr⊕sl-1=p⋅p的个数
由异或的性质:a⊕b=c⇔a⊕c=a,可得sr⊕sl-1=p⋅p⇔sr⊕p⋅p=sl-1,从而根据sr⊕p⋅p的数量求出区间的异或和为平方数的数量
由异或的性质,整个数组的异或和不会超过2*n
for(int i=1;i<=n;i++)
{
sum^=a[i];//求出当前s[r]
for(int j=0;j*j<=2*n;j++)
{
int p=(j*j)^sum;
if(p<=2*n)ans+=cnt[p];
}
cnt[sum]++;
}
最后用子区间总数减去异或和为平方数的区间数量即可
AC代码:
#include <bits/stdc++.h>
using namespace std;
#define int long long
int n;
int a[500005],cnt[500005];
void solve(){
memset(cnt,0,sizeof(cnt));
cin>>n;
int sum=0,ans=0;
for(int i=1;i<=n;i++)cin>>a[i];
cnt[sum]++;
for(int i=1;i<=n;i++)
{
sum^=a[i];
for(int j=0;j*j<=2*n;j++)
{
int p=(j*j)^sum;
if(p<=2*n)ans+=cnt[p];
}
cnt[sum]++;
}
cout<<((1+n)*n)/2-ans<<'\n';
}
signed main() {
ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
int t=1;
cin>>t;
while(t--)solve();
return 0;
}