JSFiddle - React, Tailwind, and code Playground

by Ansmtz

JavaScript

const solve = (k) => {
  // k = 4
  let res = [];
  // пока i не равно 4
  for(let i = 1; i <= k; i++) {
    // temp = res[1]
    let temp = res[res.length - 1];
    // если массив пуст или длина равна 2
    if(!temp || temp.length === Math.sqrt(k)) {
      console.log(i)
      // res = [[1]]
      res.push([i]);
    }
    else {
      temp.push(i);
      // [2]
    }
  }
  return res;
};

console.log(solve(4));