leetcode 1310. 子数组异或查询(C++、python)

☆樱花仙子☆ 提交于 2020-01-14 18:10:27

有一个正整数数组 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]

 

提示:

  • 1 <= arr.length <= 3 * 10^4
  • 1 <= arr[i] <= 10^9
  • 1 <= queries.length <= 3 * 10^4
  • queries[i].length == 2
  • 0 <= queries[i][0] <= queries[i][1] < arr.length

C++

class Solution {
public:
    vector<int> xorQueries(vector<int>& arr, vector<vector<int>>& queries) 
    {
        vector<int> res;
        int n=arr.size();
        vector<int> tmp(n);
        tmp[0]=arr[0];
        for(int i=1;i<n;i++)
        {
            tmp[i]=tmp[i-1]^arr[i];
        }
        for(auto it:queries)
        {
            int left=it[0];
            int right=it[1];
            if(0==left)
            {
                res.push_back(tmp[right]);
            }
            else
            {
                res.push_back(tmp[right]^tmp[left-1]);
            }
        }
        return res;
    }
};

python

class Solution:
    def xorQueries(self, arr: List[int], queries: List[List[int]]) -> List[int]:
        res=[]
        tmp=[]
        n=len(arr)
        tmp.append(arr[0])
        for i in range(1,n):
            tmp.append(tmp[i-1]^arr[i])
        for it in queries:
            left=it[0]
            right=it[1]
            if 0==left:
                res.append(tmp[right])
            else:
                res.append(tmp[right]^tmp[left-1])
        return res
        

 

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!