프로그래머스(Programmers) Demo Test 나머지 한 점 - 파이썬(python)

문제 설명

직사각형을 만드는 데 필요한 4개의 점 중 3개의 좌표가 주어질 때, 나머지 한 점의 좌표를 구하려고 합니다. 점 3개의 좌표가 들어있는 배열 v가 매개변수로 주어질 때, 직사각형을 만드는 데 필요한 나머지 한 점의 좌표를 return 하도록 solution 함수를 완성해주세요. 단, 직사각형의 각 변은 x축, y축에 평행하며, 반드시 직사각형을 만들 수 있는 경우만 입력으로 주어집니다.

제한사항

• v는 세 점의 좌표가 들어있는 2차원 배열입니다.
• v의 각 원소는 점의 좌표를 나타내며, 좌표는 [x축 좌표, y축 좌표] 순으로 주어집니다.
• 좌표값은 1 이상 10억 이하의 자연수입니다.
• 직사각형을 만드는 데 필요한 나머지 한 점의 좌표를 [x축 좌표, y축 좌표] 순으로 담아 return 해주세요.

---

입출력 예

vresult

[[1, 4], [3, 4], [3, 10]]	[1, 10]
[[1, 1], [2, 2], [1, 2]]	[2, 1]

입출력 예 설명

입출력 예 #1
세 점이 [1, 4], [3, 4], [3, 10] 위치에 있을 때, [1, 10]에 점이 위치하면 직사각형이 됩니다.

입출력 예 #2
세 점이 [1, 1], [2, 2], [1, 2] 위치에 있을 때, [2, 1]에 점이 위치하면 직사각형이 됩니다.

 

 

제출 코드

x, y 각 축에 평행하다고 했으니 입력변수 v의 원소 중 한 개만 있는 걸 answer로 return함

import collections

def solution(v):
    x_coords = [elem[0] for elem in v]
    y_coords = [elem[1] for elem in v]
    x_cnt = collections.Counter(x_coords)
    y_cnt = collections.Counter(y_coords)
    answer = [[key for key, value in x_cnt.items() if value == 1][0],
               [key for key, value in y_cnt.items() if value == 1][0]]
    return answer