[프로그래머스] 시소 짝꿍(JavaScript)

문제 설명[링크]

어느 공원 놀이터에는 시소가 하나 설치되어 있습니다. 이 시소는 중심으로부터 2(m), 3(m), 4(m) 거리의 지점에 좌석이 하나씩 있습니다.
이 시소를 두 명이 마주 보고 탄다고 할 때, 시소가 평형인 상태에서 각각에 의해 시소에 걸리는 토크의 크기가 서로 상쇄되어 완전한 균형을 이룰 수 있다면 그 두 사람을 시소 짝꿍이라고 합니다. 즉, 탑승한 사람의 무게와 시소 축과 좌석 간의 거리의 곱이 양쪽 다 같다면 시소 짝꿍이라고 할 수 있습니다.
사람들의 몸무게 목록 weights이 주어질 때, 시소 짝꿍이 몇 쌍 존재하는지 구하여 return 하도록 solution 함수를 완성해주세요.

풀이

function solution(weights) {
  let result = 0;
  const weightCountMemo = new Array(1001).fill(0);
  const forceMemo = new Array(4001).fill(0);

  for (const weight of weights) {
    weightCountMemo[weight] += 1;
  }

  const calculator = (number) => ((number - 1) * number) / 2;

  for (let weight = 100; weight <= 1000; weight += 1) {
    const weightCount = weightCountMemo[weight];

    if (weightCount) {
      forceMemo[weight * 2] += weightCount;
      forceMemo[weight * 3] += weightCount;
      forceMemo[weight * 4] += weightCount;

      result -= 2 * calculator(weightCount);
    }
  }

  result += forceMemo.filter((el) => el).reduce((acc, cur) => acc + calculator(cur), 0);

  return result;
}

결과