선분 3개가 평행하게 놓여 있습니다. 세 선분의 시작과 끝 좌표가 [[start, end], [start, end], [start, end]] 형태로 들어있는 2차원 배열 lines
가 매개변수로 주어질 때, 두 개 이상의 선분이 겹치는 부분의 길이를 return 하도록 solution 함수를 완성해보세요.
lines
가 [[0, 2], [-3, -1], [-2, 1]]일 때 그림으로 나타내면 다음과 같습니다.
선분이 두 개 이상 겹친 곳은 [-2, -1], [0, 1]로 길이 2만큼 겹쳐있습니다.
lines
의 길이 = 3lines
의 원소의 길이 = 2lines
의 원소는 [a, b] 형태이며, a, b는 각각 선분의 양 끝점 입니다.lines | result |
---|---|
[[0, 1], [2, 5], [3, 9]] | 2 |
[[-1, 1], [1, 3], [3, 9]] | 0 |
[[0, 5], [3, 9], [1, 10]] | 8 |
def solution(lines):
table = [set([]) for _ in range(200)] # -100~100까지 각 선분들의 등장 count 초기화
for index, line in enumerate(lines):
x1, x2 = line
for x in range(x1, x2):
table[x + 100].add(index) # 선분에 음수가 들어있을 수도 있으므로 +100
answer = 0
for line in table:
if len(line) > 1:
answer += 1
return answer
Level 0 치곤 너무 어렵게 느껴졌었다.. (나중에 다시 풀어 봐야할 문제)
“선분을 긋는다” 라는 것은 1차원 배열과 집합으로 표현
즉 (0, 2)에 선분을 긋는다 = 인덱스 0부터 1까지 해당 선분을 집합 자료형에 기록
for문을 돌때, 처음부터 끝까지 선분을 매번 긋는 방식으로 접근했다.
그리고 각 선분을 집합으로 받아서 그 길이(len
)가 1 이상이면 count하는 방식으로 접근하는 것으로 잡았다.
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])
lines
배열의 각각의 최소-최대 값을 range로 돌려 선분들을 집합체로 담았습니다.
그리고 리턴값으로 각각의 집합체(선분)들을 &
연산자로 교집합을 구합니다.
최종적으로 겹치는 선분들을 |
연산자로 합집합을 구해 총 length를 return 합니다.