본문 바로가기
📖 Coding Test/Programmers LV.0

[프로그래머스 LV.0] 겹치는 선분의 길이

by hyebin (Helia) 2023. 1. 3.

겹치는 선분의 길이

문제 설명

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

입출력 예 설명

입출력 예 #1

  • 두 번째, 세 번째 선분 [2, 5], [3, 9]가 [3, 5] 구간에 겹쳐있으므로 2를 return 합니다.

입출력 예 #2

  • 겹친 선분이 없으므로 0을 return 합니다.

입출력 예 #3

  • 첫 번째와 두 번째 선분이 [3, 5] 구간에서 겹칩니다.
  • 첫 번째와 세 번째 선분 [1, 5] 구간에서 겹칩니다.
  • 두 번째와 세 번째 선분 [3, 9] 구간에서 겹칩니다.
  • 따라서 [1, 9] 구간에 두 개 이상의 선분이 겹쳐있으므로, 8을 return 합니다.            

제출

import Foundation

func solution(_ lines:[[Int]]) -> Int {
    var table = Array(repeating: Set<Int>(), count: 200)
    for i in 0..<lines.count{
        for j in lines[i][0]..<lines[i][1]{
            table[j+100].insert(i)
        }
    }
    return table.filter{$0.count > 1}.count
}
정수형 집합을 요소로 갖고 있는 배열 table을 선언한다. (-100 ≤ a < b ≤ 100 이기 때문에 200개)
lines 배열의 요소를 하나씩 비교하여 table에 몇 번째 선분인지 추가한다.
table 배열을 filter 함수를 사용하여 선이 2개 이상인 구간(선이 겹쳐있는 구간)을 계산한 후, 그 수를 반환한다.

 

반응형

댓글