2020牛客寒假算法基础集训营4 D.子段异或

题目地址

题目描述

输入一个数列a，你需要输出其中异或值为0的不同子段的数量。一个子段 [l,r] (1 ≤ l ≤ r ≤ n)的异或值为

a[l]⊕a[l+1]⊕a[l+2]⊕…⊕a[r]，其中⊕符号代表异或运算。

两个子段被视为相同的，当且仅当其开始和结束位置均对应相同。

输入描述

第一行一个整数 n ，代表数列长度。
第二行 n 个整数，代表数列。

输出描述

输出一个整数，代表答案。

示例1

输入

5

1 2 3 2 1

输出

2

说明

子段 [1,3] 和子段 [3,5] 是合法子段。

备注

n ≤ 200000, 0 ≤ a[i] ≤ 2^30−1

题解

设[l,r]=a[l]⊕a[l+1]⊕a[l+2]⊕…⊕a[r-1]⊕a[r]

我们知道相同的数，异或的结果为0，所以[1,l-1]⊕[1,r]=[l,r] (r>l)。

题目要求[l,r]=0，所以这时[1,l-1]=[1,r]。

我们从1一直异或到n，用map记录一下再相加就可以得到答案了。

代码如下

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <queue>
#include <cmath>
#include <cctype>
#include <stack>
#include <map>
#include <cstring>
#include <sstream>
#include <set>
#include <list>
#include <fstream>
#include <iomanip>
#include <assert.h>

#define ll long long
#define ull unsigned long long
#define pii std::pair<int, int>
#define op                            \
    std::ios::sync_with_stdio(false); \
    std::cin.tie(0);

const int INF = 0x3f3f3f3f;
const int maxn = (200000 + 5);
const int mod = (1e9 + 7);

std::map<ll,int> m;
ll a[maxn],ans=0,sum=0;

int main()
{
    int n;
    scanf("%d",&n);
    for(int i=0;i<n;i++)
        scanf("%d",&a[i]);
    m[0]=1;
    for(int i=0;i<n;i++)
    {
        sum^=a[i];
        ans+=m[sum];
        m[sum]++;
    }
    printf("%lld",ans);
    return 0;
}