[백준] (Swift) 3085번 - 사탕게임 (오답노트必)

문제링크

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

3085번: 사탕 게임

 

풀이

• 좌우/상하 로 인접한 사탕을 바꿀 수 있다.

표[1]

• 표[1]과 같이 사탕이 주어졌다. 여기서 [0][0] 사탕과 [0][1] 사탕을 바꾸어보자.

표[2]

• 표[2] 와 같은 결과가 나온다. 이렇게 되면 행끼리 인접한 사탕의 최고 개수는 2, 열끼리 인접한 사탕의 개수는 3이 된다.
• 다시 표[1]로 돌아가서 표[1]에서 [0][0] 사탕과 [1][0] 사탕을 바꾸어보자.

표[3]

• 표3을 보면, 행끼리 인접한 사탕 최고 개수는 2개, 열끼리 인접한 사탕 최고 개수는 1개 이다.

이 위의 과정을 for 문을 통해서 반복해주었다. 그리고 인접한'최대의 사탕 개수'를 구하는 func을 만들었다.

▶ swapAt()

인덱스에 해당하는 값을 서로 바꾸어주는 메서드이다. 자리를 바꾸어줄 때, 복잡하게 코드로 작성하지 않고 해당 메서드를 사용하면 간단하게 인덱스 자리의 값들의 자리를 바꿀 수 있다.

아래와 같이 swapAt괄호 안에 인덱스 두개를 입력하면, 그 두개의 값이 바뀌어 출력된다.

var i = ["A", "B", "C"]

i.swapAt(1, 2)
print(i)
// prints ["A", "C", "B"]

https://developer.apple.com/documentation/swift/array/2893281-swapat

Apple Developer Documentation

 

문제 풀이 코드

• 먼저 행끼리, 열끼리 인접한 최고 개수를 세어주는 함수를 만들었다. 
• 어차피 최종 result 는 최대값을 구하는 것이기 때문에 기존 result 에  계속 max 메서드를 활용하여 초기화 시켜준다.

func row_count() {
    for i in 0..<n {
        var count = 1
        for j in 0..<n-1 {
            if input[i][j] == input[i][j+1] {
                count += 1
            } else {
                result = max(result, count)
                count = 1
            }
            result = max(result,count)
        }
    }
}

func col_count() {
    for i in 0..<n {
        var count = 1
        for j in 0..<n-1 {
            if input[j][i] == input[j+1][i] {
                count += 1
            } else {
                result = max(result, count)
                count = 1
            }
            result = max(result, count)
        }
    }
}

• 그 다음, main 코드를 짠다.
• 2차원 배열이기 때문에 for문 두번을 이용했고, 행끼리 바꿔서 result를 구해주고, 열끼리 바꾸어 result를 구해주었다.
• 그 결과 최종적으로 result를 출력해준다.

for i in 0..<n {
    for j in 0..<n-1 {
    
        // 자리 바꿔서 result 갱신해주고
        input[i].swapAt(j,j+1)
        row_count()
        col_count()
        //다시 자리 바꿔줌
        input[i].swapAt(j,j+1)
        
        //열끼리 자리 바꿀건데, 값을 temp 에 저장해뒀다가 값을 줘야해.
        var temp = input[j][i]
        input[j][i] = input[j+1][i]
        input[j+1][i] = temp
        
        row_count()
        col_count()
        
        // 원래자리로 돌려
        temp = input[j][i]
        input[j][i] = input[j+1][i]
        input[j+1][i] = temp
     
    }
}

print(result)

 

회고

• 처음푸는 유혀형이라 어떻게 해야할지 몰라서 다른 사람의 코드를 참고하여 풀었다.
• 추후 다시 오답노트로 다룰 예정이다!!
• 완전탐색이 진짜 하나하나 다 해보는 방식이구나!