[백준] 1244 스위치 켜고 끄기 (python 파이썬)

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

---

생각보다 오래 걸린 문제. 팰린드롬 처리하는 게 약한 거 같다.

그리고 특이하게 정답을 숫자 20개씩 끊어서 출력해야 한다. 덕분에 실수를 많이 했다.

---

아이디어

- 일단 배열의 시작은 0이고 스위치 번호는 1부터 시작이니까, 인덱스를 먼저 맞혀주자.

간단하게 배열 맨 앞에 -1 같은 더미 값을 추가해 줬다.

 

- 동일한 기능을 하는 코드가 중복되어 사용되므로, 함수를 만들어 사용했다. 처음 문제 풀 때는 그냥 풀긴 했다.

---

전체 코드

import sys

N = int(input())
switchs = [-1] + [int(x) for x in sys.stdin.readline().rstrip().split()]
students = int(input())

def click(x):
    if switchs[x] == 1:
        switchs[x] = 0
    else:
        switchs[x] = 1

for _ in range(students):
    commands = [int(x) for x in sys.stdin.readline().rstrip().split()]
    gender, num = commands[0], commands[1]

    if gender == 1: # 남자면
        for i in range(num, N + 1, num):
            click(i)
    else:
        click(num)
        for k in range(N // 2):
            if num + k > N or num - k < 1:
                break
            if switchs[num + k] == switchs[num - k]:
                click(num - k)
                click(num + k)
            else:
                break

sliced_lists = [switchs[i:i+20] for i in range(1, len(switchs), 20)]

for s in sliced_lists:
    print(" ".join(map(str, s)))

---

코드 설명

for _ in range(students):
    commands = [int(x) for x in sys.stdin.readline().rstrip().split()]
    gender, num = commands[0], commands[1]

    if gender == 1: # 남자면
        for i in range(num, N + 1, num):
            click(i)
    else:
        click(num)
        for k in range(N // 2):
            if num + k > N or num - k < 1:
                break
            if switchs[num + k] == switchs[num - k]:
                click(num - k)
                click(num + k)
            else:
                break

 

성별이 남자면 간단하다. for문 스텝을 num크기만큼 주면 된다.

여자면 일단 선택된 스위치 건들고, 좌우가 동일한지 확인하며 범위를 늘려주면 된다. 

 

sliced_lists = [switchs[i:i+20] for i in range(1, len(switchs), 20)]

for s in sliced_lists:
    print(" ".join(map(str, s)))

 

출력은 for문의 스텝을 응용해 봤다.