LeetCode 1310 子数组异或查询，Java解法参考

题目描述

难度： 中等

有一个正整数数组 arr，现给你一个对应的查询数组 queries，其中 queries[i] = [Li, Ri]。

对于每个查询 i，请你计算从 Li 到 Ri 的 XOR 值（即 arr[Li] xor arr[Li+1] xor … xor arr[Ri]）作为本次查询的结果。

并返回一个包含给定查询 queries 所有结果的数组。

示例 1：

输入： arr = [1,3,4,8], queries = [[0,1],[1,2],[0,3],[3,3]]
输出： [2,7,14,8]
解释：
数组中元素的二进制表示形式是：
1 = 0001
3 = 0011
4 = 0100
8 = 1000
查询的 XOR 值为：
[0,1] = 1 xor 3 = 2
[1,2] = 3 xor 4 = 7
[0,3] = 1 xor 3 xor 4 xor 8 = 14
[3,3] = 8

示例 2：

输入： arr = [4,8,2,10], queries = [[2,3],[1,3],[0,0],[0,3]]
输出： [8,0,4,4]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/xor-queries-of-a-subarray

思路 1 （两层迭代）

思路

异或运算是基于二进制的位运算，Java中的异或运算可以使用^。

简单的思路是运用两层循环，外层为结果的个数，内层为Li至Ri。结果耗时较多，但是没有超时。

代码

class Solution {
    public int[] xorQueries(int[] arr, int[][] queries) {
        int n = queries.length;
        int[] res = new int[n];
        for (int i = 0; i < n; i++) {
            for (int j = queries[i][0]; j <= queries[i][1]; j++) {
                res[i] ^= arr[j];
            }
        }
        return res;
    }
}

执行结果

思路 2 （前缀和）

思路

这种思路是先创建一个数组，第一个元素值为0，第二个元素值为 arr[0]，第三个元素为 arr[0] ^ arr[1]，第四个元素为 arr[0] ^ arr[1] ^ arr[2] ……随后只需简单的查询就可以得到答案，显著减小了时间复杂度。

假设这个新创建的数值为 arr2。以示例1 的前两次计算为例。假设输出的数组为 res。

arr2[0] = 0
arr2[1] = arr[0]
arr2[2] = arr[0] ^ arr[1]
arr2[3] = arr[0] ^ arr[1] ^ arr[2]
……

res[0] = arr[0] ^ arr[1] = arr2[2] ^ 0 = arr2[2] ^ arr2[0]

res[1] = arr[1] ^ arr[2] = (arr[0]) ^ (arr[0] ^ arr[1] ^ arr[2]) = arr2[1] ^ arr2[3]

可得通项公式

res[i] = arr2[Li] ^ arr2[Ri + 1]

代码

class Solution {
    public int[] xorQueries(int[] arr, int[][] queries) {
        int n = queries.length;
        int m = arr.length;
        int[] arr2 = new int[m + 1];
        int[] res = new int[n];
        for (int i = 1; i <= m; i++) {
            arr2[i] = arr2[i - 1] ^ arr[i - 1];
        }
        for (int i = 0; i < n; i++) {
            res[i] = arr2[queries[i][0]] ^arr2[queries[i][1] + 1];
        }
        return res;
    }
}

执行结果

写在最后

本系列博客仅记录转码过程中刷题时的解法。若有错误，欢迎交流与批评指正。官方解法请见力扣（LeetCode）官网。