Question

1486. 数组异或操作

English Version

题目描述

给你两个整数，n 和 start 。

数组 nums 定义为：nums[i] = start + 2*i（下标从 0 开始）且 n == nums.length 。

请返回 nums 中所有元素按位异或（XOR）后得到的结果。

 

示例 1：

输入：n = 5, start = 0
输出：8
解释：数组 nums 为 [0, 2, 4, 6, 8]，其中 (0 ^ 2 ^ 4 ^ 6 ^ 8) = 8 。
   "^" 为按位异或 XOR 运算符。

示例 2：

输入：n = 4, start = 3
输出：8
解释：数组 nums 为 [3, 5, 7, 9]，其中 (3 ^ 5 ^ 7 ^ 9) = 8.

示例 3：

输入：n = 1, start = 7
输出：7

示例 4：

输入：n = 10, start = 5
输出：2

 

提示：

• 1 <= n <= 1000
• 0 <= start <= 1000
• n == nums.length

解法

方法一：模拟

我们可以直接模拟算出数组中所有元素的异或结果。

时间复杂度 $O(n)$，空间复杂度 $O(1)$。其中 $n$ 为数组长度。

class Solution:
  def xorOperation(self, n: int, start: int) -> int:
    ans = 0
    for i in range(n):
      ans ^= start + 2 * i
    return ans

class Solution {
  public int xorOperation(int n, int start) {
    int ans = 0;
    for (int i = 0; i < n; ++i) {
      ans ^= start + 2 * i;
    }
    return ans;
  }
}

class Solution {
public:
  int xorOperation(int n, int start) {
    int ans = 0;
    for (int i = 0; i < n; ++i) {
      ans ^= start + 2 * i;
    }
    return ans;
  }
};

func xorOperation(n int, start int) (ans int) {
  for i := 0; i < n; i++ {
    ans ^= start + 2*i
  }
  return
}

方法二

class Solution:
  def xorOperation(self, n: int, start: int) -> int:
    return reduce(xor, ((start + 2 * i) for i in range(n)))