happy coding

[level 0] 겹치는 선분의 길이 본문

coding study/programmars

[level 0] 겹치는 선분의 길이

yeoonii 2024. 5. 6. 18:45

문제 설명

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

겹치는 부분 => 교집합을 사용해야 겠다고 생각했고. 자연스럽게 set을 사용하게 되었다.

 

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

    ans = len((s1 & s2) | (s2 & s3) | (s1 & s3))
    
    return ans

 

다른 사람 코드

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

 

여기서 최대 최소로 하지 않으면 테스트케이스 3번의 경우, 갯수가 겹치는 일이 생긴다. 처음에 당연히 순서대로 받다가 오류를 ㅇㅁㅇ..

'coding study > programmars' 카테고리의 다른 글

[level 0] 짝수는 싫어요  (0) 2024.05.25
[level 0] 중앙값 구하기  (0) 2024.05.25
[level 0] 순서쌍의 개수  (0) 2024.05.06
[level 0] 전국 대회 선발 고사  (0) 2024.04.27
[level 0] 머쓱이보다 키 큰 사람  (0) 2024.04.14
Comments