알고리즘 문제풀이/파이썬

[이코테-그리디] 기출 - 03. 문자열 뒤집기

Ella_K 2022. 5. 6. 18:15

[문제]

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

예를 들어 S=0001100일 때는 다음과 같습니다.

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

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

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

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


[입력 조건]

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

 

[출력 조건]

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

 

<입력 예시>
0001100

<출력 예시>
1

 


[책 풀이]

🔎 아이디어

  • 문자열을 모두 0으로 바꾸는 경우와 모두 1로 바꾸는 경우 2가지 모두 고려해서 더 적은 값을 출력한다

 

🔎 코드 설계

arr = input()

count0 = 0 # 문자열 모두 0으로 바꾸는 경우
count1 = 0 # 문자열 모두 1로 바꾸는 경우

# 첫번째 원소에 대하여
if arr[0] == '1':
  count0+=1
else:
  count1+=1

# 두번째 원소부터 마지막까지 확인
for i in range(1,len(arr)-1):
  if arr[i] != arr[i+1]: # 현재 수와 다음수가 같지 않을 때 다음수를 현재수와 같게 바꾼다.
    # 다음 수를 0으로 바뀌는 경우
    if arr[i+1] == '1':
      count0+=1
    # 다음 수를 1로 바뀌는 경우
    else:
      count1+=1

print(min(count0,count1))

 

 


출처: 이것이 취업을 위한 코딩 테스트다 with 파이썬

* 포스팅에 책의 내용과 개인적인 해석이 같이 포함됩니다.