[코딩테스트 입문]Day22-2
겹치는 선분의 길이
문제 설명
선분 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 |
문제 풀이
<div class = “notice”, markdown = “1”>
set: 집합 자료형
set()의 괄호 안에 리스트를 입력하여 만들거나 문자열을 입력하여 만들 수도 있습니다.
set의 자료형의 특징은 두가지가 있습니다.
- 중복을 허용하지 않습니다. - 순서가 없습니다.
리스트나 튜플은 중복을 허용하고 순서가 있어서 인덱싱으로 요솟값을 얻을 수 있지만 set은 안됩니다. set의 교집합을 구할때는 &
을 통해서 구할 수 있습니다.
a = set([1, 2, 3, 4])
b = set([3, 4, 5])
print(a & b)
>>> {3, 4}
같은 결과를 만드는 것이 intersection
입니다.a.intersection(b)
하면 같은 값이 나옵니다.
합집합을 구하려면 |
를 사용합니다. 이때 중복된 값은 하나만 나타납니다.
print(a | b)
>>> {1, 2, 3, 4, 5}
-------------------
a.union(b)
>>> {1, 2, 3, ,4, 5}
union
을 통해서도 합집합을 구할 수 있습니다.
-(빼기)를 사용하면 차집합을 구할 수 있습니다.add
를 사용하면 만들어진 set자료형에 추가 가능합니다. 한번에 여러개를 추가하고 싶을 때는 update
를 사용합니다. 특정 값을 제거하고 싶을때는 remove
를 사용합니다.
</div>
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])
위 식에서 테스트 케이스 1번을 적용해보면 sets는 [{-1~1}, {1~3}, {3~9}]입니다. 그리고 &
를 이용해서 교집합을 구해서 |
을 통해 합집합을 구한 뒤 그 값들의 len
즉, 길이를 구하는 방식입니다.
댓글남기기