[프로그래머스 코딩테스트 with JAVA] 시소 짝꿍

시소 짝꿍

Lv. 2 | 링크 https://school.programmers.co.kr/learn/courses/30/lessons/152996

어느 공원 놀이터에는 시소가 하나 설치되어 있습니다. 이 시소는 중심으로부터 2(m), 3(m), 4(m) 거리의 지점에 좌석이 하나씩 있습니다.
이 시소를 두 명이 마주 보고 탄다고 할 때, 시소가 평형인 상태에서 각각에 의해 시소에 걸리는 토크의 크기가 서로 상쇄되어 완전한 균형을 이룰 수 있다면 그 두 사람을 시소 짝꿍이라고 합니다. 즉, 탑승한 사람의 무게와 시소 축과 좌석 간의 거리의 곱이 양쪽 다 같다면 시소 짝꿍이라고 할 수 있습니다.
사람들의 몸무게 목록 weights이 주어질 때, 시소 짝꿍이 몇 쌍 존재하는지 구하여 return 하도록 solution 함수를 완성해 주세요.

 

제한 사항

• 2 ≤ weights의 길이 ≤ 100,000
• 100 ≤ weights[i] ≤ 1,000
  • 몸무게 단위는 N(뉴턴)으로 주어집니다.
  • 몸무게는 모두 정수입니다.

입출력 예

입력 예시	출력 예시
[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) 거리에 마주보고 앉으면 균형을 이룹니다.

소스코드

import java.util.*;

class Solution {
    public long solution(int[] weights) {
        long answer = 0;

        Arrays.sort(weights);
        
        for(int i = 0; i < weights.length - 1; i++){
            for(int j = i+1; j < weights.length; j++){
                if((weights[i] * 2) < weights[j]){
                    break;
                }else if(weights[i] == weights[j]){
                    answer++;
                }else if(weights[i] * 2 == weights[j]){
                    answer++;
                }else if(weights[i] * 3 == weights[j] * 2){
                    answer++;
                }else if(weights[i] * 4 == weights[j] * 3){
                    answer++;
                }
            }
        }
        
        return answer;
    }
}

이 중 for문으로 전체를 반복하게 되면 시간제한을 넘어가게 되기 때문에 중간에 break 장치를 걸어두어야 한다. 나는 오름차순 정렬을 먼저 수행한 후 균형을 이룰 수 없는 2 배 이상의 숫자가 되면 내부 for문을 중지시키도록 하였더니 통과가 되었다. 다른 사람들의 풀이를 보니 사람마다 매우 다르게 작성했던데 더 시간을 써서 작성하면 더 좋은 코드가 나오지 않을까 싶다.

 

GitHub Link

GitHub - Leeyeonjae/coding-test: 코딩 테스트 연습