백준 13413 오셀로 재배치 Kotlin (그리디)

문제 출처 : https://www.acmicpc.net/problem/13413

13413번: 오셀로 재배치

문제

로봇을 좋아하는 세희는 로봇동아리에서 카메라와 센서, 라즈베리 파이, 집게발을 이용해 로봇을 완성하였다. 이 로봇을 통해서 오셀로 재배치라는 작업을 하려고 한다. 오셀로 말은 앞면이 검정, 뒷면이 흰색으로 된 말이다. 세희의 목표는 로봇을 이용하여 처음 배치된 오셀로 말을 주어진 형태로 바꾸는 일을 하는 것이다. 아래의 예시를 참고하자.

초기 상태목표 상태

○●●○○

○●○●○

세희는 로봇을 이용해 2가지 작업 중 하나를 골라 진행할 수 있다.

1. 배치된 말 중 임의의 2개의 말을 골라 서로의 위치를 바꾼다.
2. 말 1개를 들어 뒤집어 놓아 색상을 변경한다.

위의 예시에서, 3번째와 4번째 말을 2번 작업을 통해 각각 뒤집으면 2번의 작업으로 목표 상태를 만들 수 있다. 하지만 1번 작업을 통해 3번째와 4번째 말을 골라 서로의 위치를 바꾸어주면 1번 만에 목표 상태에 도달할 수 있다. 초기 상태의 말과 목표 상태의 말이 주어질 때, 목표 상태에 도달할 수 있는 최소 횟수를 구하는 프로그램을 작성하시오.

입력

입력 데이터는 표준 입력을 사용한다. 입력은 T개의 테스트 데이터로 구성된다. 각 입력의 첫 번째 줄에는 오셀로 말의 개수 N(1 ≤ N ≤ 100,000)이 주어진다. 각 입력의 두 번째 줄과 세 번째 줄에는 각각 오셀로 말의 초기 상태와 목표 상태가 주어진다. 초기 상태와 목표 상태의 말의 개수는 항상 N과 일치한다. 흰색 면이 보이는 경우에는 W, 검은색 면이 보이는 경우에는 B로 주어진다.

출력

출력은 표준 출력을 사용한다. 입력받은 데이터에 대해, 한 줄에 1개씩 초기 상태에서 목표 상태를 만들기 위한 작업의 최소 횟수를 구한다.

알고리즘 분류

• 문자열
• 그리디 알고리즘

풀이

간단한 그리디 문제이다.

순수 완전 탐색이 가능한가? -> 아니다

- 그리디

- 백트래킹

- dp

등 고려

문제에서 주어진 초기 상태에서 목표 상태로 바꾸는 최솟값 구하기.

요 부분을 잘 생각해 보면 답이 나온다.

우선 초기 상태와 목표 상태에서 상태가 다른 말들을 찾자.

초기 상태를 기준으로 말이 다른 상태는 두 가지가 있다.

a.초기 상태가 B이고 목표 상태가 W인 경우

b.초기 상태가 W이고 목표 상태가 B인 경우

이럴 경우 a 상태와 b 상태를 서로 swap해 주면 한 번의 동작으로 두 개의 다른 상태를 해결할 수 있다.

만약 a 상태가 4개, b 상태가 2개 있다고 해 보자.

그럼 2개 까지는 서로 swap할 수 있고, 나머지 a 상태 2개는 말을 뒤집으면 된다.

따라서 a상태, b상태 둘 중 큰 값이 최소 횟수가 된다. 

 

코드

val br = System.`in`.bufferedReader()
fun getInt() = br.readLine().toInt()

fun main() = with(System.out.bufferedWriter()){

    repeat(getInt()){
        //input
        val n = getInt()
        val before = br.readLine()
        val after = br.readLine()
        val diff = intArrayOf(0,0)
        //solve
        for(i in before.indices){
            val a = before[i]
            val b = after[i]
            if(a!=b){
                if(a=='W') diff[0]++
                else diff[1]++
            }
        }
        //output
        write("${diff[0].coerceAtLeast(diff[1])}\n")
    }
    close()
}