백준 1439. 뒤집기 - 문제풀이 (Python/파이썬)

🔎 1439번. 문제 보기
https://www.acmicpc.net/problem/1439

💡 문제 풀기 전
1로 이루어진 쌍의 수와 0으로 이루어진 쌍의 수를 구한 다음,
둘 중에 더 적은 수의 쌍을 뒤집은 횟수를 답으로 출력하면 되겠다!!!

📋 코드 보기

from sys import stdin

nums = list(map(int, stdin.readline().rstrip()))
zero = 0
one = 0

for i in range(1, len(nums)):
    if nums[i] - nums[i - 1] == 1:
        one += 1
    elif nums[i] - nums[i - 1] == -1:
        zero += 1

print(one) if one > zero else print(zero)

🥕 코드 풀이 및 관련 개념

문제풀이 아이디어

그리 어렵지 않았던 문제!
입력 받은 수의 길이만큼 for문을 돌리는데, nums[i]와 nums[i-1]을 비교하기 위해
for문은 1부터 시작했다.

'0에서 1이' 또는 '1에서 0이' 나오는 경우는 총 2가지다.

nums[i] - nums[i - 1] == 1   #이건 0이 끝나고 1이 시작될 때
nums[i] - nums[i - 1] == -1  #이건 1이 끝나고 0이 시작될 때

경우에 따라 미리 만들어준 one과 zero에 +1을 해주고
마지막에 one과 zero 중 비교해서 더 큰 수를 출력해준다.