[프로그래머스 Lv.2] 교점에 별 만들기 (Python)

문제 링크: https://school.programmers.co.kr/learn/courses/30/lessons/87377

프로그래머스

풀이

레벨 2의 문제답게 크게 어렵지 않게 풀 수 있는 문제라고 생각한다.

 

아래 사진의 두 일차방정식의 교점을 구하는 공식을 알고 있기 때문에, 문제에 이를 적용함으로써 교점을 구하고 이를 사용하여 교점의 위치에 *을 출력하면 해결이 된다.

def solution(line):
    x_min = y_min = float('inf') # 그려야할 직사각형의 시작 위치를 알기 위한 최소값
    x_max = y_max = -float('inf') # 그러야할 직사각형의 끝 위치를 알기 위한 최대값
  
    
    meets = [] # 교점들
    answer = []

    for i in range(len(line)):
        first = line[i]
        for j in range(i+1, len(line)):
            second = line[j]
            
            if check_if_parallel(first, second): # 두 직선이 서로 평행이라면
                continue # 교점을 계산할 필요가 없음
            
            x, y = get_meet(first, second) # 교점을 계산
            
            if x.is_integer() and y.is_integer(): # 교점의 좌표가 정수면
                # 정수 체크 도중 float로 변경되었기에 이를 다시 정수로 변경
                x = int(x) 
                y = int(y)
                
                if (x, y) not in meets: # 이미 교점 리스트에 존재하면 추가하지 않음
                    meets.append((x, y))
                    
                # 최대값 최소값 갱신
                x_min, y_min = min(x, x_min), min(y, y_min) 
                x_max, y_max = max(x, x_max), max(y, y_max)

    for j in range(y_max, y_min - 1, -1): # 그려야하는 직사각형의 최대 y값부터 최저 y값까지 반복 (위에서 아래로)
        s = ""
        for i in range(x_min, x_max + 1): # 그려야하는 직사각형의 최소 x값부터 최대 x값까지 반복 (좌에서 우로)
            if (i, j) in meets: # 교점 위치에서는 *을 찍고
                s += "*"
            else: # 교점이 아니라면 .을 찍기
                s += "."
        answer.append(s)
            
    return answer


def check_if_parallel(first, second):
    return not (first[0] * second[1] - first[1] * second[0]) # ad - bc = 0인 경우에는 평행 또는 일치


def get_meet(first, second): # 공식을 사용하여 교점 구하기
    a = first[0]
    b = first[1]
    e = first[2]
    c = second[0]
    d = second[1]
    f = second[2]
    
    x = ((b*f-e*d)/(a*d-b*c))
    y = ((e*c-a*f)/(a*d-b*c))
    
    return x, y