💫겹치는 선분의 길이 + 스위핑

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

[나의 풀이]

(틀린 풀이)

def solution(lines):
    lines.sort(key=lambda x: x[0])  
    a, b, c = lines
    
    aa = []
    bb = []
    
    # a와 b의 겹치는 구간
    for i in range(a[0], a[1] + 1):
        if i >= b[0] and i <= b[1]:
            aa.append(i)
    
    # b와 c의 겹치는 구간
    for j in range(b[0], b[1] + 1):
        if j >= c[0] and j <= c[1]:
            bb.append(j)
    
    # aa와 bb의 합집합
    cc = list(set(aa + bb))
    
    # aa와 bb의 교집합
    intersection = set(aa) & set(bb)
    
    if intersection:  # 교집합이 있을 때
        return len(cc) - 1
    elif not aa and not bb:  # aa와 bb 둘 다 공집합일 때
        return 0
    elif not aa or not bb:  # 둘 중 하나가 공집합일 때
        return len(cc) - 1
    else:  # 교집합이 없고 둘 다 공집합이 아닐 때
        return len(cc) - 2

 

모든 케이스를 커버한다고 생각했는데 실패..

왜 안 될까? 생각해봤는데 [[1,10],[2,5],[4,7]]같은 케이스에는 만족하지 않을 수 있겠다.

 

(올바른 풀이)

def solution(lines):
    # 각 선분의 시작점과 끝점을 이벤트로 등록
    events = []
    for start, end in lines:
        events.append((start, 1))  # 시작점은 +1
        events.append((end, -1))  # 끝점은 -1
    
    # 이벤트를 좌표 기준으로 정렬 (같은 좌표면 끝점(-1)이 시작점(+1)보다 앞서도록 정렬)
    events.sort()

    # 스위핑을 통해 겹치는 구간 계산
    overlap_count = 0
    total_length = 0
    prev_position = None

    for position, event in events:
        if prev_position is not None and overlap_count >= 2:
            # 이전 위치부터 현재 위치까지 두 선분 이상 겹치는 구간의 길이 추가
            total_length += position - prev_position

        # 겹치는 선분의 수 업데이트
        overlap_count += event
        prev_position = position

    return total_length

 

[다른 풀이]

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

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

 

def solution(lines):
    starts = [min(a) for a in lines]
    ends = [max(a) for a in lines]
    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