在JavaScript中查找具有相等总和的n个子数组
我们需要编写一个JavaScript函数,该函数将整数数组作为第一个参数,将整数作为第二个参数。
该函数应检查我们是否可以从原始数组创建n个(第二个参数)子数组,以使所有子数组具有相等的总和。
例如-
如果输入是-
const arr = [4, 3, 2, 3, 5, 2, 1]; const num = 4;
输出应该为true,因为子数组为:[5],[1、4],[2、3],[2、3]的总和等于5。
示例
以下是代码-
const arr = [4, 3, 2, 3, 5, 2, 1];
const num = 4;
const canFormSubarray = (arr = [], num) => {
const total = arr.reduce((sum, num) => sum + num, 0);
if (total % num !== 0) {
return false;
}
const target = total / num;
const visited = new Array(arr.length).fill(false);
const canPartition = (start, numberOfSubsets, currentSum) => {
if (numberOfSubsets === 1) {
return true;
}
if (currentSum === target) {
return canPartition(0, numberOfSubsets - 1, 0);
};
for (let i = start; i < arr.length; i++) {
if (!visited[i]) {
visited[i] = true;
if (canPartition(i + 1, numberOfSubsets, currentSum + arr[i])) {
return true;
}
visited[i] = false;
};
};
return false;
};
return canPartition(0, num, 0);
};
console.log(canFormSubarray(arr, num));输出结果以下是控制台输出-
true