시소 짝꿍 다시정리

magicdrill·2025년 6월 25일
0

시소 짝꿍

import java.util.*;

class Solution {
    public long solution(int[] weights) {
        long answer = 0;
        Map<Double, Integer> map = new HashMap<>();

        Arrays.sort(weights);
        
        for (int w : weights) {
            double[] ratios = {
                w * 1.0,
                w * 2.0 / 3,
                w * 2.0 / 4,
                w * 3.0 / 4
            };

            for (double r : ratios) {
                if (map.containsKey(r)) {
                    answer += map.get(r);
                }
            }

            map.put((w * 1.0), map.getOrDefault((w * 1.0), 0) + 1);
        }

        return answer;
    }
}

0개의 댓글