그리디 알고리즘 : 문자열 뒤집기 [백준 1439, Kotlin]

문제 

다솜이는 0과 1로만 이루어진 문자열 S를 가지고 있다. 다솜이는 이 문자열 S에 있는 모든 숫자를 전부 같게 만들려고 한다. 다솜이가 할 수 있는 행동은 S에서 연속된 하나 이상의 숫자를 잡고 모두 뒤집는 것이다. 뒤집는 것은 1을 0으로, 0을 1로 바꾸는 것을 의미한다.

예를 들어 S=0001100 일 때,

1. 전체를 뒤집으면 1110011이 된다.
2. 4번째 문자부터 5번째 문자까지 뒤집으면 1111111이 되어서 2번 만에 모두 같은 숫자로 만들 수 있다.

하지만, 처음부터 4번째 문자부터 5번째 문자까지 문자를 뒤집으면 한 번에 0000000이 되어서 1번 만에 모두 같은 숫자로 만들 수 있다.

문자열 S가 주어졌을 때, 다솜이가 해야하는 행동의 최소 횟수를 출력하시오.

 

첫 번째 시도 

import java.io.BufferedReader
import java.io.InputStreamReader

fun main(){

    val br = BufferedReader(InputStreamReader(System.`in`))
    val numList = br.readLine().trim()
    val num = mutableListOf<Int>()
    numList.forEach {
        num.add(it.digitToInt())
    }

    var previous = num[1]
    var zeroCount = 0
    var oneCount = 0

    // 구간이 몇개 있는지를 확인
    if(previous == 0){
        zeroCount += 1
    }else{
        oneCount += 1
    }
    for(i in 2 until num.size - 1){
        if(num[i] != previous){
            if(previous == 0){
                oneCount += 1
                previous = 1
            }else{
                zeroCount += 1
                previous = 0
            }
        }
    }

    // 0인 경우의 수
    if(num.count{ it == 0} < num.count{ it == 1}){
        print(zeroCount)
    }else{
        print(oneCount)
    }
}

 

두 번째 시도

import java.io.BufferedReader
import java.io.InputStreamReader

fun main(){

    val br = BufferedReader(InputStreamReader(System.`in`))

    val numList = br.readLine().map{ it.digitToInt() }
    var previous = numList[0]
    var zeroCount = 0 // 0의 개수
    var oneCount = 0 // 1의 개수

    // 구간이 몇개 있는지를 확인
    if(previous == 0){
        zeroCount += 1
    }else{
        oneCount += 1
    }
    for(i in 1 until numList.size){
        if(numList[i] != previous){ // 이전 숫자와 현재 숫자가 다른 경우
            if(previous == 0){
                oneCount += 1
                previous = 1
            }else{
                zeroCount += 1
                previous = 0
            }
        }
    }
    if(zeroCount < oneCount) print(zeroCount) else print(oneCount)
}

 

풀이 

하나의 숫자로 통일되도록 뒤집는 문제이다. 뒤집는 것이 결국 하나의 연속된 구간을 뒤집는 것이기에, 연속된 구간을 찾는 문제로 생각했다. 0과 1중 연속된 구간이 가장 적은 수가 최솟값이 된다. 

 

https://github.com/jeongjaino/Problem_solving/blob/master/src/main/kotlin/day1/Day1-3.kt