[이코테/greedy]문자열 뒤집기

문제

다솜이는 0과 1로만 이루어진 문자열 S를 가지고 있다. 다솜이는 이 문자열 S에 있는 모든 숫자를 전부 같게 만들려고 한다. 다솜이가 할 수 있는 행동은 S에서 연속된 하나 이상의 숫자를 잡고 모두 뒤집는 것이다. 뒤집는 것은 1을 0으로, 0을 1로 바꾸는 것을 의미한다.
예를 들어 S=0001100 일 때,
전체를 뒤집으면 1110011이 된다.4번째 문자부터 5번째 문자까지 뒤집으면 1111111이 되어서 2번 만에 모두 같은 숫자로 만들 수 있다.
하지만, 처음부터 4번째 문자부터 5번째 문자까지 문자를 뒤집으면 한 번에 0000000이 되어서 1번 만에 모두 같은 숫자로 만들 수 있다.
문자열 S가 주어졌을 때, 다솜이가 해야하는 행동의 최소 횟수를 출력하시오.

 

내가푼답

strs = list(map(int,input()))
cnt=1 
for s in strs:
    ck = strs[0]
    if s != ck:
        ck = s
        cnt+=1
            
print(cnt//2)

 

정답

data = input()
count0 = 0 # 전부 0으로 바꾸는 경우
count1 = 0 # 전부 1로 바꾸는 경우

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

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

print(min(count0, count1))

 

풀이

내가 푼 답은 출제자가 의도한 풀이방법과 좀 다른 것 같기는 한데 일단 내 풀이를 설명해보자면

문자열에서 연속된 숫자를 한 덩이라고 쳤을 때,

구성요소가 0과 1 두 가지밖에 없기 때문에 결국 문자열이 얼마나 길어지든 상관없이 

'01'의 경우와 '010'의 경우로 나뉠거라고 생각했다. 

 

다시말해서, 0으로 이루어진 덩이와 1로 이루어진 덩이의 개수를 구했을 때

어떤 문자열이든 두 덩이의 개수가 같은 경우 or 둘 중 한덩이가 1개 더 많은 경우 로 나뉜다고 생각했고

 

0과 1의 구분 없이 그냥 덩이의 개수만 cnt 변수에 구했을 때, 

첫 번째 경우라면 (짝수가 될 것임) 어떤 숫자를 뒤집든 필요한 횟수는 똑같으니 그냥 2로 나누면 되고

두 번째 경우라면 (홀수가 될 것임) 최소 횟수를 구해야 하니까 -1을 한 후 2로 나누면 작은 수가 나올 것이다

하지만 두 경우 모두 몫을 구하는 n/2연산을 통해 답을 구할 수 있다.  (파이썬은 n//2 연산)

 

 

 

답지에서 요구한 풀이방법은

문자열을 전부 0으로 바꾸는 경우와 전부 1로 바꾸는 경우를 모두 구해서 최소값을 출력하는 것이었다. 

 

 

 

참조

이것이 취업을 위한 코딩테스트다(with. 파이썬)