코테 배열 비교하기 로직 리팩토링

2024-03-21

코딩테스트
리팩토링

문제설명 📌

이 문제에서 두 정수 배열의 대소관계를 다음과 같이 정의합니다.
두 배열의 길이가 다르다면, 배열의 길이가 긴 쪽이 더 큽니다.
배열의 길이가 같다면 각 배열에 있는 모든 원소의 합을 비교하여 다르다면 더 큰 쪽이 크고, > 같다면 같습니다.
두 정수 배열 arr1과 arr2가 주어질 때, 위에서 정의한 배열의 대소관계에 대하여 arr2가 > 크다면 -1, arr1이 크다면 1, 두 배열이 같다면 0을 return 하는 solution 함수를 작성해 주세요.


입출력 예 1

예제 1번에서는 arr1의 길이는 2이고 arr2의 길이는 3으로 arr2의 길이가 더 깁니다. 따라서 arr2가 arr1보다 크므로 -1을 return 합니다.

입출력 예 2

예제 2번에서는 arr1의 길이과 arr2의 길이가 4로 같습니다. arr1의 모든 원소의 합은 100 + 17 + 84 + 1 = 202이고 arr2의 모든 원소의 합은 55 + 12 + 65 + 36 = 168으로 arr1의 모든 원소의 합이 더 큽니다. 따라서 arr1이 arr2보다 크므로 1을 return 합니다.

입출력 예 3

예제 3번에서는 arr1의 길이와 arr2의 길이가 5로 같고 각 배열의 모든 원소의 합 또한 15로 같습니다. 따라서 arr1과 arr2가 같으므로 0을 return 합니다.


코딩 풀이

내가 풀이한 코드
function solution(arr1, arr2) {
  let arr1SumResult = 0; //arr1 총합
  let arr2SumResult = 0; //arr2 총합
 
  for (const arr1Sum of arr1) {
    arr1SumResult += arr1Sum;
  }
  for (const arr2Sum of arr2) {
    arr2SumResult += arr2Sum;
  }
 
  // 배열의 길이가 같다면
  if (arr1.length === arr2.length) {
    //배열에 있는 모든 원소의 합을 비교
    if (arr1SumResult > arr2SumResult) {
      return 1;
    } else if (arr1SumResult < arr2SumResult) {
      return -1;
    } else {
      return 0;
    }
  } else if (arr1.length > arr2.length) {
    return 1;
  } else {
    return -1;
  }
  return 0;
}

배열을 비교하는 코테 문제를 풀었는데 좀 복잡시러워 보인다.
더 가독성 높게 리팩토링 해보려한다.

리팩토링 코드
function solution(arr1, arr2) {
  const arr1Sum = arr1.reduce((acc, cur) => acc + cur, 0); // arr1의 총합
  const arr2Sum = arr2.reduce((acc, cur) => acc + cur, 0); // arr2의 총합
 
  if (arr1.length === arr2.length) {
    // 두 배열의 길이가 같을 때
    if (arr1Sum > arr2Sum) {
      return 1;
    } else if (arr1Sum < arr2Sum) {
      return -1;
    } else {
      return 0; // 두 배열의 합이 같으면 0을 반환
    }
  } else {
    // 두 배열의 길이가 다를 때
    return arr1.length > arr2.length ? 1 : -1;
  }
}

풀이 ✏️

1. for of 반복문 대신 reduce

reduce() 메서드는 배열의 각 요소에 대해 주어진 콜백 함수를 실행하고 하나의 결과 값을 반환하는 배열 메서드이다. reduce는 누적된 결과를 반환하며, 각 요소를 순회하면서 현재까지의 누적된 값과 현재 요소를 이용하여 계산을 수행한다.

reduce
array.reduce(
  callback(accumulator, currentValue, currentIndex, array),
  initialValue
);

© 2023 chosule blog by Next.js