【洛谷】P5057 简单题

题目地址：

https://www.luogu.com.cn/problem/P5057

题目描述：
有一个$n$个元素的数组，每个元素初始均为$0$。有$m$条指令，要么让其中一段连续序列数字反转——$0$变$1$，$1$变$0$（操作$1$），要么询问某个元素的值（操作$2$）。例如当$n=20$时，$10$条指令如下：

输入格式：
第一行包含两个整数$n,m$，表示数组的长度和指令的条数；以下$m$行，每行的第一个数$t$表示操作的种类：
若$t=1$，则接下来有两个数$L,R$，表示区间$[L,R]$的每个数均反转；若$t=2$，则接下来只有一个数$i$，表示询问的下标。

输出格式：
每个操作$2$输出一行（非$0$即$1$），表示每次操作$2$的回答。

数据范围：
对于$50\%$的数据，$1\le n\le 10^3$，$1\le m\le 10^4$；
对于$100\%$的数据，$1\le n\le 10^5$，$1\le m\le 5\times 10^5$，保证$L\le R$。

可以考虑该数组的异或差分数组，这样两个操作就转为单点修改和求前缀和，可以用树状数组来做。代码如下：

#include <iostream>
using namespace std;

const int N = 1e5 + 10;
int n, m;
int tr[N];

int lowbit(int x) {
  return x & -x;
}

void add(int k, int x) {
  for (; k <= n; k += lowbit(k)) tr[k] ^= x;
}

int sum(int k) {
  int res = 0;
  for (; k; k -= lowbit(k)) res ^= tr[k];
  return res;
}

int main() {
  scanf("%d%d", &n, &m);
  while (m--) {
    int t, l, r, k;
    scanf("%d", &t);
    if (t == 1) {
      scanf("%d%d", &l, &r);
      add(r + 1, 1), add(l, 1);
    } else {
      scanf("%d", &k);
      printf("%d\n", sum(k));
    }
  }
}

每次操作时间复杂度$O(\log n)$，空间$O(n)$。