🤖 옌지니어의 IT 기록 🤖

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
def solution(lines):
    answer = 0
    count = [0]*200
    
    starts = []
    ends = []

    for line in lines:
        start, end = line

        starts.append(start)
        ends.append(end)

        for num in range(start, end):
            count[num+100] += 1

    for cnt in count[100+min(starts):100+max(ends)+1]:
        if cnt > 1:
            answer += 1

    return answer

풀이 # 1

• 원소가 0인 길이 200의 리스트 count 생성
  • 선분이 점 (-100,100) 범위에 있으므로 길이가 200
• 각 선분의 시작 및 끝 좌표를 담을 starts / ends 리스트 생성
• linelist 돌기
  • 각 선분의 시작 및 끝 좌표 추가
  • 시작과 끝 좌표 사이 숫자에 +1 (count 리스트 범위가 -100~100이므로 인덱스에 +100)
• 시작 좌표 중 가장 작은 수와 끝 좌표 중 가장 큰 수 범위에서 count가 2보다 크면 선분이 겹치는 것이므로 +1

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

풀이 #2

• 각 라인의 시작 좌표부터 끝 좌표까지의 숫자를 세트로 생성 (s1 / s2 / s3)
• 선분끼리의 교집합(&)들의 합집합(|) 길이를 출력

(-100,100) 범위를 만들어 놓는 걸 생각하지 못했었는데 범위를 만들고 나니 간단하게 풀 수 있는 문제였다. 굳이 시작과 끝 좌표의 min/max는 구하지 않아도 괜찮을 거 같다. 두 번째 풀이처럼 세트에서 합집합과 교집합을 이용하면 더 간단하게 풀 수 있는 문제인데...방법을 생각해 내는 게 항상 어렵다...