실버III. 백준 1244 스위치 켜고 끄기

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

1244번: 스위치 켜고 끄기

단순 구현문제이지만, 파이썬 입력받는 코드가 익숙하지 않아 많이 헤맸다..

import sys

N = 0
switches = []
students = 0


def solve():
    init()
    switch()
    answer()

def init():
    global N, switches, students

    N = int(input())
    switches = [-99] + list(map(int, input().split()))
    students = int(input())
    return

def switch():
    global N, switches, students

    for _ in range(students):
        gender, switch_number = list(map(int, input().split()))
        if gender == 1:
            for i in range(switch_number, N + 1, switch_number):
                change(i)
        else:
            change(switch_number)
            move = 1
            while True:
                left = switch_number - move
                right = switch_number + move

                if left < 1 or right > N:
                    break
                if switches[left] == switches[right]:
                    change(left)
                    change(right)
                    move +=1
                else:
                    break

def change(switch_number):
    global switches
    if switches[switch_number] == 1:
        switches[switch_number] = 0
    else:
        switches[switch_number] = 1
        
def answer():
    global switches
    for i in range(1, N + 1):
        print(switches[i], end=" ")
        if i % 20 == 0:
            print()

if __name__ == '__main__':
    solve()