[programmers] 시소 짝꿍 - Java

- 시소 짝꿍 문제는 두 사람이 시소에서 완전한 균형을 이루는 쌍의 수를 찾는 문제이다. - 시소는 2m, 3m, 4m 거리에 좌석이 있으며, 무게와 거리의 곱이 양쪽이 같아야 균형을 이룬다. - HashMap을 활용하여 각 무게의 발생 횟수를 저장하고, 같은 무게와 다른 무게의 쌍을 각각 계산한다. - 가능한 거리 비율 (1:1, 2:3, 3:4, 2:4)을 고려하여 균형을 이루는 모든 쌍의 수를 구한다.

DriedPollack

Dec 23, 2024

Contents

시소 짝꿍 solution.java 핵심 키워드 결론!

시소 짝꿍

문제 설명

어느 공원 놀이터에는 시소가 하나 설치되어 있습니다. 이 시소는 중심으로부터 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) 거리에 마주보고 앉으면 균형을 이룹니다.

solution.java


import java.util.*;

class Solution {
    public long solution(int[] weights) {
        // HashMap을 사용하여 각 가중치의 발생 횟수를 계산
        Map<Integer, Long> weightCount = new HashMap<>();
        for (int weight : weights) {
            weightCount.put(weight, weightCount.getOrDefault(weight, 0L) + 1);
        }

        long totalPairs = 0;

        // 가능한 거리 비율
        int[][] ratios = {{1, 1}, {2, 3}, {3, 4}, {2, 4}};

        for (Map.Entry<Integer, Long> entry : weightCount.entrySet()) {
            int weight = entry.getKey();
            long count = entry.getValue();

            // 같은 무게를 가진 pair 수 세기
            if (count > 1) {
                totalPairs += (count * (count - 1)) / 2;
            }

            // 서로 다른 가중치를 가진 pair 수 세기
            for (int[] ratio : ratios) {
                int num = ratio[0];
                int denom = ratio[1];
                if (weight * denom % num == 0) {
                    int pairedWeight = weight * denom / num;
                    if (pairedWeight > weight && weightCount.containsKey(pairedWeight)) {
                        totalPairs += count * weightCount.get(pairedWeight);
                    }
                }
            }
        }

        return totalPairs;
    }
}