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

---

🔗 문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/152996

프로그래머스

---

반응형

728x90

👩‍💻 코드

#include <string>
#include <vector>
#include <unordered_map>
#include <algorithm>

using namespace std;

long long solution(vector<int> weights) {
    unordered_map<float, long long> weight_count;
    long long answer = 0;
    
    sort(weights.begin(), weights.end());
    
    for (long long weight : weights) {
        weight_count[weight]++;
    }
    
    for (long long weight : weights) {
        weight_count[weight]--;
        
        answer += weight_count[weight];
        answer += weight_count[weight * 2];
        answer += weight_count[(float)weight * 3 / 2];
        answer += weight_count[(float)weight * 4 / 3];
    }
    
    return answer;
}

---

📝 풀이

하나의 weight 값을 기준으로 만들 수 있는 균형을 구하여 다른 weight값과 비교하며 답을 찾아낼겁니다.

쌍을 구하는 것은 중복을 허용하지 않기 때문에 weights를 오름차순으로 정렬합니다.

무게와 횟수를 나타내는 unordered_map에 weights의 정보를 저장합니다.

 

현재 차례의 무게는 뺍니다.

균형을 이룰 수 있는 경우는 같은 무게일 때,  1: 2 (2: 4), 2 : 3 , 3 : 4의 경우로 총 4가지 입니다.

4가지 경우를 모두 확인하여 존재한다면 answer을 증가시킵니다.

이 때 주의할 점은 / 연산의 경우 소수점을 발생시킬 수 있으므로 float로 형변환을 해주어야 합니다.

 

모든 경우의 시소 짝꿍의 확인이 끝나면 answer을 return 해줍니다.