Swift - Algorithm
스위프트를 사용하여 알고리즘 문제를 풀고 기록합니다.
📈 백준
백준 - 알고리즘 분류 - 수학
🔢 수학
1000 - A+B
if let input = readLine() {
let numbers = input.split(separator: " ").compactMap { Int($0) }
let sum = numbers.reduce(0, +)
print(sum)
}
if let 바인딩(강제 언래핑(!)에 비해 안전한 방식)을 사용해서 nil 체크와 값 추출을 동시에 수행했다. 입력이 nil인 경우 블록 내부의 코드가 실행되지 않으므로, 런타임 오류를 방지할 수 있다.
compactMap은 map과 유사하게 배열의 각 요소를 변환하여 새로운 배열을 생성하지만, 변환 작업 후에 nil인 요소를 제거한다는 것에 차이가 있다. compactMap을 사용함으로써, 변환 작업 후에 유효하지 않은 요소들을 제거하고 유효한 정수 값들로만 이루어진 배열을 얻을 수 있다.
reduce는 배열의 모든 요소를 합하는 기능을 제공한다.
2739 - 구구단
if let input = readLine(), let n = Int(input) {
Array(1...9).forEach { print("\(n) * \($0) = \(n * $0)") }
}
콤마(,)를 통해 두 개의 옵셔널 바인딩을 동시에 수행했다.
1...9 범위를 사용하여 1부터 9까지의 숫자를 나타내는 배열을 생성했다. forEach 메서드를 사용하여 배열의 각 요소에 대해 클로저를 반복적으로 실행했다. 추후에 스위프트의 클로저와 자바스크립트의 클로저를 비교해서 정리할 때, 더 자세히 다뤄볼 예정이다.
2753 - 윤년
if let input = readLine(), let num = Int(input) {
if (num % 400 == 0) || (num % 4 == 0 && num % 100 != 0) {
print("1")
} else {
print("0")
}
}
주어진 연도가 윤년인 경우는 아래 두 가지 조건 중 하나를 만족할 때이다. 조건 1 (num % 400 == 0) : 400의 배수일 때 조건 2 (num % 4 == 0 && num % 100 != 0) : 4의 배수이면서 100의 배수가 아닐 때
25304 - 영수증
guard let x = Int(readLine()!), let n = Int(readLine()!) else {
fatalError("error")
}
var price = 0
for _ in 0..<n {
guard let input = readLine()?.split(separator: " ").compactMap({ Int($0) }), input.count == 2 else {
fatalError("error")
}
price += input[0] * input[1]
}
price == x ? print("Yes") : print("No")
readLine() 함수는 한 번에 한 줄씩만 입력을 받기 때문에 여러 줄을 입력 받기 위해서는 여러번의 호출이 필요하다.
guard let 구문을 사용한 옵셔널 바인딩을 수행했다. guard let 구문은 if let 구문과는 달리, 옵셔널 바인딩이 실행되는 블록 밖에서도 바인딩된 값을 사용할 수 있다는 차이가 있다.
guard let 구문에서 여러 개의 옵셔널 바인딩을 동시에 수행하거나 여러 개의 조건을 동시에 확인하기 위해 콤마를 사용할 수 있다.
10039 - 평균 점수
var sum = 0
for _ in 0..<5 {
guard let score = Int(readLine()!) else {
fatalError("error")
}
sum += max(score, 40)
}
let average = sum / 5
print(average)
각 점수를 독립적으로 옵셔널 바인딩하고 최저 점수 조건을 수행하기 위해 여러 개의 옵셔널 바인딩을 동시에 수행하는 것이 아닌, for 구문을 통해 반복적으로 수행하는 방식을 택했다.
복합 할당 연산자 +=를 통해 변수 sum을 갱신시켰다. 참고로 스위프트에서는 ++을 지원하지 않기 때문에 +=1을 사용해야 한다.
2839 - 설탕 배달
if let input = readLine(), let N = Int(input) {
var result = -1
let maxFiveBags = N / 5
for i in stride(from: maxFiveBags, through: 0, by: -1) {
let remainder = N - (5 * i)
if remainder % 3 == 0 {
result = i + (remainder / 3)
break
}
}
print(result)
}
stride 함수는 일정한 간격으로 반복하고자 할 때, 일반적인 for 구문보다 더 간결하고 더 가독성이 좋은 코드를 작성할 수 있도록 도와준다. 또한 stride 함수는 반복 범위 내에서 실제로 필요한 값만 생성하므로 메모리와 연산 비용을 절약할 수 있다.
1002 - 터렛
if let N = Int(readLine()!) {
for _ in 0..<N {
if let input = readLine() {
let values = input.split(separator: " ").compactMap { Int($0) }
if values.count == 6 {
let x1 = values[0]
let y1 = values[1]
let r1 = values[2]
let x2 = values[3]
let y2 = values[4]
let r2 = values[5]
let d = (x2 - x1) * (x2 - x1) + (y2 - y1) * (y2 - y1)
let d1 = (r1 + r2) * (r1 + r2)
let d2 = (r1 - r2) * (r1 - r2)
if x1 == x2 && y1 == y2 {
if r1 == r2 { print(-1) }
else { print(0) }
} else {
if d > d1 || d < d2 { print(0) }
else if d == d1 || d == d2 { print(1) }
else { print(2) }
}
}
}
}
}
다른 분들 풀이를 보면 종종 튜플 할당 let (x1, y1, r1, x2, y2, r2) = (values[0], values[1], values[2], values[3], values[4], values[5])을 사용하는 것을 볼 수 있는데, 확인해보면 위 코드처럼 개별적인 상수 할당이 더 빠르게 동작(12ms -> 8ms)하는 것을 볼 수 있다. 튜플 할당은 각 변수에 접근할 때마다 튜플에서 해당 요소를 추출해야 하기 때문에 추가적인 연산과 메모리 접근을 필요로 하는 것에 반해, 개별적인 상수 할당은 배열 요소에 직접 접근하는 방식이다.
🪜 구현
1966 - 프린터 큐
struct Document {
let priority: Int
let isTarget: Bool
}
func findPrintOrder(documents: [Document]) -> Int {
var printQueue = documents
var printOrder = 0
while !printQueue.isEmpty {
let highestPriority = printQueue.max(by: { $0.priority < $1.priority })!.priority
let front = printQueue.removeFirst()
if front.priority < highestPriority {
printQueue.append(front)
} else {
printOrder += 1
if front.isTarget {
return printOrder
}
}
}
return printOrder
}
let T = Int(readLine()!)!
for _ in 0..<T {
let input = readLine()!.split(separator: " ").map { Int($0)! }
let targetIndex = input[1]
let priorities = readLine()!.split(separator: " ").map { Int($0)! }
var documents = [Document]()
for (index, priority) in priorities.enumerated() {
let isTarget = (index == targetIndex)
documents.append(Document(priority: priority, isTarget: isTarget))
}
let printOrder = findPrintOrder(documents: documents)
print(printOrder)
}
max() 함수의 by는 (비교 연산을 정의하는) 클로저를 받는 매개변수이다. 위에서 클로저 내의 $0과 $1은 클로저에서 첫 번째와 두 번째 매개변수를 나타낸다.
스위프트에서 enumerated() 함수는 컬렉션의 각 요소에 대해 인덱스와 값을 함께 제공하는 반복자를 생성하는 데 사용된다.
11866 - 요세푸스 문제 0
func josephus(_ n: Int, _ k: Int) {
var cirBuffer = Array(1...n)
var permutation: [String] = []
var idx = 0
while !cirBuffer.isEmpty {
idx = (idx + k - 1) % cirBuffer.count
let removedElement = cirBuffer.remove(at: idx)
permutation.append(String(removedElement))
}
print("<" + permutation.joined(separator: ", ") + ">")
}
let input = readLine()!.split(separator: " ").map { Int($0)! }
let n = input[0], k = input[1]
josephus(n, k)
조세푸스 순열(Josephus Permutation)은 주어진 인원에서 일정한 간격으로 요소를 제거하는 과정을 반복하여 마지막에 남는 요소들로 이루어진 순열을 말한다.
