라떼는말이야

[solved.ac 실버5] 1439_뒤집기 (파이썬, 그리디) 본문

알고리즘/코딩 테스트

[solved.ac 실버5] 1439_뒤집기 (파이썬, 그리디)

MangBaam 2022. 6. 14. 04:33
반응형

밑의 사진을 클릭하면 문제 링크로 이동합니다

제한 사항

 

문제

다솜이는 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만보다 작다.

 

출력

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

예제


나의 풀이

연속된 숫자를 하나로 압축해서 생각하면 훨씬 쉽다.

예를 들어 11001100110011000001 라는 숫자는 101010101 이라는 수로 압축해서 생각해보자.

101010101에서 맨 앞의 1을 뒤집으면 압축하기 전의 맨 앞 11을 뒤집는 것과 같은 효과이다.

101010101은 0이 4개, 1이 5개로 이루어졌다. 0을 다 뒤집으면 4번만에 모두 1이 되고, 1을 다 뒤집으면 5번만에 모두 0이 된다.

더 적은 수인 0을 4번 뒤집으면 된다.

정리하자면 다음과 같다.

  1. 연속된 숫자를 하나로 압축하자
  2. 1과 0의 개수를 세서 더 적은 것을 출력하자
num = input()
answer = [0, 0]
answer[int(num[0])] += 1
for i in range(1, len(num)):
    if num[i-1] != num[i]:
        if num[i] == '0': answer[0] += 1
        else: answer[1] += 1
print(min(answer))

입력된 숫자를 순회하며 이전의 수와 달라졌을 때 개수를 세어줬다. 이때 맨 앞의 숫자는 세지 못했기 때문에 먼저 맨 앞의 수를 더해준다.

answer 리스트의 0번 인덱스에는 0의 개수가, 1번 인덱스에는 1의 개수가 저장된다.

채점 결과

 

반응형
Comments