有一個正整數數組 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]
解題思路
維護一個前綴數組sum,sum[i]代表原數組中arr[0] xor arr[1]…arr[i]的結果
因此如果要查詢某個區間[i…j]的異或的結果,那么只需要計算sum[i-1]^sum[j]
因為
- sum[j]=arr[0] xor arr[1]…xor arr[i-1]…xor arr[j]
- sum[i-1]=arr[0] xor arr[1]…xor arr[i-1]
- sum[i-1] xor arr[i]…xor arr[j]=sum[j]
所以可得 arr[i]…xor arr[j]=sum[i-1]^sum[j]
代碼
func xorQueries(arr []int, queries [][]int) []int {n,m := len(arr),len(queries)sum := make([]int, n)res:=make([]int,m)sum[0]=arr[0]for i := 1; i <n ; i++ {sum[i]=arr[i]^sum[i-1]}for i, query := range queries {if query[0]==0{res[i]=sum[query[1]]}else {res[i]=sum[query[0]-1]^sum[query[1]]}}return res
}