[프로그래머스] 겹치는 선분의 길이 (Python)

문제 설명

선분 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

입출력 예

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

입출력 예 설명

• 입출력 예 #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):
    answer = 0
    a,b,c = lines
    dict={}
    start = min(a[0],b[0],c[0])
    end = max(a[1],b[1],c[1])
    for i in range(start,end):
        dict[i]=0
        if a[0]<=i and i<a[1]:
            dict[i] += 1
        if b[0]<=i and i<b[1]:
            dict[i] += 1
        if c[0]<=i and i<c[1]:
            dict[i] += 1
    for i in dict.values():
        if i>1:
            answer += 1
    return answer

• 가독성은 좋으나, 너무 중복되는 코드가 많은 것 같다.

정답 풀이1 (교집합 원리)

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])

&는 집합의 교집합을, |는 집합의 합집합을 추출

정답 풀이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))

정답 풀이3 (정렬)

def solution(lines):
    starts = [min(a) for a in lines] # 시작점
    ends = [max(a) for a in lines] # 종료점
    # print(starts, ends) 
    # ex) [0, 2, 3] [1, 5, 9]
    starts.sort()
    ends.sort()
    answer = 0
    answer += max(0,ends[0] - starts[1])
    answer += max(0, ends[1] - starts[2])
    answer -= max(0, ends[0] - starts[2])
    return answer