백준 1244 스위치 켜고 끄기 - 파이썬

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

1244번: 스위치 켜고 끄기

 

 

# 스위치 저장할 리스트

귀찮게 스위치 번호가 1번부터 시작한다.

이걸 계산할 때마다 고려해줘도 되고, 리스트 앞에 0번을 추가해놓아도 된다. 나는 후자로 놓고 풀었다.

취향껏 작성하면 된다.

단, 후자는 여학생이 있지도 않은 0번 스위치을 켜고 끄지 않도록 유의해야 한다.

 

 

# 스위치 바꾸는 함수

이것도 취향껏 작성하면 된다.

dictionary에 {1:0, 0:1} 이런 식으로 넣어도 되고 if문 빡구현해도 된다.

 

 

# 남자일 때

for문을 사용해도 된다. 나는 while문을 이용해서 i를 1부터 하나씩 늘려 가면서 범위에 맞는 no * i를 바꿔줬다.

 

 

# 여자일 때

일단 누른 스위치의 상태를 바꿔준다.

이후 i를 1부터 하나씩 증가시키며 스위치 ± i 가 리스트 범위 내이고, 둘의 상태가 같다면 둘 다 상태를 바꿔준다.

조건을 만족시키지 못할 때 바로 break해서 빠져 나와 준다.

 

 

# 출력

1번 ~ n번 스위치의 상태를 출력해준다.

20개를 출력할 때마다 꼭 줄바꿈을 해야한다.

 

 

 

가능한 오답 원인 : 출력 시 줄바꿈, 범위 체크 다시 하기(여학생이 switch 리스트에서 index 0번 만지고 있는 지)

 

 

n = int(input())
switch = [0] + list(map(int, input().split())) # 스위치 상태 저장

# 스위치 바꾸는 함수
def change(x):
    global switch
    switch[x] = abs(switch[x]-1)


for _ in range(int(input())):
    gender, no = map(int, input().split())
    i = 1
    # 남자일 때
    if gender == 1:
        while no * i <= n:
            change(no * i)
            i += 1
    # 여자일 때
    elif gender == 2:
        change(no)
        while 1 <= no-i and no+i <= n and switch[no-i] == switch[no+i]:
            change(no-i)
            change(no+i)
            i += 1

# 출력
for i in range(1, n+1):
    print(switch[i], end=" ")
    if not i % 20:
        print()

 

 

---

스위치 번호를 하나씩 줄여서 우겨 넣어서 풀면 아래와 같다.

+ map을 이용해서 스위치 상태를 바꿔주었다.

사실 똑같다.

 

n = int(input())
switch = list(map(int, input().split())) # 스위치 상태 저장할 리스트
change = {1: 0, 0: 1} # 스위치 바꾸는 dictionary

for _ in range(int(input())):
    gender, no = map(int, input().split())
    i = 1
    # 남자일 때
    if gender == 1:
        while no * i - 1 < n:
            switch[no*i-1] = change[switch[no*i-1]]
            i += 1
    # 여자일 때
    elif gender == 2:
        switch[no-1] = change[switch[no-1]]
        while 1 <= no-i and no+i < n+1 and switch[no-i-1] == switch[no-1+i]:
            switch[no-1-i] = change[switch[no-1-i]]
            switch[no-1+i] = change[switch[no-1+i]]
            i += 1

# 출력
for i in range(n):
    print(switch[i], end=" ")
    if i % 20 == 19:
        print()