[Swift Algorithm] 뒤집기 BOJ #1439

문제 설명

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

예를 들어 S=0001100 일 때,

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

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

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

 

입력

첫째 줄에 문자열 S가 주어진다. S의 길이는 100만보다 작다.

 

출력

첫째 줄에 다솜이가 해야하는 행동의 최소 횟수를 출력한다.

 

입출력 예제

입력

0001100

출력

1

 

내 문제 풀이

import Foundation

var input = Array(readLine()!).map({Int(String($0))!})
var zero = 0 // 0 블록의 개수
var one = 0 // 1 블록의 개수
var check = true // 현재 1 블록이 유지되고 있다면 true, 0 블록이 유지되고 있다면 false

if input[0] == 0 { check = false; zero += 1 }
else { one += 1 }

for i in input[1...input.count-1] {
    if i == 0 && check == true { check = false; zero += 1 }
    else if i == 1 && check == false { check = true; one += 1 }
}

zero > one ? print(one) : print(zero)

👉 이 문제는 굉장히 단순하게도, 0으로 연속된 블록과 1로 연속된 블록의 개수를 찾는 것이 바로 핵심!

• if, else를 이용하여 맨 처음 시작하는 숫자를 파악해준 후,
  for문을 이용하여 0블록의 개수와 1블록의 개수를 알아냈다.
• 그리고 구한 0블록, 1블록의 개수 중 더 작은 것의 개수를 출력했다.

 

💡 피드백

• 지문은 복잡해보이지만, 굉장히 단순한 문제라 금방 풀 수 있었다.

 

---

 

문제

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