squareOrSquareRoot.js (8kyu 37)

in about Dev on Algorithm

Codewars 알고리즘 풀이


Problem

  • If the number has an integer square root, take this, otherwise square the number.
    • 숫자가 정수 제곱근을 가지면 제곱근을 반환하고, 그렇지 않으면 제곱한다.


Tips

제곱근 (Square Root)

어떤 수 a를 두 번 곱하여 x가 되었을 때, a를 x에 대하여 이르는 말.


Solution 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를 추가하고, 새로운 길이를 반환한다.


Solution 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() 메소드

가장 가까운 정수로 반내림하고, 결과를 반환한다.


Solution 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에 대해, 호출한 함수의 결과를 모아 새 배열로 반환한다.