[백준] 12927_배수 스위치 python

김동완·2022년 4월 19일
0

알고리즘

목록 보기
21/55
post-thumbnail

배수_스위치

문제

강호는 전구 N개를 가지고 있다. 전구는 1번부터 N번까지 번호가 매겨져 있으며, 일렬로 놓여져 있다. 전구는 켜져있거나 꺼져있다.

강호는 모든 전구를 끄려고 한다. 강호는 전구를 켜고 끌 수 있는 스위치 N개를 가지고 있고, 스위치도 1번부터 N번까지 번호가 매겨져 있다. i번 스위치는 i의 배수 번호를 가지는 전구의 상태를 모두 반전시킨다.

현재 전구의 상태가 주어졌을 때, 모든 전구를 끄기 위해서 스위치를 몇 번 눌러야하는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 전구의 상태가 1번 전구부터 차례대로 주어진다. Y는 전구가 켜 있는 경우, N은 전구가 꺼져있는 경우이다. 전구의 개수는 1보다 크거나 같고 1,000보다 작거나 같은 자연수이다.

출력

모든 전구를 끄기 위해서 스위치를 몇 번 눌러야 하는지 출력한다. 만약, 모든 전구를 끌 수 없다면 -1을 출력한다.

해결 방안

  • 스위치가 모두 꺼질때가지 반복한다.
  • 스위치를 돌면서 현재 스위치가 N이면 건너뛰고
  • 현재 스위치가 y이면 현재 스위치의 배수를 다 반전시킨다.
switch = [0]+list(input())
cnt = 0

for i in range(1,len(switch)) :
    if switch == [0]+['N']*(len(switch)-1):
        break 
    if switch[i] == 'N' :
        continue
    for j in range(i,len(switch)) :
        if j%i == 0 :
            if switch[j] == 'Y':
                switch[j] = 'N'
            else :
                switch[j] = 'Y'
    cnt += 1 
print(cnt)
profile
내가 공부한 내용들이 누군가에게 도움이 될지 몰라서 쓰는 벨로그

0개의 댓글