leetcode每日一题—1442. 形成两个异或相等数组的三元组数目

最新推荐文章于 2022-05-08 17:56:42 发布

柒年时光

最新推荐文章于 2022-05-08 17:56:42 发布

阅读量143

点赞数

分类专栏： leetcode Python

本文链接：https://blog.csdn.net/jqq125/article/details/116989625

版权

Python 同时被 2 个专栏收录

484 篇文章 1 订阅

订阅专栏

leetcode

461 篇文章 1 订阅

订阅专栏

题目：
给你一个整数数组 arr 。

现需要从数组中取三个下标 i、j 和 k ，其中 (0 <= i < j <= k < arr.length) 。

a 和 b 定义如下：

a = arr[i] ^ arr[i + 1] ^ … ^ arr[j - 1]
b = arr[j] ^ arr[j + 1] ^ … ^ arr[k]
注意：^ 表示按位异或操作。

请返回能够令 a == b 成立的三元组 (i, j , k) 的数目。
在这里插入图片描述

思路：
a=b 等价于 a^b=0 。故本题可转化为：arr[i] ^ arr[i + 1] ^ … ^ arr[j - 1] ^ arr[j] ^ arr[j + 1] ^ … ^ arr[k]=0，仅需确定符合条件的i和k即可,i+1<=j<k。即j 有k-i种取法。

解答：

class Solution:
    def countTriplets(self, arr: List[int]) -> int:
        count=0
        n=len(arr)
        for i in range(n):
            sum=0
            for k in range(i,n):
                sum^=arr[k]
                if sum==0 and k>i:
                    count+=(k-i)   
        return count