[Swift] 프로그래머스 기지국 설치

Problem

코딩테스트 연습 - 기지국 설치

---

Solution

1. stations를 순회하며 전파가 오지 않는 구간을 구합니다.

전파가 안오는 구간은 현재 설치된 기지국들의 전파 도달거리 사이입니다.

문제에 나왔던 예로 들면 4와 11에 현재 기지국들이 설치되어 있고 전파 도달거리 사이는 1~2, 6~9 입니다.

2. 전파가 오지 않는 구간에 기지국이 몇 개 필요한지 계산한다.

전파가 오지 않는 구간의 수가 n이라면 필요한 기지국의 수는 n/w*2+1 을 올림한 수입니다.

위의 예에서 6~9까지는 n = 4이고 4/1*1+1 은 1.33333... 이므로 2개의 기지국이 필요합니다.

3. 현재 설치된 가장 마지막 기지국과 n까지의 빈 공간 수를 구한다.

현재 설치된 기지국들의 사이를 구하고 가장 마지막 기지국과 n(끝) 사이의 수를 구합니다.

(아래로 예를 들면 12~16입니다.)

4. 필요한 기지국 수를 반환한다.

위에서 구한 필요한 기지국 수들을 모두 더해 반환해줍니다.

---

Source Code

	import Foundation
	func solution(_ n:Int, _ stations:[Int], _ w:Int) -> Int{
	var baseStationCount:Int = 0
	var start:Int = 1
	for station in stations + [n+w+1] {
	let end = station-(w+1)
	let emptyCount = Int(ceil(Double(end - start + 1)/Double(w*2+1)))
	baseStationCount += emptyCount > 0 ? emptyCount : 0
	start = station + w + 1
	}
	return baseStationCount
	}

view raw 기지국 설치.swift hosted with ❤ by GitHub