1442. 形成两个异或相等数组的三元组数目(动态规划)

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

给你一个整数数组 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

标签：输出,arr,示例,三元组,异或,let,1442,输入
来源： https://blog.51cto.com/u_15201483/2794277