일기 대신 코드 슬쩍

[프로그래머스][JAVA] Lv0. 평행 본문

코딩테스트/프로그래머스(JAVA)

[프로그래머스][JAVA] Lv0. 평행

코코자 2024. 3. 14. 22:07

문제 설명

점 네 개의 좌표를 담은 이차원 배열  dots가 다음과 같이 매개변수로 주어집니다.

  • [[x1, y1], [x2, y2], [x3, y3], [x4, y4]]

주어진 네 개의 점을 두 개씩 이었을 때, 두 직선이 평행이 되는 경우가 있으면 1을 없으면 0을 return 하도록 solution 함수를 완성해보세요.


제한사항

  • dots의 길이 = 4
  • dots의 원소는 [x, y] 형태이며 x, y는 정수입니다.
    • 0 ≤ x, y ≤ 100
  • 서로 다른 두개 이상의 점이 겹치는 경우는 없습니다.
  • 두 직선이 겹치는 경우(일치하는 경우)에도 1을 return 해주세요.
  • 임의의 두 점을 이은 직선이 x축 또는 y축과 평행한 경우는 주어지지 않습니다.

입출력 예

dots result

[[1, 4], [9, 2], [3, 8], [11, 6]] 1
[[3, 5], [4, 1], [2, 4], [5, 10]] 0

아이디어

마찬가지로 전에 파이썬으로 푼 적이 있는 문제이다.

그때 코드를 살펴보니, combinations을 이용해서 조합을 다 찾아주었다.

from itertools import combinations
def solution(dots):
    answer = []
    dap = 0
    for i in combinations(dots,2):
            scope = (i[1][1]-i[0][1]) / (i[1][0]-i[0][0]) # 기울기 구하는 공식 y증가량/x증가량 아시죠?
            if scope in answer: # 기울기가 같다면 1 출력
                dap = 1
                break
            else:
                answer.append(scope)
    return dap

하지만 JAVA에는 combination을 해주는 라이브러리가 없다고 한다…

그래서 생각해보니 사실상 4개의 좌표만 존재하고 나올 수 있는 경우의 수는 3가지 뿐이라 그냥 다 코드로 작성하는게 나을 것이라고 생각했다.

아 그리고 기울기가 정수가 아닐 수도 있으므로 double type으로 바꾸어주었다.

class Solution {
    public int solution(int[][] dots) {
        int answer = 0;
        if ((double)(dots[0][1]-dots[1][1]) / (dots[0][0]-dots[1][0]) == (double)((dots[2][1]-dots [3][1]) / (dots[2][0]-dots[3][0]))) answer +=1;
        if ((double)(dots[0][1]-dots[2][1]) / (dots[0][0]-dots[2][0]) == (double)((dots[1][1]-dots [3][1]) / (dots[1][0]-dots[3][0]))) answer +=1;
        if ((double)(dots[0][1]-dots[3][1]) / (dots[0][0]-dots[3][0]) == (double)((dots[2][1]-dots [1][1]) / (dots[2][0]-dots[1][0]))) answer +=1;
        
        return answer;
    }
}

테스트 5,6,11을 통과하지 못했다. 무엇이 원인일까…

아 난 멍청이인가?

나는 바보같이 평행인 경우가 2개거나 3개이면 2랑 3을 return하도록 했는데

문제를 다시 읽어보니 그냥 평행인 경우와 평행이 아닌 경우만 고려해서 1과 0으로 return하는 거였다.

오늘 멘탈이 많이 안 좋아서 문제를 이해조차 못했나보다…


문제 풀이

class Solution {
    public int solution(int[][] dots) {
        
        if ((double)(dots[0][1]-dots[1][1]) / (dots[0][0]-dots[1][0]) == (double)((dots[2][1]-dots [3][1]) / (dots[2][0]-dots[3][0]))) return 1;
        if ((double)(dots[0][1]-dots[2][1]) / (dots[0][0]-dots[2][0]) == (double)((dots[1][1]-dots [3][1]) / (dots[1][0]-dots[3][0]))) return 1;
        if ((double)(dots[0][1]-dots[3][1]) / (dots[0][0]-dots[3][0]) == (double)((dots[2][1]-dots [1][1]) / (dots[2][0]-dots[1][0]))) return 1;
        
        return 0;
    }
}