[프로그래머스/Level2] 시소 짝꿍 (JAVA)

📖 문제

프로그래머스

 

 

💡  풀이 방식

• Map<무게, 무게 갯수>

1. 무게 배열을 오름차순 정렬한다.

 → 오름차순 정렬했으므로 무게가 같은 경우와 (2/3, 2/4, 3/4)일 때만 고려해 접근

 

2. 기존 key값과 같으면 체크해 정답 + 1

 

 

해당 예제 [100, 180, 360, 100, 270]의 경우

Map에 입력되는 값은 (100, 2), (180, 1), (270,1) (360,1)이다.

 

정답이 업데이트 되는 상황은

1) 두 번째 100일 때 +1

2) 270도일 때, (b) 270 * 2.0 / 3.0 = 180.0이므로 +1

3) 360도 일 때, (c) 360 * 1.0 / 2.0 에서 +1, (d) 360 * 3.0 / 4.0에서 +1된다.

 

 

💥 유의사항

시간 초과 이슈로 인해 for문을 한 번만 사용해야 한다.

→ 이를 위해 사전 정렬해야 한다.

 

 

🔺 코드

import java.util.*;
 
class Solution {
    public long solution(int[] weights) {
        long answer = 0;
        
        Arrays.sort(weights);   // 오름차순 정렬
        
        Map<Double, Integer> map = new HashMap<>(); // 무게 weight, 무게 갯수
        
        for(int w : weights) {
            double a = w * 1.0;             // 무게가 같은 경우
            double b = (w * 2.0) / 3.0;     // 2/3인 경우
            double c = (w * 1.0) / 2.0;     // 2/4인 경우
            double d = (w * 3.0) / 4.0;     // 3/4인 경우
            
            // 기존 key값과 같으면 체크해 정답 + 1            
            if(map.containsKey(a))  answer += map.get(a);
            if(map.containsKey(b))  answer += map.get(b);
            if(map.containsKey(c))  answer += map.get(c);
            if(map.containsKey(d))  answer += map.get(d);
            
            map.put((w * 1.0), map.getOrDefault((w * 1.0), 0) + 1);
        }
        
        return answer;
    }
}

 

 

 

 

➕ 다른 풀이 방식

이분 탐색 활용

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

 

---

💦 어려웠던 점

Map을 사용할 생각을 하지 못 했다.

자꾸 시간초과가 떠서 어떻게 해야할지 몰랐다.

 

 

🧐 새로 알게 된 내용

시간초과 이슈 해결 - for문을 한 번만 돌린다. - 배열을 정렬한다.

 

 

1회독	2회독	3회독	4회독	5회독
V

---

(참고)

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