0

我试图让这个堆算法返回一个排列数组,而不是如下打印。我知道这可以通过在函数外部声明一个数组并推送到它来完成,但是我想避免这种方法。如何在不使用外部数组的情况下使其返回排列数组?

function heaps(arr, n) {
  if (n === undefined) n = arr.length;
  if (n <= 1) console.log(arr);
  else {
    for (let i = 0; i <= n - 1; i++)
     {
      heaps(arr, n-1);
      if 
        (n % 2 === 0) [arr[n-1], arr[i]] = [arr[i], arr[n-1]];
      else             
        [arr[n-1], arr[0]] = [arr[0], arr[n-1]];
    }
  }
}
4

1 回答 1

2

只需将其return作为数组的结果,并从循环中的递归调用中收集返回值:

function heaps(arr, n = arr.length) {
  if (n <= 1) return [arr.slice()];
  let result = [];
  for (let i = 0; i <= n - 1; i++) {
    result.push(...heaps(arr, n-1));
    if (n % 2 === 0)
      [arr[n-1], arr[i]] = [arr[i], arr[n-1]];
    else             
      [arr[n-1], arr[0]] = [arr[0], arr[n-1]];
  }
  return result;
}

或者,制作一个确实推送到外部数组的内部辅助函数

function heaps(arr) {
  let result = [];
  function helper(n) {
    if (n <= 1)
      result.push(arr.slice());
    else
      for (let i = 0; i <= n - 1; i++) {
        heaps(n-1);
        if (n % 2 === 0)
          [arr[n-1], arr[i]] = [arr[i], arr[n-1]];
        else             
          [arr[n-1], arr[0]] = [arr[0], arr[n-1]];
      }
  }
  helper(arr.length);
  return result;
}

如果您坚持不使用“外部”数组,请使result数组和辅助函数的输入arr显式参数。

于 2018-10-08T20:27:06.710 回答