26-jung0115

[jung0115]

🔗 문제 링크

수학 | 프로그래머스 - 시소 짝꿍(Lv.2)

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

✔️ 소요된 시간

30분

✨ 수도 코드

우선, 시소에 두 사람이 앉을 때 거리 비율은 1:1, 2:3, 1:2(= 2:4), 3:4 중 하나입니다

그래서 처음에는 사람들을 몸무게 오름차순으로 정렬하고,
적은 몸무게인 사람부터 기준으로 잡아서 뒷순서의 사람과 위 비율 중 하나로 앉아서 균형을 맞출 수 있는지 완전 탐색으로 구현했습니다

하지만 같은 무게인 사람도 중복해서 계산해 일부 케이스에서 시간초과가 발생했습니다.
아래는 시간 초과가 발생한 코드의 구현부입니다!

❌ 시간 초과 코드

for(i: Int in 0..people.size - 1) {
  for(j: Int in i+1..people.size - 1) {
    if(people[i] == people[j]) {
      answer++
      continue
    }
    var checked = false
    for(m1: Int in 2..3) {
      for(m2: Int in m1+1..4) {
        if(people[i] * m2 == people[j] * m1) {
          answer++
          checked = true
          break
        }
      }
          
      if(checked) break
    }
  }
}

중복되는 연산을 줄이기 위해 몸무게�마다 해당하는 사람 수를 세어주고, 가능한 쌍을 연산해줬습니다

만약 같은 몸무게인 사람이 2명 이상일 경우,
그 사람들끼리는 모두 1:1 거리로 앉을 수 있기 때문에 쌍을 조합할 수 있는 수인n * (n-1) / 2을 누적해줬습니다

나머지 2:3, 1:2, 3:4 비율의 거리로 앉는 경우는 같은 몸무게끼리와 달리 중복되는 사람이 없으므로
현재 기준이 되는 몸무게의 사람 수와 해당 비율로 앉을 수 있는 몸무게의 사람 수를 곱해주면 쌍의 수를 구할 수 있습니다.

각 비율마다 if문으로 체크해주고, 해당할 경우 경우의 수를 누적해줬습니다

비율 체크를 한 방법을 2:3의 비율을 기준으로 설명하면,
현재 탐색 중인 무게가 왼쪽, 2에 해당한다고 생각합니다!

1. 모든 몸무게가 정수이기 때문에 2로 나누어 떨어져야 비율이 성립하므로 나누어 떨어지는지를 먼저 계산합니다
2. 그리고 기준 몸무게에 3을 곱하고 2를 나눈 값이 존재하는지를 찾습니다 (containsKey 이용)

이렇게 찾은 몸무게를 위에서 설명한 방식으로 계산해주면 최종적으로 시소 짝꿍의 쌍 수를 찾을 수 있습니다!

✅ 최종 코드

class Solution {
  fun solution(weights: IntArray): Long {
    var answer: Long = 0
    val weightCount = mutableMapOf<Int, Long>()

    for(weight in weights) {
      weightCount[weight] = weightCount.getOrDefault(weight, 0L) + 1
    }
    
    for((weight, count) in weightCount) {
      // 같은 무게인 사람이 2명 이상 -> 1:1로 앉을 수 있음
      if(count > 1) {
        answer += count * (count - 1) / 2
      }
      
      // 2:3 비율
      if (weight % 2 == 0 && weightCount.containsKey(weight * 3 / 2)) {
        answer += count * weightCount[weight * 3 / 2]!!
      }
      
      // 2:4 = 1:2 비율
      if (weightCount.containsKey(weight * 2)) {
        answer += count * weightCount[weight * 2]!!
      }

      // 3:4 비율
      if (weight % 3 == 0 && weightCount.containsKey(weight * 4 / 3)) {
        answer += count * weightCount[weight * 4 / 3]!!
      }
    }
    
    return answer
  }
}

📚 새롭게 알게된 내용