[백준 문제풀이] 얼렁뚱땅 1244 스위치 켜고 끄기 풀이

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

1244번: 스위치 켜고 끄기

 

n = int(input())
arr = [-1]+list(map(int,input().split()))
stu_num = int(input())
stu_arr = [list(map(int,input().split())) for _ in range(stu_num)]

for i in range(stu_num) :
    if stu_arr[i][0] == 1 :
        start = stu_arr[i][1]
        new_start = start
        how = 2
        while True :
            if new_start <= n :
                if arr[new_start] == 1 :
                    arr[new_start]= 0
                else :
                    arr[new_start] = 1
                new_start = start * how
                how += 1
            else :
                break
    else :
        start = stu_arr[i][1]
        down = start - 1
        up = start + 1
        if arr[start] == 0 :
            arr[start]= 1
        else :
            arr[start] = 0
        while True:

            if 1<=down<=n and 1<=up<=n :
                if arr[down] == arr[up] :
                    if arr[down] == 1 :
                        arr[down]= 0
                        arr[up] = 0
                    else :
                        arr[down] = 1
                        arr[up] = 1
                    down -= 1
                    up += 1
                else :
                    break
            else :
                break



arr = arr[1:]

for i in range(len(arr)//20 + 1) :
    if (i+1)*20 + 1 > n :
        print(*arr[i*20:])
    else :
        print(*arr[i*20:(i+1)*20])