[프로그래머스/Lv.0] 겹치는 선분의 길이 | 파이썬

문제

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

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

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

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

 

제한사항

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

 

입출력 예

입출력 예 #1

• 두 번째, 세 번째 선분 [2, 5], [3, 9]가 [3, 5] 구간에 겹쳐있으므로 2를 return 합니다.

입출력 예 #2

• 겹친 선분이 없으므로 0을 return 합니다.

입출력 예 #3

• 첫 번째와 두 번째 선분이 [3, 5] 구간에서 겹칩니다.
• 첫 번째와 세 번째 선분 [1, 5] 구간에서 겹칩니다.
• 두 번째와 세 번째 선분 [3, 9] 구간에서 겹칩니다.
• 따라서 [1, 9] 구간에 두 개 이상의 선분이 겹쳐있으므로, 8을 return 합니다.

 

나의 풀이

def solution(lines):
    base = [0] * 200
    
    for line in lines:
        start, end = line
        
        for i in range(start, end):
            if i <= 0:
                base[abs(i)] += 1
            else:
                base[100+i] += 1

    stacked = list(filter(lambda x: x >= 2, base))

    return len(stacked)

선분의 최대 길이를 크기로 가지는 `base` 리스트를 생성해서

각 선분의 좌표가 위치하는 원소에 `+1`을 했다.

이때 음수는 `0~99`에, 양수는 `100~199`에 저장했다.

 

좌표의 위치를 `base` 리스트에 모두 카운트한 후에

`base` 리스트에서 `2` 이상을 값으로 가지는 원소(선분이 겹치는 부분)만 필터링한 후

필터링한 리스트의 길이를 최종 출력한다.

 

다른 사람의 풀이

def solution(lines):
    s1 = set([i for i in range(lines[0][0], lines[0][1])])
    s2 = set([i for i in range(lines[1][0], lines[1][1])])
    s3 = set([i for i in range(lines[2][0], lines[2][1])])
    return len((s1 & s2) | (s2 & s3) | (s1 & s3))

set의 교집합, 합집합 기능을 이용한 풀이!

선분을 각각 두 개씩 비교해서 교집합(겹치는 부분)을 구하고,

세 개의 교집합의 합집합을 구하면 두 개 이상의 선분이 겹치는 부분의 길이를 구할 수 있다.

 

회고

풀면서 `set()`을 이용하는 방식은 전혀 떠올리지 못했기 때문에 다른 사람의 풀이를 보고 감탄이 나왔다 😮