문제

  • 숫자가 정수 제곱근을 가지면 제곱근을 반환하고, 그렇지 않으면 제곱한다.

Tips

  • 제곱근 (square root): 어떤 수 a를 두 번 곱하여 x가 되었을 때, a를 x에 대하여 이르는 말.

풀이 01

function squareOrSquareRoot(arr) {
  let result = [];

  for (let i = 0; i < arr.length; i++) {
    if (Math.sqrt(arr[i]) % 1 === 0) {
      result.push(Math.sqrt(arr[i]));
    } else {
      result.push(arr[i] * arr[i]);
    }
  }
  return result;
}

squareOrSquareRoot([1, 2, 3, 4]);        // [1, 4, 9, 2]
squareOrSquareRoot([16, 20, 100, 121]);  // [4, 400, 10, 11]
  • Math.sqrt(): 숫자의 제곱근을 반환한다.
  • push(): 배열의 끝에 새 element를 추가하고, 새로운 길이를 반환한다.

풀이 02

function squareOrSquareRoot(arr) {
  let result = [];

  for (let i = 0; i < arr.length; i++) {
    if (Math.sqrt(arr[i]) === Math.floor(Math.sqrt(arr[i]))) {
      result.push(Math.sqrt(arr[i]));
    } else {
      result.push(arr[i] * arr[i]);
    }
  }
  return result;
}

squareOrSquareRoot([1, 2, 3, 4]);        // [1, 4, 9, 2]
squareOrSquareRoot([16, 20, 100, 121]);  // [4, 400, 10, 11]
  • Math.floor(): 가장 가까운 정수로 반내림하고, 결과를 반환한다.

풀이 03

function squareOrSquareRoot(arr) {
  return arr.map(i => Math.sqrt(i) % 1 === 0 ? Math.sqrt(i) : i * i);
}

squareOrSquareRoot([1, 2, 3, 4]);        // [1, 4, 9, 2]
squareOrSquareRoot([16, 20, 100, 121]);  // [4, 400, 10, 11]
  • map(): 배열 내 모든 element에 대해, 호출한 함수의 결과를 모아 새 배열로 반환한다.