문제
도현이는 바구니를 총 N개 가지고 있고, 각각의 바구니에는 1번부터 N번까지 번호가 순서대로 적혀져 있다. 바구니는 일렬로 놓여져 있고, 가장 왼쪽 바구니를 1번째 바구니, 그 다음 바구니를 2번째 바구니, ..., 가장 오른쪽 바구니를 N번째 바구니라고 부른다.
도현이는 앞으로 M번 바구니의 순서를 역순으로 만들려고 한다. 도현이는 한 번 순서를 역순으로 바꿀 때, 순서를 역순으로 만들 범위를 정하고, 그 범위에 들어있는 바구니의 순서를 역순으로 만든다.
바구니의 순서를 어떻게 바꿀지 주어졌을 때, M번 바구니의 순서를 역순으로 만든 다음, 바구니에 적혀있는 번호를 가장 왼쪽 바구니부터 출력하는 프로그램을 작성하시오.
입력
첫째 줄에 N (1 ≤ N ≤ 100)과 M (1 ≤ M ≤ 100)이 주어진다.
둘째 줄부터 M개의 줄에는 바구니의 순서를 역순으로 만드는 방법이 주어진다. 방법은 i j로 나타내고, 왼쪽으로부터 i번째 바구니부터 j번째 바구니의 순서를 역순으로 만든다는 뜻이다. (1 ≤ i ≤ j ≤ N)
도현이는 입력으로 주어진 순서대로 바구니의 순서를 바꾼다.
출력
모든 순서를 바꾼 다음에, 가장 왼쪽에 있는 바구니부터 바구니에 적혀있는 순서를 공백으로 구분해 출력한다.
예제 입력 1
5 4
1 2
3 4
1 4
2 2
예제 출력 1
3 4 1 2 5
풀이
replaceSubrange를 사용
import Foundation
let MN = readLine()!.split(separator: " ").map { Int($0)! }
let m = MN[1]
var buckets = [Int](1...MN[0])
for _ in 0...m - 1 {
let change = readLine()!.split(separator: " ").map { Int($0)! }
let start = change[0] - 1
let end = change[1] - 1
buckets.replaceSubrange(start...end, with: buckets[start...end].reversed())
}
buckets.map { num in
print(num, terminator: " ")
}
- 분명 array 에서 특정 index를 수정할 수 있는 방법이 있을 거라고 생각했는데 replaceSubrange가 있었다.
- 처음에는 newArr를 사용해서 직접 넣으려 했는데 더 간단하게
array[closedRange] 타입을 사용해서 수정할 수 있었다.
for문을 사용한 정렬
import Foundation
let MN = readLine()!.split(separator: " ").map { Int($0)! }
let m = MN[1]
var buckets = [Int](1...MN[0])
for _ in 0...m - 1 {
let change = readLine()!.split(separator: " ").map { Int($0)! }
let start = change[0] - 1
let end = change[1] - 1
for i in start..<end {
let endNum = buckets[end]
buckets.remove(at: end)
buckets.insert(endNum, at: i)
}
}
buckets.map { num in
print(num, terminator: " ")
}
- 사실상 replaceSubrange를 직접 구현한 것과 같다.
replaceSubrange, Array[n...m] 을 사용하는 방법을 알게 되었다...
'Algorithm' 카테고리의 다른 글
[Algorithm] Swift ASCII 출력하기 (0) | 2023.11.09 |
---|---|
[Algorithm] 백준 1546번 평균 (0) | 2023.11.08 |
[Algorithm] 백준 3052번 나머지 (1) | 2023.11.06 |
[Algorithm] 백준 5597번, 과제 안 내신분..? (0) | 2023.11.04 |
[알고리즘]백준 10813번 : 공 바꾸기 (1) | 2023.10.23 |