[프로그래머스/JavaScript] Lv.2 시소 짝꿍

https://school.programmers.co.kr/learn/courses/30/lessons/152996

프로그래머스

 

 

---

1. 문제 설명

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

 

제한사항

 

---

2. 입출력

입출력

 

 

---

3. 문제 풀이

* 가능한 비율 CASE
- a = b * 1
- a * 2 = b * 3
- a * 2 = b * 4
- a * 3 = b * 4

case 배열 선언 → 큰 값부터 모든 case에 대해 짝꿍 가능 여부를 판단 → Map에 weights 값을 카운트하며 저장 → Map에 있는 값과 짝꿍 값을 찾으면, answer증가 

 

function solution(weights) {
  let answer = 0;
  const map = new Map();
  
  // 1) 비율 case 
  const ratio = [1, 3 / 2, 2, 4 / 3];

  // 2) 큰 값부터 시소 짝궁이 가능한 값을 찾기 위해 내림차순 정렬
  weights.sort((a, b) => b - a).forEach(w => {
  
    // 3) 가능한 모든 비율을 적용하여 짝궁 찾고,
    // 짝궁을 찾으면 answer에 추가, 중복되는 짝궁의 개수만큼 추가한다.
    ratio.forEach(r => {if (map.has(w * r)) answer += map.get(w * r)});
    
    // 4) 중복되는 값만큼 추가하기 위해 중복 값 카운트
    map.set(w, (map.get(w) || 0) + 1);
  });

  return answer;
}