[baekjoon]python #1244 스위치 켜고 끄기

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

1244번: 스위치 켜고 끄기

N = int(input()) #스위치 개수
#스위치 상태 리스트. 스위치 번호 헷갈리니까 0번인덱스에 0과1이 아닌 아무거나 넣어줌
switch=[-1]+ list(map(int,input().split()))
S = int(input()) # 학생 수
for i in range(S): #학생 성별, 받은 수
    gen,num = map(int,input().split())

    if gen == 1: #학생이 남자일 때
        for j in range(num,N+1,num): #배수를 찾아 스위치 바꾸기
            if switch[j] == 0:
                switch[j] = 1
            else:
                switch[j] = 0


    elif gen == 2: #학생이 여자일 때

        #자기 자신의 스위치 바꾸기
        if switch[num] == 1:
            switch[num] = 0
        else:
            switch[num] = 1

        #대칭 되는 값 찾기
        #어디가 끝인지 모르니까 while문 이용하기

        move=1 #이동하기 위한 변수
        while(num-move > 0 and num+move < N+1 and switch[num+move]==switch[num-move]):
            #대칭 되는 값들 바꿔주기
            if switch[num+move] == 1:
                switch[num+move],switch[num-move] =0,0
            elif switch[num+move] == 0:
                switch[num + move], switch[num - move] = 1,1
            move+=1

# 처음에 넣어준 0번 인덱스 삭제
result = switch[1:]

#20개씩 출력하기
for k in range(0,N,20):
    print(*result[k:k+20])