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

Question

[Дмитрий Лукьянчук]

Есть такая вот задачка. При помощи функции создавать двумерный массив n*n. Сама же функция принимает параметр n
В итоге должна получится матрица такого типа
(не четное количество строк и столбцов)
[2,1,1,1,2],
[0,2,1,2,0],
[0,0,2,0,0],
[0,2,1,2,0],
[2,1,1,1,2],

(четное количество)
[2,1,1,2],
[0,2,2,0],
[0,2,2,0],
[2,1,1,2],

Но как не крути у меня получается следующее

К примеру в матрице на 5 столбцов в индексах 1 и 3, между двойками, стоит ноль. А должна быть единица

function arrCreate(size) {
    let arr = [];

    for (let i = 0; i < size; i++) {
        arr[i] = []
        let count = 0;

        for (let k = 0; k < size; k++) {
            if (i === k || i === size - k - 1) {
                arr[i][k] = 2
                count++;
            } else {
                if (count == 1) {
                    arr[i][k] = 1
                } else {
                    arr[i][k] = 0
                    count--
                }
            }
        }
    }
    return arr
}


console.log(arrCreate(5))

Кодпен почему-то не прикрепляется, сори(

Answer 1

[0xD34F]

const createArr = length =>
  Array.from({ length }, (_, i) =>
    Array.from({ length }, (_, j) =>
      (i === j || i === length - j - 1) +
      ((i <= j && i <= length - j - 1) || (i >= j && i >= length - j - 1))
    )
  );

Comments

[historydev]

Выглядит страшно)

[Дмитрий Лукьянчук]

Прошу прощения за тупость))

Но я не могу понять, куда именно этот кусок кода идет)
Я только учу JS и с подобными моментами бывает туговато. Могли бы вы немного детальнее разобрать что и куда идет?

[Дмитрий Лукьянчук]

К тому же, это должно быть внутри функции, как правильно это сделать?

[0xD34F]

Дмитрий Лукьянчук, это и есть функция.

Первое слагаемое ((i === j || i === length - j - 1)) - это условие для получения 2. Это вы и сами должны были увидеть - оно у вас уже есть (имена переменных только отличаются).

Второе слагаемое - условие для 1 и 2. Чтобы было только 1, надо убрать равенства.

То есть, как вам исправить свой код:

1. Вырезаете переменную count, она не нужна, это мусор.
2. Второй if - копируете в него второе слагаемое из моего кода, заменив имена переменных на свои и >=, <= на >, < соответственно.

[Дмитрий Лукьянчук]

0xD34F, Спасибо большое. Все работает)