码迷,mamicode.com
首页 > 编程语言 > 详细

1442. 形成两个异或相等数组的三元组数目

时间:2021-05-24 14:31:19      阅读:0      评论:0      收藏:0      [点我收藏+]

标签:bsp   三元组   function   rip   -o   col   @param   UNC   ==   

给你一个整数数组 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) 的数目。
解题思路:  动态规划
    1、我们观察到所求的数组区间是[i,j-1][j,k],满足a==b,根据异或的特性 a^b=0,可以将问题转化为求合并后区间[i,k] arr[i]^...^arr[k] = 0
    2、arr[i]^...^arr[k] = 0 可以转化为 arr[0]^...^a[i-1] ^ arr[0]^...^arr[i-1]^arr[i]^...^arr[k] = 0 
        即 arr[0]^...^a[i-1] ^arr[0]^...^a[k] =0
    3、假设dp[i] 为 arr[0]^...^arr[i], 则有 dp[i-1]^dp[k] = 0
    4、得到[i,k]区间后,再求j
/**
 * @param {number[]} arr
 * @return {number}
 */
var countTriplets = function(arr) {
    let n = arr.length;
    let dp = new Array(n).fill(0);
    let res = [];
    let count=0;
    for(let i = 0;i<n;i++){
        dp[i] = (i==0?0:dp[i-1]) ^ arr[i];
        if(dp[i] == 0){
            res.push({‘i‘:0,‘k‘:i});
        }
    }
    for(let i=0;i<n;i++){
        for(let j=i+1;j<n;j++){
            if((dp[i]^dp[j]) === 0){
                res.push({‘i‘:i+1,‘k‘:j});
            }
        }
    }
    while(res.length>0){
        let cur = res.pop();
        for(let j=cur.i+1;j<=cur.k;j++){
            let left = (cur.i === j-1)?(dp[cur.i]^dp[cur.i-1]):(dp[cur.i-1]^dp[j-1]);
            let right = (cur.k===j)?(dp[cur.k]^dp[cur.k-1]):(dp[cur.k]^dp[j-1]);
            if( left === right){
                count++;
            }
        }
    }
    return count;
};
let arr = [2,3,1,6,7]
console.log(arr, countTriplets(arr))
arr = [1,1,1,1,1]
console.log(arr, countTriplets(arr))
arr =[2,3]
console.log(arr, countTriplets(arr))
arr = [1,3,5,7,9]
console.log(arr, countTriplets(arr))
arr = [7,11,12,9,5,2,7,17,22]
console.log(arr, countTriplets(arr))

示例 1:

输入:arr = [2,3,1,6,7]
输出:4
解释:满足题意的三元组分别是 (0,1,2), (0,2,2), (2,3,4) 以及 (2,4,4)
示例 2:
输入:arr = [1,1,1,1,1]
输出:10
示例 3:
输入:arr = [2,3]
输出:0
示例 4:
输入:arr = [1,3,5,7,9]
输出:3
示例 5:
输入:arr = [7,11,12,9,5,2,7,17,22]
输出:8
 
提示:
1 <= arr.length <= 300
1 <= arr[i] <= 10^8

1442. 形成两个异或相等数组的三元组数目

标签:bsp   三元组   function   rip   -o   col   @param   UNC   ==   

原文地址:https://www.cnblogs.com/yanjianjiang/p/14780140.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!