1486. 数组异或操作

Problem#

给你两个整数 n 和 start。

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

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

https://leetcode.cn/problems/xor-operation-in-an-array/description/

示例 1：

输入：n = 5, start = 0 输出：8 解释：nums = [0, 2, 4, 6, 8]，0 ^ 2 ^ 4 ^ 6 ^ 8 = 8

示例 2：

输入：n = 4, start = 3 输出：8 解释：nums = [3, 5, 7, 9]，3 ^ 5 ^ 7 ^ 9 = 8

约束：

1 <= n <= 1000
0 <= start <= 1000

思路#

循环生成每个元素并累加异或即可。也可以用数学公式优化，但 n 很小，直接循环最清晰。

1
class Solution:
2
    def xorOperation(self, n: int, start: int) -> int:
3
        result = 0
4
        for i in range(n):
5
            result ^= start + 2 * i
6
        return result

1
import unittest
2

3

4
class TestXorOperation(unittest.TestCase):
5
    def setUp(self):
6
        self.solution = Solution()
7

8
    def test_example1(self):
9
        self.assertEqual(self.solution.xorOperation(5, 0), 8)
10

11
    def test_example2(self):
12
        self.assertEqual(self.solution.xorOperation(4, 3), 8)
13

14
    def test_single(self):
15
        self.assertEqual(self.solution.xorOperation(1, 7), 7)
16

17
    def test_large(self):
18
        self.assertEqual(self.solution.xorOperation(10, 5), 2)
19

20

21
if __name__ == '__main__':
22
    unittest.main()