점의 위치 구하기

2024. 8. 1. 18:00코딩테스트 입문

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr


문제 설명
사분면은 한 평면을 x축과 y축을 기준으로 나눈 네 부분입니다. 사분면은 아래와 같이 1부터 4까지 번호를매깁니다.

 

  • x 좌표와 y 좌표가 모두 양수이면 제1사분면에 속합니다.
  • x 좌표가 음수, y 좌표가 양수이면 제2사분면에 속합니다.
  • x 좌표와 y 좌표가 모두 음수이면 제3사분면에 속합니다.
  • x 좌표가 양수, y 좌표가 음수이면 제4사분면에 속합니다.

x 좌표 (x, y)를 차례대로 담은 정수 배열 dot이 매개변수로 주어집니다. 좌표 dot이 사분면 중 어디에 속하는지 1, 2, 3, 4 중 하나를 return 하도록 solution 함수를 완성해주세요.


제한사항

  • dot의 길이 = 2
  • dot[0]은 x좌표를, dot[1]은 y좌표를 나타냅니다
  • -500 ≤ dot의 원소 ≤ 500
  • dot의 원소는 0이 아닙니다.

입출력 예

dot result
[2, 4] 1
[-7, -9] 2

입출력 예 설명
입출력 예 #1

  • dot이 [2, 4]로 x 좌표와 y 좌표 모두 양수이므로 제 1 사분면에 속합니다. 따라서 1을 return 합니다.

입출력 예 #2

  • dot이 [-7, 9]로 x 좌표가 음수, y 좌표가 양수이므로 제 2 사분면에 속합니다. 따라서 2를 return 합니다.

dot의 x, y가 양수, 음수인지에 따라 어느 사분면에 속하는지 구하는 문제입니다.

총 4가지 조건에 따라서 속하는 위치를 구별해주면 되는 조건문입니다.

이중 if문을 사용하면 간단하게 풀 수 있습니다.

def solution(dot):
    # dot에서 x, y값을 가져옵니다.
    x, y = dot
    # x가 0보다 크다면 1, 4분면에 속합니다.
    if x > 0:
        # y가 0보다 크다면 1사분면
        if y > 0:
            return 1
        # y가 0보다 작다면 4사분면
        else:
            return 4
    # x가 0보다 작다면 2, 3분면에 속합니다.
    else:
        # y가 0보다 크다면 2사분면
        if y > 0:
            return 2
        # y가 0보다 작다면 3사분면
        else:
            return 3

 

다른 사람들의 풀이 중에 재밌는 풀이가 있어서 가져와봤습니다.

def solution(dot):
    quad = [(3,2),(4,1)]
    return quad[dot[0] > 0][dot[1] > 0]

조건 분기를 2번 사용하는 방법입니다.

True는 1, False는 0으로 판정되는 걸 사용한 건데요, dot[0]과 dot[1], 즉 x, y 값에 따라서 True-False 조건 분기를 설정해서 quad에서 값을 가져오는 방법을 사용했습니다.

 

이런 풀이는 볼 때마다 신선하네요.

'코딩테스트 입문' 카테고리의 다른 글

공 던지기  (0) 2024.08.03
2차원으로 만들기  (0) 2024.08.02
구슬을 나누는 경우의 수  (0) 2024.07.31
가위 바위 보  (0) 2024.07.30
모스부호 (1)  (0) 2024.07.29