[Python] 백준 2138번: 전구와 스위치

https://www.acmicpc.net/problem/2138

2138번: 전구와 스위치

import sys
import copy
input = sys.stdin.readline


def switch(index, map):
    for i in range(index-1, index+2):
        if 0 <= i <= N-1:
            map[i] = not map[i]

    return map


def getSwitchCount(map, count):
    switchCount = count
    newMap = copy.deepcopy(map)

    for i in range(1, N):
        if newMap[i-1] == True:
            newMap = switch(i, newMap)
            switchCount += 1

    # 모두 False인지 확인
    for i in range(N):
        if newMap[i] == True:
            return -1

    return switchCount


N = int(input())

input1 = input().rstrip()
input2 = input().rstrip()
# True -> False로 바꿔야함
map = [input1[i] != input2[i] for i in range(N)]

switchCountWithoutSwitchingZero = getSwitchCount(map, 0)
if switchCountWithoutSwitchingZero != -1:
    print(switchCountWithoutSwitchingZero)
else:
    print(getSwitchCount(switch(0, map), 1))

 

우선 입력1, 입력2를 비교하여 다른 것들을 True, 같은 것들은 False로 저장하는 리스트를 하나 만든다. 목표는 모든 True를 False로 바꾸는 것이다.

다음은 getSwitchCount() 함수에 대한 설명이다. for문을 i에 대해 1~N-1까지 돌면서 i-1이 True이면 i에 대해 스위치 연산을 수행(switch() 함수)한다. 이후에 그렇게해서 바뀐 맵이 모두 False인지 확인하고 이에 따라 -1 또는 switchCount를 리턴해준다.

다만 특이점은, 1~N-1에 대해 for문을 돌기 때문에 0번째 인덱스에 대한 고려를 하지 못한다는 것이다. 따라서, 0번째 인덱스에 대해 2가지 경우를 나누어서 고려하면 된다.