1310.子数组异或查询Java

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/

思路

方法一：最寻常的做法，一步一步按照queries数组进行遍历，但是这样时间复杂度会很高。

方法二：为了减少时间复杂度，提前对arr数组进行一个预先的异或结果数组的计算，这样在遍历queries数组的时候不用再一遍遍遍历arr数组，而是直接在预先的异或数组中取结果就可以了，这样就把O(n)的查询时间降低为O(1)。

算法分析

1. 时间复杂度O(m*n），其中m为queries的长度，n为arr的长度；空间复杂度为O(1)
2. 时间复杂度为O(n+m)，空间复杂度为O(n)

求解函数

//方法一：普通遍历
public int[] xorQueries(int[] arr, int[][] queries) {
        int[] rst = new int[queries.length];
        for (int i = 0; i < queries.length; ++i) {
            int sum = 0;
            for (int j = queries[i][0]; j <= queries[i][1]; ++j) {
                sum ^= arr[j];
            }
            rst[i] = sum;
        }
        return rst;
    }

//方法二：提前计算前缀异或数组 
public int[] xorQueries(int[] arr, int[][] queries) {
        int[] rst = new int[queries.length];
        int[] xorSum = new int[arr.length + 1]; // 前缀异或和数组
        for (int i = 0; i < arr.length; ++i) {
            xorSum[i + 1] = xorSum[i] ^ arr[i];
        }
        for (int j =0; j < queries.length; ++j) {
            rst[j] = xorSum[queries[j][0]] ^ xorSum[queries[j][1] + 1];
        }
        return rst;
    }

主函数调用

class Solution1310 {
    public static void main(String[] args) {
        int[] arr = new int[]{1, 3, 4, 8};
        int[][] queries = new int[][]{{0, 1}, {1, 2}, {0, 3}, {3, 3}};
        Solution1310 sol = new Solution1310();
        int[] res = sol.xorQueries(arr, queries);
        for (int i = 0; i < res.length; ++i) {
            System.out.print(res[i] + ", ");
        }
    }

    public int[] xorQueries(int[] arr, int[][] queries){...}
}

输出结果