[그리디] 이코테 (파이썬) 문자열 뒤집기 풀이

[문제]

다솜이는 0과 1로만 이루어진 문자열 S를 가지고 있습니다. 다솜이는 이 문자열 S에 있는 모든 숫자를 전부 같게

 

만들려고 합니다. 다솜이가 할 수 있는 행동은 S에서 연속된 하나 이상의 숫자를 잡고 모두 뒤집는 것입니다.

 

뒤집는 것은 1을 0으로, 0을 1로 바꾸는 것을 의미합니다. 예를 들어 S = 0001100일 때는 다음과 같습니다.

 

1. 전체를 뒤집으면 1110011이 됩니다.

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

 

하지만, 처음부터 4번째 문자부터 5번째 문자까지 문자를 뒤집으면 한 번에 0000000이 되어서 1번 만에 모두

 

같은 숫자로 만들 수 있습니다.

 

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

 

[입력 조건]

1. 첫째 줄에 0과 1로만 이루어진 문자열 S가 주어집니다. S의 길이는 100만보다 작습니다.

 

[출력 조건]

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

<입력 예시>
0001100
<출력 예시>
1

 

[풀이]

s = input()

count0 = 0 # 전부 0으로 바꾸는 경우
count1 = 0 # 전부 1로 바꾸는 경우

# 첫 번째 원소에 대해서 처리
if s[0] == '1' :
  count0 += 1
else :
  count1 += 1

# 두 번째 원소부터 모든 원소를 확인하며
for i in range(len(s) - 1) :
  if s[i] != s[i + 1] :
    # 다음 수에서 1로 바뀌는 경우
    if s[i + 1] == '1' :
      count0 += 1
    # 다음 수에서 0으로 바뀌는 경우
    else :
      count1 += 1

print(min(count0, count1))

 

입력받은 문자열 s의 첫 번째 원소에 대하여 그 값이 1이라면 count0을 1 증가시키고 그렇지 않다면 count1을 1 증가시킨다.

 

반복문을 통해 두 번째 원소부터 모든 원소를 확인하는데, 현재의 값과 그 다음 값이 다를 경우 다시 조건문을 통해

 

다음 수가 1일 경우 0으로 바꿔준다 생각하여 count0을 1 증가시킨다. 반대로 다음 수가 0일 경우 1로 바꿔준다 생각하여

 

count1을 1 증가시킨다. 이와 같은 작업을 반복하고 최종적으로 count0과 count1 중 더 작은 값을 출력한다.

 

---

출처

이것이 코딩 테스트다 with 파이썬 - 나동빈 저