[Greedy] 이코테 - 문자열 뒤집기(py)

문제

1439번: 뒤집기

 

---

풀이

 

- 내 풀이

s = input()
one = s.split('0')
zero = s.split('1')
one_len = 0
zero_len = 0

for i in one:
  if i != '':
    one_len += 1
for i in zero:
  if i != '':
    zero_len += 1
    
print(zero_len) if zero_len < one_len else print(one_len)

0과 1로 이루어진 문자열의 연속된 부분들을 부분 집합으로 생각해

바꿔야 할 최소 횟수(다솜이가 해야 하는 행동의 최소 횟수)는 0 또는 1의 부분 집합 개수 중 더 작은 부분 집합의 개수와 같다는 아이디어를 가지고 풀었다.

 

- 답안

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로 바꾸는 경우 중에서 더 적은 횟수를 가지는 경우를 계산하는 방식.

전체 리스트의 원소를 앞에서부터 하나씩 확인하며 0에서 1로 변경하거나 1에서 0으로 변경하는 경우를 확인하는 방식으로 해결한다.

 

최소 횟수는 min()으로 간단하게 구할 수 있다.

 

두 풀이 메모리, 시간은 모두 같다.

(tistory 코드블럭 입력 화면에서는 코드가 정상적으로 보이다 나온 화면에서는 들여쓰기가 이상하게 표현이 되는데 tab 대신 띄어쓰기 4번으로 바꾸니 정상적으로 잘 보인다.)

 

---

 

출처: 나동빈, 「이것이 취업을 위한 코딩테스트다 with 파이썬」, 한빛미디어, p.313