[프로그래머스 Lv2] 시소 짝꿍 (딕셔너리)

[프로그래머스 Lv2] 시소 짝꿍

문제 설명

어느 공원 놀이터에는 시소가 하나 설치되어 있습니다. 이 시소는 중심으로부터 2(m), 3(m), 4(m) 거리의 지점에 좌석이 하나씩 있습니다.

이 시소를 두 명이 마주 보고 탄다고 할 때, 시소가 평형인 상태에서 각각에 의해 시소에 걸리는 토크의 크기가 서로 상쇄되어 완전한 균형을 이룰 수 있다면 그 두 사람을 시소 짝꿍이라고 합니다. 즉, 탑승한 사람의 무게와 시소 축과 좌석 간의 거리의 곱이 양쪽 다 같다면 시소 짝꿍이라고 할 수 있습니다.

사람들의 몸무게 목록 weights이 주어질 때, 시소 짝꿍이 몇 쌍 존재하는지 구하여 return 하도록 solution 함수를 완성해주세요.

---

제한 사항

• 2 ≤ weights의 길이 ≤ 100,000
• 100 ≤ weights[i] ≤ 1,000
  • 몸무게 단위는 N(뉴턴)으로 주어집니다.
  • 몸무게는 모두 정수입니다.

---

입출력 예

weights result

[100,180,360,100,270]

4

---

입출력 예 설명

{100, 100} 은 서로 같은 거리에 마주보고 앉으면 균형을 이룹니다.

{180, 360} 은 각각 4(m), 2(m) 거리에 마주보고 앉으면 균형을 이룹니다.

{180, 270} 은 각각 3(m), 2(m) 거리에 마주보고 앉으면 균형을 이룹니다.

{270, 360} 은 각각 4(m), 3(m) 거리에 마주보고 앉으면 균형을 이룹니다.

 

---

 

[내 풀이]

※ 소요시간 : 15 분

※ 풀이 전략

- 결국 짝은 1:1, 1:2, 2:3, 3:4 인 관계를 구하는 것이다.

# 15분
from collections import Counter
def solution(weights):
    answer = 0
    
    # 1:1, 1:2, 2:3, 3:4  관계인 개수 구하기

    # 딕셔너리를 이용한 풀이
    w_dict = Counter(weights) # N = 10만 
    for k, v in w_dict.items(): # N = 40만 
        if v >= 2: 
            answer += v*(v-1)//2
        for c in [2, 3/2, 4/3]:
            if k*c in w_dict: 
                answer += v * w_dict[k*c] 
    return answer

 

※ 중요 포인트

- 딕셔너리를 이용하면, 찾는 데 드는 시간이 1이므로 시간제한 걱정 없이 빠르게 구할 수 있다

---

[다른 사람의 풀이]

※ 풀이 전략

- 다른 분들의 풀이도 나와 거의 같았다.