9-alstjr7437

[alstjr7437]

🔗 문제 링크

전구와 스위치

✔️ 소요된 시간

1시간 20분

✨ 수도 코드

그리디의 개념을 잘 생각하면서 전구를 쳐다보고 있으니 문뜩 떠오르다구요..
처음에는 예제에서 000 -> 010 만들기 였는데
0 0 0 -> 1 1 0 -> 1 0 1 -> 0 1 0 이 있고
0 0 0 -> 1 1 0 -> 0 0 1 -> 0 1 0 이 있었습니다.
위 두개의 차이점으로

1. 첫번째는 0번 2번 1번을 누르는 순서로 가져갔고
2. 두번째는 0번 1번 2번을 누르는 순서로 가져갔습니다!

그렇게 해서 앞 쪽 전구부터 계속해서 결과값에 맞춰가면 되지 않을까 생각해서 아아디어를 냈습니다
예를들어 5번까지 있으면 0번 먼저 타겟에 맞게 1번 맞게 쭉~~~ 하면 되더라구요.

위와 같이 하는데 뒷 부분은 그냥 교체를 해주면 되는데

💡 0번 버튼의 예외를 따로 둬야하더라고요

이유는 0번을 누르냐 안누르냐에 따라 계산 방식 전부가 달라져서 그렇습니다.
아래 그림 예제를 그리면서 이해하니 쉽게 이해가 가더라구요

위와 같이 그리디를 통한 현재 상황에서 가장 좋아보이는 버튼을 눌러 전구를 키도록 했습니다!

코드도 몇번 햇갈리긴 했지만

1. 각 0번 버튼을 위해서 함수로 만들었습니다.
2. 버튼을 1번부터 N까지 순회
   1. i-1전구를 비교해서 목표와 같으면 넘어가고
   2. 다르다면 i-1 ~ i+2(2번 버튼이면 1번, 2번, 3번을 바꾸기 위해)을 바꿔줬습니다!
   3. 그리고 최종적으로 target과 같으면 count return
   4. 다르면 -1 return하도록 했습니다!
3. 출력하기
   1. 결과1이 -1이면 결과2 출력
   2. 결과2가 -1이면 결과1 출력
   3. 둘다 -1이 아니면 둘 중 작은 부분 출력 -> 둘다 진행은 되지만 더 작은게 있을 수도 있음

위에서 i-1을 하는 이유는 i로 하게 되면 최종적으로 i+1번일때 다시 바뀔 수 있으므로 마지막으로 건드리는 i-1을 선택하게 됐습니다!

사실 출력을 더 깔끔하게 만들 수 있을 것 같은데 뇌가 비명을 질러서...

⭐️ 최종 코드

def solution(first, target, count):
    # 첫번째 버튼을 누른 경우 처리
    if count == 1:
        first[0] = 1 - first[0]
        first[1] = 1 - first[1]
    # 버튼 순회
    for i in range(1,n):
        if first[i-1] == target[i-1]:   # 바꿔야하는지 비교
            continue
        count += 1
        for j in range(i-1, i+2):   # 1번이면 0~2번까지 전구 바꾸기
            if j < n:   # n이 3일때 3을 누르면 2,3만 되도록
                first[j] = 1 - first[j]
    if first == target:
        return count
    else :
        return -1
    
n = int(input())
first = list(map(int, input()))
target = list(map(int, input()))

# 첫번째 전구의 스위치를 누르는 경우
result1 = solution(first[:], target, 1)
# 첫번째 전구의 스위치를 누르지 않는 경우
result2 = solution(first[:], target, 0)

if result1 == -1:
    print(result2)
elif result2 == -1:
    print(result1)
else : 
    print(min(result1, result2))

📚 새롭게 알게된 내용

그리디 개념을 어제 어느정도 정리하고 오늘 문제를 풀어봤는데 풀려서 기분이 좋네요!! 근데 0번 버튼 처리를 따로 해줘야 하는 부분에서 막혀서 잠시 인터넷 참조를 했었습니다!

[wkdghdwns199]

해설을 보면서 이해가 안되는 부분이 있는데, 순차적으로 버튼을 누르는 것이 아닌 다른 위치의 버튼을 먼저 누르는 것이 최소의 버튼 누르는 횟수가 될 수도 있지 않나요? 다른 해설들도 보는데 민석님과 같은 해설이더라구요..! 🤔

[fnzksxl]

import sys

input = sys.stdin.readline

N = int(input())

_init = list(map(int, input().rstrip()))
_goal = list(map(int, input().rstrip()))
tr_init = _init.copy()
tr_init[:2] = [0 if j else 1 for j in _init[:2]]

def return_cnt(_init, _goal):
  cnt = 0
  for i in range(1,len(_init)):
    if _init[i-1] == _goal[i-1]:
      continue
    
    _init[i-1:i+2] = [0 if j else 1 for j in _init[i-1:i+2]]
    cnt += 1
  
  if _init == _goal:
    return cnt
  else:
    return -1

tr = return_cnt(tr_init, _goal)
not_tr = return_cnt(_init, _goal)

if not_tr >= 0:
  print(not_tr)
elif tr >= 0:
  print(tr+1)
else:
  print(-1)

처음에는 다른 전구에서 변환을 시도했었는데 그러면 앞의 전구 상태가 보장이 안 되더라구요.
그래서 바로 뒤 전구에서 변환하는 걸로 해서 코드를 짰는데 막막해서 민석님 풀이를 좀 참고했습니다.
0번째 인덱스 변화 유무가 중요했던 거네요..

[alstjr7437]

해설을 보면서 이해가 안되는 부분이 있는데, 순차적으로 버튼을 누르는 것이 아닌 다른 위치의 버튼을 먼저 누르는 것이 최소의 버튼 누르는 횟수가 될 수도 있지 않나요? 다른 해설들도 보는데 민석님과 같은 해설이더라구요..! 🤔

1 0 1 1

1번 클릭 > 4번 클릭
4번 클릭 > 1번 클릭

0 1 0 0

0 0 0 0 0

3번 클릭

0 1 1 1 0

위와 같이 어짜피 어떤 버튼을 누르는지는 똑같게 나오고 버튼 횟수의 차이는 없을 것 같아요!!
대신 다른 위치의 버튼을 먼저 누르게 되면 2번째 예제에서 처럼 1번 2번 안누르고 3번에 가는 것과 3번 바로 가는 것으로
답을 찾는 속도가 빨라지니 시간복잡도가 좋겠지만
태원님이 위에 말하신대로 앞에 전구의 상태가 보장이 안되서 아예 답이 안되는 경우가 생길 수도 있어서 그런 것 같네요!(개인적인 생각입니다)