[백준] (Swift) 3085번 - 사탕게임 (두번째 풀이)

🔖 TIL - swapAt(_ i: ,_  j: )

배열 내부에서 두 개의 원소의 자리를 바꿔야할 때가 있다. 그럴때 물론 아래와 같은 방법으로도 변경이 가능하다. 
temp라는 임시 변수에 현재 위치의 값을 저장해둔 후, 바뀐 자리에 temp를 넣어주는 방법이다.

var array = [1, 2, 3]
for i in 0..<2 {ㄷ
	var temp = array[i]
    array[i] = array[i+1]
    array[i+1] = temp
}

물론, 2차원 배열에서 다른 배열끼리의 값 바꿈이 필요하다면 이렇게 작성해야한다. 하지만, 한 배열 안에서 자리를 바꾸는 방식은 굳이 이렇게 하지 않고 swapAt() 이라는 메서드를 쓰면 간단하다.

var array = [1,2,3]
array.swayAt(1, 2)

이렇게 하게 되면, index가 1인 것과 2인것의 위치가 바뀐다. 위 예시에서는 2와 3의 위치가 바뀌겠지? 

 

🟠 문제

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

3085번: 사탕 게임

 

🟠 나의 풀이

아니 왜,, 도대체 왜 이런 두려움이 생겼는지 모르겟는데, 코드가 길어지면 안된다는 이상한 강박이 생겼다. 시간복잡도는 결코 길이와 관련되는게 아닌데 왜 이런 두려움이 생겼을까 어아아 아아아 머리아파! 일단 풀어보자.

놀랍게도 처음 한 풀이와 또 같은 코드가 작성되었다. (첫번째 풀이)

먼저, 예시를 2차원 배열로 입력 받을 것이다. 예제1번을 보면, 해당 입력은 [[C, C, P], [C, C, P], [P, P, C]] 이렇게 입력이 될 것이다. 여기서 행 방향이든, 열 방향이든 자리를 한번만 바꾸고, 인접한 것들의 max 값을 찾아주면 된다. 복잡하지 않았던게, 인접한 놈들을 세어줄 때도 행방향과 열방향으로만 진행되어 갯수가 세어지기 때문에 간단했다.

• for문을 이용해서 차례대로 행끼리 바꿔준다. (현재 기준 오른쪽 옆에있는 사탕과 자리 바꿔줌)
• 행끼리 인접한 애들의 갯수를 세어본다. 
• 열끼리 인접한 애들의 갯수를 세어본다.
• 다시 자리를 원래대로 돌려준다

• 다시 열끼리 자리를 바꿔준다. (위아래 위치한 사탕들의 위치를 바꿔줌)
• 행끼리 인접한 애들의 갯수를 세어본다.
• 열끼리 인접한 애들의 갯수를 세어본다.
• 다시 원래대로 자리를 돌려놔준다.

• (반복)

이렇게 구현했다!

🟠 정답코드

갯수를 세어주는 함수는 rowCount(), colCount()로 해서 따로 구현해줬다.

import Foundation

private func solution() {
    var result = [Int]()
    let n = Int(readLine()!)!
    var candy = [Array<Character>]()
    for _ in 0..<n {
        candy.append(Array(readLine()!))
    }

    for i in 0..<n {
        for j in 0..<n-1 {

            // 행끼리 바꿈
            candy[i].swapAt(j, j+1)
            result.append(colCount(n: n, candy: candy))
            result.append(rowCount(n: n, candy: candy))
            candy[i].swapAt(j, j+1)

            // 열바꿈
            var temp = candy[j][i]
            candy[j][i] = candy[j+1][i]
            candy[j+1][i] = temp
            result.append(colCount(n: n, candy: candy))
            result.append(rowCount(n: n, candy: candy))
            temp = candy[j][i]
            candy[j][i] = candy[j+1][i]
            candy[j+1][i] = temp
        }
    }
    print(result.max()!)
}

private func colCount(n: Int, candy: [Array<Character>]) -> Int {
    var result = -999
    for i in 0..<n {
        var count = 1
        for j in 0..<n-1 {
            if candy[i][j] == candy[i][j+1] {
                count += 1
            } else {
                result = max(result, count)
                count = 1
            }
        }
        result = max(result, count)
    }
    return result
}

private func rowCount(n: Int, candy: [Array<Character>]) -> Int {
    var result = -999
    for i in 0..<n {
        var count = 1
        for j in 0..<n-1 {
            if candy[j][i] == candy[j+1][i] {
                count += 1
            } else {
                result = max(result, count)
                count = 1
            }
        }
        result = max(result, count)
    }
    return result
}

solution()

 

 

💬 번외

그냥 colCount와 rowCount가 고작 i와 j 의 위치만 다른데, 코드가 중복되어 다른 함수로 존재하는게 마음에 안들었다. enum을 통해서 type을 분리해주고, 하나의 function으로 합치면 가독성이 더 좋아지지 않을까 생각되어, 그냥,, 그렇게 다시 해줬다..

import Foundation

private enum type {
    case colCount
    case rowCount
}

private func solution() {
    var result = [Int]()
    let n = Int(readLine()!)!
    var candy = [Array<Character>]()
    for _ in 0..<n {
        candy.append(Array(readLine()!))
    }

    for i in 0..<n {
        for j in 0..<n-1 {

            // 행끼리 바꿈
            candy[i].swapAt(j, j+1)
            result.append(countCloseCandy(type: type.rowCount, n: n, candy: candy))
            result.append(countCloseCandy(type: type.colCount, n: n, candy: candy))
            candy[i].swapAt(j, j+1)

            // 열바꿈
            var temp = candy[j][i]
            candy[j][i] = candy[j+1][i]
            candy[j+1][i] = temp
            result.append(countCloseCandy(type: type.rowCount, n: n, candy: candy))
            result.append(countCloseCandy(type: type.colCount, n: n, candy: candy))
            temp = candy[j][i]
            candy[j][i] = candy[j+1][i]
            candy[j+1][i] = temp
        }
    }
    print(result.max()!)
}

private func countCloseCandy(type: type, n: Int, candy: [Array<Character>]) -> Int {
    var result = -999
    for i in 0..<n {
        var count = 1
        for j in 0..<n-1 {
            switch type {
            case .colCount:
                if candy[i][j] == candy[i][j+1] {
                    count += 1
                } else {
                    result = max(result, count)
                    count = 1
                }
            case .rowCount:
                if candy[j][i] == candy[j+1][i] {
                    count += 1
                } else {
                    result = max(result, count)
                    count = 1
                }
            }
        }
        result = max(result, count)
    }
    return result
}

solution()

 

진짜 끝!