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

문제 설명

어느 공원 놀이터에는 시소가 하나 설치되어 있습니다. 이 시소는 중심으로부터 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

풀이

처음에는 공배수를 구해서 풀어야 하나 고민했는데 문제에 2,3,4 지점에 좌석이 하나씩 있다는 문구를 다시 확인하니 간단한 문제라는 것을 깨달았습니다. 두 수를 2,3,4 와 곱하고 값이 같은지 확인하면 되는 문제입니다.

#include <string>
#include <vector>
#include <algorithm>
using namespace std;

bool seesaw(const int& a, const int& b){
    if(a == b){
        return true;
    }
    
    if( a * 4 == b * 3){
        return true;
    }
    if( a * 4 == b * 2){
        return true;
    }
    if( a * 3 == b * 2){
        return true;
    }
    return false;
}

long long solution(vector<int> weights) {
    long long answer = 0;
    sort(weights.begin(), weights.end());
    for(int i = 0; i < weights.size() - 1; i++){
        for(int j = i+1; j < weights.size(); j++){
            if(seesaw(weights[i], weights[j])){
                answer += 1;
            }
        }
    }
    
    return answer;
}