BOJ: #3048 - 개미

두 그룹을 더할 때, 튜플을 이용해서 방향을 같이 저장했다. false는 왼쪽에서 오른쪽, true는 오른쪽에서 왼쪽으로 이동하는 개미를 뜻한다.

처음엔 이중 for문을 이용했는데, 개미들을 스왑하는 경우 i를 한 번 더 건너뛰어야 해서 while문으로 변경했다. 그리고 서로 방향이 다른 경우에 전부 스왑해줬었는데, 7% 쯤에서 틀려서 다른 풀이를 찾아봤다. 왼쪽으로 이동하고 있는 개미랑 만난 경우에는 바꿔줄 필요가 없다는 거였다. (←, →) 서로 만나지 않으니까! 그래서 앞에 있는 개미가 오른쪽을 보고 있을 때 조건도 추가했다.

문제

https://www.acmicpc.net/problem/3048

내가 작성한 코드

Swift

let n = readLine()!.split(separator: " ").map { Int(String($0))! }
let a = Array(readLine()!.reversed())
let b = Array(readLine()!)
let t = Int(readLine()!)!
var c = a.map { return ($0, false) } + b.map { return ($0, true) }
let m = n.reduce(0, +)

for _ in 0..<t {
    var i = 1
    while i < m {
        if c[i].1 != c[i-1].1 && c[i - 1].1 == false { // 방향이 다른 경우
            c.swapAt(i, i-1)
            i += 1
        }
        i += 1
    }
}

print(c.map { String($0.0) }.joined())