프로그래머스 - 겹치는 선분의 길이 **

dobyming·2022년 12월 29일
0

문제 설명

선분 3개가 평행하게 놓여 있습니다. 세 선분의 시작과 끝 좌표가 [[start, end], [start, end], [start, end]] 형태로 들어있는 2차원 배열 lines가 매개변수로 주어질 때, 두 개 이상의 선분이 겹치는 부분의 길이를 return 하도록 solution 함수를 완성해보세요.

lines가 [[0, 2], [-3, -1], [-2, 1]]일 때 그림으로 나타내면 다음과 같습니다.

https://grepp-programmers.s3.ap-northeast-2.amazonaws.com/files/production/e4122d8b-9ce2-49ce-a360-3d1284babd8a/line_2.png

선분이 두 개 이상 겹친 곳은 [-2, -1], [0, 1]로 길이 2만큼 겹쳐있습니다.


제한사항

  • lines의 길이 = 3
  • lines의 원소의 길이 = 2
  • 모든 선분은 길이가 1 이상입니다.
  • lines의 원소는 [a, b] 형태이며, a, b는 각각 선분의 양 끝점 입니다.
    • 100 ≤ a < b ≤ 100

입출력 예

linesresult
[[0, 1], [2, 5], [3, 9]]2
[[-1, 1], [1, 3], [3, 9]]0
[[0, 5], [3, 9], [1, 10]]8

내 코드

def solution(lines):
    table = [set([]) for _ in range(200)] # -100~100까지 각 선분들의 등장 count 초기화
    for index, line in enumerate(lines):
        x1, x2 = line
        for x in range(x1, x2):
            table[x + 100].add(index) # 선분에 음수가 들어있을 수도 있으므로 +100

    answer = 0
    for line in table:
        if len(line) > 1:
            answer += 1

    return answer

Level 0 치곤 너무 어렵게 느껴졌었다.. (나중에 다시 풀어 봐야할 문제)

“선분을 긋는다” 라는 것은 1차원 배열과 집합으로 표현
즉 (0, 2)에 선분을 긋는다 = 인덱스 0부터 1까지 해당 선분을 집합 자료형에 기록

for문을 돌때, 처음부터 끝까지 선분을 매번 긋는 방식으로 접근했다.
그리고 각 선분을 집합으로 받아서 그 길이(len)가 1 이상이면 count하는 방식으로 접근하는 것으로 잡았다.

다른 사람의 코드

def solution(lines):
    sets = [set(range(min(l), max(l))) for l in lines]
    return len(sets[0] & sets[1] | sets[0] & sets[2] | sets[1] & sets[2])

lines 배열의 각각의 최소-최대 값을 range로 돌려 선분들을 집합체로 담았습니다.

그리고 리턴값으로 각각의 집합체(선분)들을 & 연산자로 교집합을 구합니다.
최종적으로 겹치는 선분들을 | 연산자로 합집합을 구해 총 length를 return 합니다.

0개의 댓글