[코딩테스트/Python] 프로그래머스 코딩테스트 - 나머지 한 점

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

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

 

입출력 예

 

문제 풀이

def solution(v):
    x = [i[0] for i in v] # x 좌표값들을 리스트로 모음
    y = [i[1] for i in v] # y 좌표값들을 리스트로 모음
    answer = []
    for i in x:
        if x.count(i) == 1: # x좌표 중 유일하게 있는 것
            answer.append(i)
    for i in y:
        if y.count(i) == 1:# y좌표 중 유일하게 있는 것
            answer.append(i)

    return answer

입출력 예를 들여다 보니, 어쨌든 사각형은 x좌표와 y좌표 어느 한 점에서 각 좌표쌍이 만나야 하는 것이므로 각 좌표끼리 모았을 때 짝이 없는 값이 정답에 들어가야 할 좌표라는 것을 알아냈다.

 

따라서, x/y 좌표끼리 리스트로 모은 다음에 짝이 없는(카운트가 1인) 값을 찾아 정답 리스트에 추가해 주었다.