Как решить задачку, сохранив предыдущий элемент массива в замыкании?

Question

[lil_web]

Я на собеседовании решил задачку:

/**
 * @param {Array<number>} array
 * @return {Array<number>}
 */
const getSumNestElem = (array) => {
  const result = [];

  array.forEach((item, i) => {
    const newValue = (array[i - 1] || 0) + (array[i + 1] || 0);

    result.push(newValue);
  });

  return result;
};

const array = [0, 1, 2, 3, 4];
const result = getSumNestElem(array);
console.log(result); // [1, 2, 4, 6, 3]

Затем интервьюер усложнил задание: попросил отформатировать текущий массив:

/**
 * @param {Array<number>} array
 * @return {void}
 */
const formatSumNestElem = (array) => {
  // make each item is equal to: array[i - 1] + array[i + 1]
};

const array = [0, 1, 2, 3, 4];
formatSumNestElem(array);
console.log(array); // [1, 2, 4, 6, 3]

Я безуспешно поигрался с замыканием:

const formatSumNestElem = (array) => {
  for (let i = 0; i < array.length; i++) {
    const item = array[i];

    (() => {
      const newValue = (array[i - 1] || 0) + (array[i + 1] || 0);

      array[i] = newValue;
    })();
  }
};

Подскажите, пожалуйста, как хорошо решить эту задачку: отформатировать массив заданным способом.

Answer 1

[0xD34F]

Не надо никуда ничего сохранять:

arr.splice(0, arr.length, ...arr.map((n, i, a) => (a[i - 1] ?? 0) + (a[i + 1] ?? 0)));

Comments

[lil_web]

Вау, здорово. Спасибо. Получается, первую задачу можно было решить вот так. А я думал, что с map так нельзя.

return array.map((item, i) => (array[i - 1] || 0) + (array[i + 1] || 0));

Но ведь у вас для более сложной задачи временная сложность квадратичная?

[0xD34F]

lil_web, нет.

[lil_web]

0xD34F, разве Array.prototype.splice не O(n)?

[0xD34F]

lil_web, n. В данном случае. Дальше-то что? Кажется, вы не понимаете, что происходит при вызове splice (нет, ничего объяснять вам не буду - сами разбирайтесь, если надо).

[lil_web]

0xD34F, ок. Но интервьюёр сказал, что можно решить, сохраняя значение в замыкании. Хотелось бы этот вариант рассмотреть. (Я не прошу обязательно вас).

[0xD34F]

lil_web, ну, вот с сохранением предыдущего:

let prev = 0;
for (let i = 0; i < arr.length; i++) {
  const curr = arr[i];
  arr[i] = prev + (arr[i + 1] ?? 0);
  prev = curr;
}